[DNM] Add simple slice test to cuda (#3242)

Co-authored-by: Martin Durant <[email protected]> Co-authored-by: Jim Pivarski <[email protected]>
scikit-hep · Nov 7, 2024 · 307352f · 307352f
1 parent f3124c6
commit 307352f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests-cuda/test_3140_cuda_slicing.py b/tests-cuda/test_3140_cuda_slicing.py
@@ -457,6 +457,12 @@ def test_1904_drop_none_RecordArray():
     ]
 
 
+def test_simple_slice():
+    arrg = ak.Array([[1, 2, 3], [0], [4, 5]], backend="cuda")
+    out = arrg[:, 0]
+    assert out.tolist() == [1, 0, 4]
+
+
 def test_2246_slice_not_packed():
     index = ak.Array(
         ak.contents.ListOffsetArray(