changed tolerances

ricsinaruto · Mar 27, 2023 · 472b7ca · 472b7ca
1 parent ddc40f4
commit 472b7ca
Show file tree

Hide file tree

Showing 10 changed files with 10 additions and 10 deletions.
diff --git a/tests/test_lda_nn_chn_multiclass.py b/tests/test_lda_nn_chn_multiclass.py
@@ -21,7 +21,7 @@
 
     # assert that losses are the same as true_losses to 4 decimal places
     for loss, val in zip(losses, true_losses):
-        assert round(loss, 2) == round(val, 2)
+        assert abs(loss-val) < 0.005
 
 # check that model.pt25 exists
 path = os.path.join(directory, 'model.pt25')

diff --git a/tests/test_lda_nn_fe_multiclass.py b/tests/test_lda_nn_fe_multiclass.py
@@ -12,7 +12,7 @@
     loss = float(lines[0].strip())
 
     # assert that loss is the same as val to 4 decimal places
-    assert round(loss, 2) == round(val, 2)
+    assert abs(loss-val) < 0.005
 
 # check that model.pt25 exists
 path = os.path.join(directory, 'model.pt25')

diff --git a/tests/test_lda_nn_fe_multiclass2pairwise.py b/tests/test_lda_nn_fe_multiclass2pairwise.py
@@ -11,4 +11,4 @@
 loss = float(lines[0].strip())
 
 # assert that loss is the same as val to 4 decimal places
-assert round(loss, 2) == round(val, 2)
+assert abs(loss-val) < 0.005
diff --git a/tests/test_lda_nn_fe_multiclass_spatialPFI.py b/tests/test_lda_nn_fe_multiclass_spatialPFI.py
@@ -14,7 +14,7 @@
 
 # assert that pfi is the same as true_pfi to 4 decimal places
 # use numpy
-assert np.allclose(pfi, true_pfi, rtol=1e-05, atol=1e-08)
+assert np.allclose(pfi, true_pfi, rtol=0.02, atol=0.002)
 
 # assert that args_saved.py is same as args.py
 with open('args.py', 'r') as f:

diff --git a/tests/test_lda_nn_fe_multiclass_spatiotemporal.py b/tests/test_lda_nn_fe_multiclass_spatiotemporal.py
@@ -15,7 +15,7 @@
 
 # assert that pfi is the same as true_pfi to 4 decimal places
 # use numpy
-assert np.allclose(pfi, true_pfi, rtol=1e-05, atol=1e-08)
+assert np.allclose(pfi, true_pfi, rtol=0.02, atol=0.002)
 
 # assert that args_saved.py is same as args.py
 with open('args.py', 'r') as f:

diff --git a/tests/test_lda_nn_fe_multiclass_spectralPFI.py b/tests/test_lda_nn_fe_multiclass_spectralPFI.py
@@ -14,7 +14,7 @@
 
 # assert that pfi is the same as true_pfi to 4 decimal places
 # use numpy
-assert np.allclose(pfi, true_pfi, rtol=1e-05, atol=1e-08)
+assert np.allclose(pfi, true_pfi, rtol=0.02, atol=0.002)
 
 # assert that args_saved.py is same as args.py
 with open('args.py', 'r') as f:

diff --git a/tests/test_lda_nn_fe_multiclass_temporalPFI.py b/tests/test_lda_nn_fe_multiclass_temporalPFI.py
@@ -14,7 +14,7 @@
 
 # assert that pfi is the same as true_pfi to 4 decimal places
 # use numpy
-assert np.allclose(pfi, true_pfi, rtol=1e-05, atol=1e-08)
+assert np.allclose(pfi, true_pfi, rtol=0.02, atol=0.002)
 
 
 # assert that args_saved.py is same as args.py

diff --git a/tests/test_lda_nn_fe_multiclass_temporospectralPFI.py b/tests/test_lda_nn_fe_multiclass_temporospectralPFI.py
@@ -15,7 +15,7 @@
 
 # assert that pfi is the same as true_pfi to 4 decimal places
 # use numpy
-assert np.allclose(pfi, true_pfi, rtol=1e-05, atol=1e-08)
+assert np.allclose(pfi, true_pfi, rtol=0.02, atol=0.002)
 
 # assert that args_saved.py is same as args.py
 with open('args.py', 'r') as f:

diff --git a/tests/test_lda_nn_sw_multiclass.py b/tests/test_lda_nn_sw_multiclass.py
@@ -21,7 +21,7 @@
 
     # assert that losses are the same as true_losses to 4 decimal places
     for loss, val in zip(losses, true_losses):
-        assert round(loss, 2) == round(val, 2)
+        assert abs(loss-val) < 0.005
 
 # check that model.pt5 ... model.pt105 exists
 for i in range(5, 105):

diff --git a/tests/test_nn.py b/tests/test_nn.py
@@ -17,7 +17,7 @@
 
     # assert that loss is the same as val to 4 decimal places
     for k in keys:
-        assert round(float(d[k]), 2) == round(float(values[k]), 2)
+        assert abs(float(d[k])-float(values[k])) < 0.005
 
 # check that model.pt exists
 for n in ['_end', '_epoch', '_init', '']: