diff --git a/python/rainbow/util/test_tools.py b/python/rainbow/util/test_tools.py index d4caf88..a0212de 100644 --- a/python/rainbow/util/test_tools.py +++ b/python/rainbow/util/test_tools.py @@ -6,7 +6,7 @@ """ -def is_array_equal(arr1, arr2, dec=8): +def is_array_equal(arr1, arr2): """ 2022-04-06 Kenny TODO: Write proper documentation. @@ -15,7 +15,7 @@ def is_array_equal(arr1, arr2, dec=8): :param dec: :return: """ - return None == np.testing.assert_array_almost_equal(arr1, arr2, dec) + return None == np.testing.assert_allclose(arr1, arr2, rtol=1e-5, atol=1e-8) def is_array_not_equal(arr1, arr2):