Declare anytree onn top to pass test_is_constant_and_can_evaluate

pybamm-team · Nov 9, 2023 · 700ab5a · 700ab5a
1 parent 8d6db99
commit 700ab5a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/unit/test_util.py b/tests/unit/test_util.py
@@ -14,6 +14,8 @@
 
 def test_function(arg):
     return arg + arg
+
+anytree = sys.modules['anytree']
 class TestUtil(TestCase):
     """
     Test the functionality in util.py
@@ -31,6 +33,7 @@ def test_rmse(self):
             pybamm.rmse(np.ones(5), np.zeros(3))
 
     def test_is_constant_and_can_evaluate(self):
+        sys.modules['anytree'] = anytree
         symbol = pybamm.PrimaryBroadcast(0, "negative electrode")
         self.assertEqual(False, pybamm.is_constant_and_can_evaluate(symbol))
         symbol = pybamm.StateVector(slice(0, 1))