Skip to content

Commit 6084d37

Browse files
agriyakhetarpalBradyPlanden
authored andcommitted
Add a test
1 parent f2c86a4 commit 6084d37

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

tests/unit/test_parameters/test_parameter_values.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1298,6 +1298,31 @@ def test_to_json_with_filename(self):
12981298
finally:
12991299
os.remove(temp_path)
13001300

1301+
def test_roundtrip_with_keyword_args(self):
1302+
def func_no_kwargs(x):
1303+
return 2 * x
1304+
1305+
def func_with_kwargs(x, y=1):
1306+
return 2 * x
1307+
1308+
x = pybamm.Scalar(2)
1309+
func_param = pybamm.FunctionParameter("func", {"x": x})
1310+
1311+
parameter_values = pybamm.ParameterValues({"func": func_no_kwargs})
1312+
assert parameter_values.evaluate(func_param) == 4.0
1313+
1314+
serialized = parameter_values.to_json()
1315+
parameter_values_loaded = pybamm.ParameterValues.from_json(serialized)
1316+
assert parameter_values_loaded.evaluate(func_param) == 4.0
1317+
1318+
parameter_values = pybamm.ParameterValues({"func": func_with_kwargs})
1319+
assert parameter_values.evaluate(func_param) == 4.0
1320+
1321+
serialized = parameter_values.to_json()
1322+
parameter_values_loaded = pybamm.ParameterValues.from_json(serialized)
1323+
1324+
assert parameter_values_loaded.evaluate(func_param) == 4.0
1325+
13011326
def test_convert_symbols_in_dict_with_interpolator(self):
13021327
"""Test convert_symbols_in_dict with interpolator (covers lines 1154-1170)."""
13031328
import numpy as np

0 commit comments

Comments
 (0)