Fixing a few typos

pybamm-team · Sep 28, 2023 · 10c7270 · 10c7270
1 parent 3a75c16
commit 10c7270
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/unit/test_expression_tree/test_symbol.py b/tests/unit/test_expression_tree/test_symbol.py
@@ -387,7 +387,7 @@ def test_symbol_repr(self):
         )
 
     def test_symbol_visualise(self):
-        with TemporaryDirectory as dir_name:
+        with TemporaryDirectory() as dir_name:
             test_stub = os.path.join(dir_name, "test_visualize")
             test_name = f"{test_stub}.png"
             c = pybamm.Variable("c", "negative electrode")

diff --git a/tests/unit/test_parameters/test_lithium_ion_parameters.py b/tests/unit/test_parameters/test_lithium_ion_parameters.py
@@ -11,7 +11,7 @@
 
 class TestLithiumIonParameterValues(TestCase):
     def test_print_parameters(self):
-        with TemporaryDirectory as dir_name:
+        with TemporaryDirectory() as dir_name:
             parameters = pybamm.LithiumIonParameters()
             parameter_values = pybamm.lithium_ion.BaseModel().default_parameter_values
             output_file = os.path.join(dir_name, "lithium_ion_parameters.txt")