fixing numerical round off errors in test

bussilab · Nov 28, 2024 · aae8a82 · aae8a82
1 parent fa4ec90
commit aae8a82
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/test_loss_and_minimizer.py b/tests/test_loss_and_minimizer.py
@@ -12,8 +12,10 @@ def assertEqualObjs(self, obj1, obj2):
 
         if isinstance(obj1, np.ndarray) or isinstance(obj1, jnp.ndarray):
             if obj1.shape == (1,): obj1 = obj1[0]
+            elif obj1.shape == (): obj1 = np.array([obj1])[0]
         if isinstance(obj2, np.ndarray) or isinstance(obj2, jnp.ndarray):
             if obj2.shape == (1,): obj2 = obj2[0]
+            elif obj2.shape == (): obj2 = np.array([obj2])[0]
 
         if isinstance(obj1, dict) and isinstance(obj2, dict):
             self.assertSetEqual(set(obj1.keys()), set(obj2.keys()))

diff --git a/tests/test_mdrefinement.py b/tests/test_mdrefinement.py
@@ -11,8 +11,10 @@ def assertEqualObjs(self, obj1, obj2, tol = 1e-4, if_relative = False):
 
         if isinstance(obj1, np.ndarray) or isinstance(obj1, jnp.ndarray):
             if obj1.shape == (1,): obj1 = obj1[0]
+            elif obj1.shape == (): obj1 = np.array([obj1])[0]
         if isinstance(obj2, np.ndarray) or isinstance(obj2, jnp.ndarray):
             if obj2.shape == (1,): obj2 = obj2[0]
+            elif obj2.shape == (): obj2 = np.array([obj2])[0]
 
         if isinstance(obj1, dict) and isinstance(obj2, dict):
             self.assertSetEqual(set(obj1.keys()), set(obj2.keys()))