renamed assertEqual to assert_equal

baseclasses/testing/assertions.py

@@ -3,7 +3,7 @@
 DEFAULT_TOL = 1e-12
 
 
-def assertEqual(a, b):
+def assert_equal(a, b):
     """
     This replaces the assertEqual functionality provided by unittest.TestCase
     so that we can call this outside the class
@@ -17,7 +17,7 @@ def assertEqual(a, b):
         if set(a.keys()) != set(b.keys()):
             raise AssertionError("The two dictionaries do not have the same keys")
         for k in a.keys():
-            assertEqual(a[k], b[k])
+            assert_equal(a[k], b[k])
     elif not a == b:
         raise AssertionError(f"{a} and {b} are not equal.")
 
@@ -63,7 +63,7 @@ def assert_dict_not_allclose(actual, desired, atol=DEFAULT_TOL, rtol=DEFAULT_TOL
     """
     The opposite of assert_dict_allclose
     """
-    assertEqual(set(actual.keys()), set(desired.keys()))
+    assert_equal(set(actual.keys()), set(desired.keys()))
     for key in actual.keys():
         if np.allclose(actual[key], desired[key], atol=atol, rtol=rtol):
             raise AssertionError(f"Dictionaries are close! Got {actual} and {desired} for key {key}")

tests/test_fileIO.py

@@ -2,7 +2,7 @@
 import os
 from parameterized import parameterized
 from baseclasses.testing.decorators import require_mpi
-from baseclasses.testing.assertions import assertEqual
+from baseclasses.testing.assertions import assert_equal
 from baseclasses.utils import readPickle, writePickle, readJSON, writeJSON
 import numpy as np
 
@@ -35,7 +35,7 @@ def test_pickle(self, name, obj):
         self.fileName = f"{name}.pkl"
         writePickle(self.fileName, obj, comm=self.comm)
         newObj = readPickle(self.fileName, comm=self.comm)
-        assertEqual(obj, newObj)
+        assert_equal(obj, newObj)
 
     @parameterized.expand(
         [
@@ -48,7 +48,7 @@ def test_JSON(self, name, obj):
         self.fileName = f"{name}.json"
         writeJSON(self.fileName, obj, comm=self.comm)
         newObj = readJSON(self.fileName, comm=self.comm)
-        assertEqual(obj, newObj)
+        assert_equal(obj, newObj)
 
     def tearDown(self):
         if self.comm.rank == 0: