Add boolean array indexing test for multi-dim array

cupy · Nov 27, 2016 · 0e70665 · 0e70665
1 parent c95fa9f
commit 0e70665
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/cupy_tests/core_tests/test_ndarray_indexing.py b/tests/cupy_tests/core_tests/test_ndarray_indexing.py
@@ -16,6 +16,8 @@
      'indexes': (slice(None), slice(None, -1), slice(-2))},
     {'shape': (2, 3, 4), 'transpose': (2, 0, 1),
      'indexes': (slice(None), slice(None, 1), slice(2))},
+    {'shape': (2, 3, 4), 'transpose': None,
+     'indexes': numpy.random.choice([False, True], (2, 3, 4))},
     {'shape': (2, 3, 5), 'transpose': None,
      'indexes': (slice(None, None, -1), slice(1, None, -1), slice(4, 1, -2))},
     {'shape': (2, 3, 5), 'transpose': (2, 0, 1),