Merge pull request #1542 from priyanshuone6/issue-1540

Skip print_name setter
pybamm-team · Jul 9, 2021 · d62c862 · d62c862
2 parents c421a91 + 66fe1bc
commit d62c862
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/pybamm/expression_tree/printing/print_name.py b/pybamm/expression_tree/printing/print_name.py
@@ -41,6 +41,10 @@
 def prettify_print_name(name):
     """Prettify print_name using regex"""
 
+    # Skip prettify_print_name() for cases like `new_copy()`
+    if "{" in name:
+        return name
+
     # Return print_name if name exists in the dictionary
     if name in PRINT_NAME_OVERRIDES:
         return PRINT_NAME_OVERRIDES[name]

diff --git a/tests/unit/test_expression_tree/test_printing/test_print_name.py b/tests/unit/test_expression_tree/test_printing/test_print_name.py
@@ -29,6 +29,12 @@ def test_prettify_print_name(self):
         # Test greek letters
         self.assertEqual(param1.delta_phi_n.print_name, r"\delta_\phi_n")
 
+        # Test new_copy()
+        param2 = pybamm.LeadAcidParameters()
+        x_n = pybamm.standard_spatial_vars.x_n
+        a_n = param2.a_n(x_n)
+        a_n.new_copy()
+
 
 if __name__ == "__main__":
     print("Add -v for more debug output")