Add negative index test
@@ -54,6 +54,12 @@ class TestSplit(unittest.TestCase):
split = xp.split(a, (2, 4, 9), -1)
return xp.concatenate(split, -1)
+ @testing.numpy_cupy_array_equal()
+ def test_split_by_sections3(self, xp):
+ a = testing.shaped_arange((3, 11), xp)
+ split = xp.split(a, (-9, 4, -2), 1)
+ return xp.concatenate(split, 1)
+
@testing.numpy_cupy_array_equal()
def test_vsplit(self, xp):
a = testing.shaped_arange((12, 3), xp)
