MAINT,TST: Reduce np.testing to IS_PYPY

Co-authored-by: Melissa Weber Mendonça <melissawm@gmail.com>

1 files changed, 11 insertions, 11 deletions

diff --git a/numpy/f2py/tests/test_size.py b/numpy/f2py/tests/test_size.py
index 3360e2a3d..bd2c349df 100644
--- a/numpy/f2py/tests/test_size.py
+++ b/numpy/f2py/tests/test_size.py
@@ -1,7 +1,7 @@
 import os
 import pytest
+import numpy as np
 
-from numpy.testing import assert_equal
 from . import util
 
 
@@ -11,35 +11,35 @@ class TestSizeSumExample(util.F2PyTest):
     @pytest.mark.slow
     def test_all(self):
         r = self.module.foo([[]])
-        assert_equal(r, [0], repr(r))
+        assert r == [0]
 
         r = self.module.foo([[1, 2]])
-        assert_equal(r, [3], repr(r))
+        assert r == [3]
 
         r = self.module.foo([[1, 2], [3, 4]])
-        assert_equal(r, [3, 7], repr(r))
+        assert np.allclose(r, [3, 7])
 
         r = self.module.foo([[1, 2], [3, 4], [5, 6]])
-        assert_equal(r, [3, 7, 11], repr(r))
+        assert np.allclose(r, [3, 7, 11])
 
     @pytest.mark.slow
     def test_transpose(self):
         r = self.module.trans([[]])
-        assert_equal(r.T, [[]], repr(r))
+        assert np.allclose(r.T, np.array([[]]))
 
         r = self.module.trans([[1, 2]])
-        assert_equal(r, [[1], [2]], repr(r))
+        assert np.allclose(r, [[1.], [2.]])
 
         r = self.module.trans([[1, 2, 3], [4, 5, 6]])
-        assert_equal(r, [[1, 4], [2, 5], [3, 6]], repr(r))
+        assert np.allclose(r, [[1, 4], [2, 5], [3, 6]])
 
     @pytest.mark.slow
     def test_flatten(self):
         r = self.module.flatten([[]])
-        assert_equal(r, [], repr(r))
+        assert np.allclose(r, [])
 
         r = self.module.flatten([[1, 2]])
-        assert_equal(r, [1, 2], repr(r))
+        assert np.allclose(r, [1, 2])
 
         r = self.module.flatten([[1, 2, 3], [4, 5, 6]])
-        assert_equal(r, [1, 2, 3, 4, 5, 6], repr(r))
+        assert np.allclose(r, [1, 2, 3, 4, 5, 6])