Fix test

apache · Feb 7, 2020 · 1faf352 · 1faf352
1 parent d71f23b
commit 1faf352
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/python/mxnet/test_utils.py b/python/mxnet/test_utils.py
@@ -52,7 +52,6 @@
 from .symbol.numpy import _Symbol as np_symbol
 from .util import use_np  # pylint: disable=unused-import
 from .runtime import Features
-from .numpy_extension import get_cuda_compute_capability
 
 
 def default_context():

diff --git a/tests/python/unittest/test_numpy_interoperability.py b/tests/python/unittest/test_numpy_interoperability.py
@@ -35,6 +35,9 @@
 _INT_DTYPES = [np.int8, np.int32, np.int64, np.uint8]
 _FLOAT_DTYPES = [np.float16, np.float32, np.float64]
 _DTYPES = _INT_DTYPES + _FLOAT_DTYPES
+_SKIP_ASSERT_EQUAL_LIST = [
+    'empty_like',
+]
 
 
 class OpArgMngr(object):
@@ -1941,11 +1944,17 @@ def _check_interoperability_helper(op_name, *args, **kwargs):
         assert type(out) == type(expected_out)
         for arr, expected_arr in zip(out, expected_out):
             if isinstance(arr, np.ndarray):
-                assert_almost_equal(arr.asnumpy(), expected_arr, rtol=1e-3, atol=1e-4, use_broadcast=False, equal_nan=True)
+                if op_name in _SKIP_ASSERT_EQUAL_LIST:
+                    assert arr.shape == expected_arr.shape
+                else:
+                    assert_almost_equal(arr.asnumpy(), expected_arr, rtol=1e-3, atol=1e-4, use_broadcast=False, equal_nan=True)
             else:
                 _np.testing.assert_equal(arr, expected_arr)
     elif isinstance(out, np.ndarray):
-        assert_almost_equal(out.asnumpy(), expected_out, rtol=1e-3, atol=1e-4, use_broadcast=False, equal_nan=True)
+        if op_name in _SKIP_ASSERT_EQUAL_LIST:
+            assert out.shape == expected_out.shape
+        else:
+            assert_almost_equal(out.asnumpy(), expected_out, rtol=1e-3, atol=1e-4, use_broadcast=False, equal_nan=True)
     else:
         _np.testing.assert_almost_equal(out, expected_out)