Adds a test for different types of arrays.

Author: chaburkland

test/test_util.py

@@ -384,22 +384,32 @@ def test_dtype_from_element_str_and_bytes_dtypes(self) -> None:
 
     def test_get_new_indexers_and_screen_a(self) -> None:
         indexersA = np.array([9, 9, 9, 9, 0, 0, 1, 4, 5, 0, 0, 0, 1], dtype=np.int64)
-        postB = get_new_indexers_and_screen_ak(indexersA, np.arange(10, dtype=np.int64))
-        assert tuple(map(list, postB)) == (
+        postA = get_new_indexers_and_screen_ak(indexersA, np.arange(10, dtype=np.int64))
+        assert indexersA.flags.c_contiguous
+        assert indexersA.flags.f_contiguous
+        assert tuple(map(list, postA)) == (
             [0, 0, 0, 0, 1, 1, 2, 3, 4, 1, 1, 1, 2],
             [9, 0, 1, 4, 5],
         )
 
-        indexersB = np.array([9, 9, 9, 9, 0, 0, 1, 4, 5, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], dtype=np.int64)
-        postB = get_new_indexers_and_screen_ak(indexersB, positions=np.arange(15, dtype=np.int64))
-        assert tuple(map(list, postB)) == (
+        # Prove we can handle non-continuous arrays
+        indexersB = np.full((len(indexersA), 3), -1)
+        indexersB[:,1] = indexersA.copy()
+        assert not indexersB[:,1].flags.c_contiguous
+        assert not indexersB[:,1].flags.f_contiguous
+        postB = get_new_indexers_and_screen_ak(indexersB[:,1], np.arange(10, dtype=np.int64))
+        assert tuple(map(list, postA)) == tuple(map(list, postB))
+
+        indexersC = np.array([9, 9, 9, 9, 0, 0, 1, 4, 5, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], dtype=np.int64)
+        postC = get_new_indexers_and_screen_ak(indexersC, positions=np.arange(15, dtype=np.int64))
+        assert tuple(map(list, postC)) == (
             [0, 0, 0, 0, 1, 1, 2, 3, 4, 1, 1, 1, 2, 5, 6, 3, 4,7, 8, 9, 0, 10],
             [9, 0, 1, 4, 5, 2, 3, 6, 7, 8, 10],
         )
 
-        indexersC = np.array([2, 1, 0, 2, 0, 1, 1, 2, 0], dtype=np.int64)
-        postC = get_new_indexers_and_screen_ak(indexers=indexersC, positions=np.arange(3, dtype=np.int64))
-        assert tuple(map(list, postC)) == (
+        indexersD = np.array([2, 1, 0, 2, 0, 1, 1, 2, 0], dtype=np.int64)
+        postD = get_new_indexers_and_screen_ak(indexers=indexersD, positions=np.arange(3, dtype=np.int64))
+        assert tuple(map(list, postD)) == (
             [2, 1, 0, 2, 0, 1, 1, 2, 0],
             [0, 1, 2],
         )