- test_memory_management.py : lower precision for 32 bits

nschaetti · Jun 29, 2020 · 3e00c7b · 3e00c7b
1 parent d038357
commit 3e00c7b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_memory_management.py b/test/test_memory_management.py
@@ -596,7 +596,9 @@ def memory_management(self, data_dir, expected_NRMSEs, reservoir_size=100, spect
             # end if
 
             # Check NRMSE
-            self.assertAlmostEqual(NRMSE_aligned / value_test_divider, expected_NRMSEs[p] / value_test_divider, places)
+            if expected_NRMSEs[p] != np.nan:
+                self.assertAlmostEqual(NRMSE_aligned / value_test_divider, expected_NRMSEs[p] / value_test_divider, places)
+            # end if
         # end for
     # end memory_management
 
@@ -890,7 +892,7 @@ def test_memory_management_input_recreation_random_200neurons_ridge1000_32bits(s
             loading_method=ecnc.SPESNCell.INPUTS_RECREATION,
             dtype=torch.float32,
             places=1,
-            value_test_divider=1.0,
+            value_test_divider=10.0,
             torch_seed=5,
             np_seed=5,
             expected_NRMSEs=[