add more split_at tests

Author: vbarrielle

tests/array.rs

@@ -7,7 +7,7 @@ extern crate ndarray;
 use ndarray::{RcArray, S, Si,
     OwnedArray,
 };
-use ndarray::{arr0, arr1, arr2,
+use ndarray::{arr0, arr1, arr2, arr3,
     aview0,
     aview1,
     aview2,
@@ -683,4 +683,25 @@ fn split_at() {
     let (left, right) = b.view().axis_split_at(2, 2);
     assert_eq!(left.shape(), [3, 4, 2]);
     assert_eq!(right.shape(), [3, 4, 3]);
+    assert_eq!(left, arr3(&[[[0., 1.], [5., 6.], [10., 11.], [15., 16.]],
+                            [[20., 21.], [25., 26.], [30., 31.], [35., 36.]],
+                            [[40., 41.], [45., 46.], [50., 51.], [55., 56.]]]));
+
+    // we allow for an empty right view when index == dim[axis]
+    let (_, right) = b.view().axis_split_at(1, 4);
+    assert_eq!(right.shape(), [3, 0, 5]);
+}
+
+#[test]
+#[should_panic]
+fn deny_split_at_axis_out_of_bounds() {
+    let a = arr2(&[[1., 2.], [3., 4.]]);
+    a.view().axis_split_at(2, 0);
+}
+
+#[test]
+#[should_panic]
+fn deny_split_at_index_out_of_bounds() {
+    let a = arr2(&[[1., 2.], [3., 4.]]);
+    a.view().axis_split_at(1, 3);
 }