Specifically predicate np.newaxis (not None) in indices tests

Co-authored-by: Zac Hatfield-Dodds <zac.hatfield.dodds@gmail.com>
HypothesisWorks · Aug 28, 2021 · 6758426 · 6758426
1 parent 8e7117a
commit 6758426
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hypothesis-python/tests/numpy/test_gen_data.py b/hypothesis-python/tests/numpy/test_gen_data.py
@@ -1053,8 +1053,8 @@ def test_advanced_integer_index_can_generate_any_pattern(shape, data):
     [
         lambda ix: Ellipsis in ix,
         lambda ix: Ellipsis not in ix,
-        lambda ix: None in ix,
-        lambda ix: None not in ix,
+        lambda ix: np.newaxis in ix,
+        lambda ix: np.newaxis not in ix,
     ],
 )
 def test_basic_indices_options(condition):
@@ -1121,7 +1121,7 @@ def test_basic_indices_generate_valid_indexers(
             assert 0 <= len(indexer) <= len(shape) + int(allow_ellipsis)
         else:
             assert 1 <= len(shape) + int(allow_ellipsis)
-        assert None not in shape
+        assert np.newaxis not in shape
     if not allow_ellipsis:
         assert Ellipsis not in shape