TST: unfail test_numeric/TestBoolCmp

Author: ev-br

torch_np/tests/numpy_tests/core/test_numeric.py

@@ -453,7 +453,6 @@ def test_logical_and_or_xor(self):
         assert_array_equal(self.im ^ False, self.im)
 
 
-@pytest.mark.xfail(reason="TODO: needs fancy indexing")
 class TestBoolCmp:
     def setup_method(self):
         self.f = np.ones(256, dtype=np.float32)
@@ -515,10 +514,6 @@ def test_float(self):
             r3 = 0 != self.f[i:]
             assert_array_equal(r, r2)
             assert_array_equal(r, r3)
-            # check bool == 0x1
-            assert_array_equal(r.view(np.int8), r.astype(np.int8))
-            assert_array_equal(r2.view(np.int8), r2.astype(np.int8))
-            assert_array_equal(r3.view(np.int8), r3.astype(np.int8))
 
             # isnan on amd64 takes the same code path
             assert_array_equal(np.isnan(self.nf[i:]), self.ef[i:])
@@ -527,6 +522,17 @@ def test_float(self):
             assert_array_equal(np.isinf(self.inff[i:]), self.efnonan[i:])
             assert_array_equal(np.signbit(self.signf[i:]), self.ef[i:])
 
+    @pytest.mark.xfail(reason='no .view method')
+    def test_float_2(self):
+        for i in range(4):
+            r = self.f[i:] != 0
+            r2 = self.f[i:] != np.zeros_like(self.f[i:])
+            r3 = 0 != self.f[i:]
+            # check bool == 0x1
+            assert_array_equal(r.view(np.int8), r.astype(np.int8))
+            assert_array_equal(r2.view(np.int8), r2.astype(np.int8))
+            assert_array_equal(r3.view(np.int8), r3.astype(np.int8))
+
     def test_double(self):
         # offset for alignment test
         for i in range(2):
@@ -541,10 +547,6 @@ def test_double(self):
             r3 = 0 != self.d[i:]
             assert_array_equal(r, r2)
             assert_array_equal(r, r3)
-            # check bool == 0x1
-            assert_array_equal(r.view(np.int8), r.astype(np.int8))
-            assert_array_equal(r2.view(np.int8), r2.astype(np.int8))
-            assert_array_equal(r3.view(np.int8), r3.astype(np.int8))
 
             # isnan on amd64 takes the same code path
             assert_array_equal(np.isnan(self.nd[i:]), self.ed[i:])
@@ -553,6 +555,18 @@ def test_double(self):
             assert_array_equal(np.isinf(self.infd[i:]), self.ednonan[i:])
             assert_array_equal(np.signbit(self.signd[i:]), self.ed[i:])
 
+    @pytest.mark.xfail(reason='no .view method')
+    def test_double_2(self):
+        for i in range(4):
+            r = self.d[i:] != 0
+            r2 = self.d[i:] != np.zeros_like(self.d[i:])
+            r3 = 0 != self.d[i:]
+
+            # check bool == 0x1
+            assert_array_equal(r.view(np.int8), r.astype(np.int8))
+            assert_array_equal(r2.view(np.int8), r2.astype(np.int8))
+            assert_array_equal(r3.view(np.int8), r3.astype(np.int8))
+
 
 @pytest.mark.xfail(reason="TODO")
 class TestSeterr: