check against all

Bchass · Jun 7, 2024 · 984b14a · 984b14a
1 parent ee74ddd
commit 984b14a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tinynumpy/tests/test_tinynumpy.py b/tinynumpy/tests/test_tinynumpy.py
@@ -469,7 +469,7 @@ def test_reshape():
     # Test reshape
     reshaped_a = a.reshape((4, 2))
     reshaped_b = b.reshape((4, 2))
-    assert np.array_equal(reshaped_a, reshaped_b)
+    assert(reshaped_a == reshaped_b).all()
 
     # Fail
     with raises(ValueError):  # Invalid shape