File tree Expand file tree Collapse file tree 1 file changed +25
-0
lines changed
tests/unit/test_parameters Expand file tree Collapse file tree 1 file changed +25
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments