#709 flake8

pybamm-team · Jan 31, 2020 · 69c2aa2 · 69c2aa2
1 parent ce574ed
commit 69c2aa2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/pybamm/parameters/parameter_values.py b/pybamm/parameters/parameter_values.py
@@ -388,7 +388,7 @@ def process_model(self, unprocessed_model, inplace=True):
     def update_model(self, model, disc):
         raise NotImplementedError(
             """
-            update_model functionality has been deprecated. 
+            update_model functionality has been deprecated.
             Use pybamm.InputParameter to quickly change a parameter value instead
             """
         )

diff --git a/tests/unit/test_parameters/test_parameter_values.py b/tests/unit/test_parameters/test_parameter_values.py
@@ -502,6 +502,11 @@ def test_process_model(self):
         with self.assertRaises(KeyError):
             parameter_values.process_model(model)
 
+    def test_update_model(self):
+        param = pybamm.ParameterValues({})
+        with self.assertRaises(NotImplementedError):
+            param.update_model(None, None)
+
     def test_inplace(self):
         model = pybamm.lithium_ion.SPM()
         param = model.default_parameter_values