Skip to content

Commit d2c9675

Browse files
committed
changed array comparison
1 parent fad7d59 commit d2c9675

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import unittest
55
from pathlib import Path
66
import numpy as np
7-
from numpy.testing import assert_array_almost_equal
7+
from numpy.testing import assert_allclose
88

99

1010
class BaseTestCase(unittest.TestCase):
@@ -67,6 +67,6 @@ def compare_files(cls, simulation, expected_folder, sim_type):
6767
result_calc = np.loadtxt(calc_path, skiprows=1)
6868
result_expected = np.loadtxt(expected_path, skiprows=1)
6969
if 'fit_toc' in file_name:
70-
assert_array_almost_equal(result_calc, result_expected, decimal=2)
70+
assert_allclose(result_calc, result_expected, atol=0, rtol=1E-6)
7171
else:
72-
assert_array_almost_equal(result_calc, result_expected)
72+
assert_allclose(result_calc, result_expected, atol=0, rtol=1E-3)

0 commit comments

Comments
 (0)