add more tests

tests-cuda/test_3140_cuda_slicing.py

@@ -680,8 +680,8 @@ def test_0127_tomask_operation():
 
 
 def test_simple_slice_cpu():
- arrg_cpu = ak.Array([[1, 2, 3], [0], [4, 5]])
- out = arrg_cpu[:, 0]
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]])
+ out = arr[:, 0]
  expected = [1, 0, 4]
  result = out.tolist()
  cp.testing.assert_array_list_equal(
@@ -692,12 +692,76 @@ def test_simple_slice_cpu():
 
 
 def test_simple_slice_gpu():
- arrg_gpu = ak.Array([[1, 2, 3], [0], [4, 5]], backend="cuda")
- out = arrg_gpu[:, 0]
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]], backend="cuda")
+ out = arr[:, 0]
  expected = [1, 0, 4]
  result = out.tolist()
  cp.testing.assert_array_list_equal(
  result,
  expected,
  err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}",
  )
+
+
+def test_simple_slice_cpu1():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]])
+ out = arr[:,1:]
+ expected = [[2, 3], [], [5]] 
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")
+
+
+def test_simple_slice_gpu1():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]], backend="cuda")
+ out = arr[:,1:]
+ expected = [[2, 3], [], [5]] 
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")
+
+
+def test_simple_slice_cpu2():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]])
+ out = arr[:,:1]
+ expected = [[1], [0], [4]]
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")
+
+
+def test_simple_slice_gpu2():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]], backend="cuda")
+ out = arr[:,:1]
+ expected = [[1], [0], [4]]
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")
+
+
+def test_simple_slice_cpu3():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]])
+ out = arr[:,1::2]
+ expected = [[2], [], [5]]
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")
+
+
+def test_simple_slice_gpu3():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]], backend="cuda")
+ out = arr[:,1::2]
+ expected = [[2], [], [5]]
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")
+
+
+def test_simple_slice_cpu4():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]])
+ out = arr[:,::-1]
+ expected = [[3, 2, 1], [0], [5, 4]]
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")
+
+
+def test_simple_slice_gpu4():
+ arr = ak.Array([[1, 2, 3], [0], [4, 5]], backend="cuda")
+ out = arr[:,::-1]
+ expected = [[3, 2, 1], [0], [5, 4]]
+ result = out.tolist()
+ cp.testing.assert_array_list_equal(result, expected, err_msg=f"Slice of [[1, 2, 3], [0], [4, 5]] should be {expected}, but got {result}")