Add test case for history plot

optimagic-dev · Nov 13, 2024 · 63a4e93 · 63a4e93
1 parent f1276f2
commit 63a4e93
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/optimagic/visualization/test_history_plots.py b/tests/optimagic/visualization/test_history_plots.py
@@ -157,8 +157,10 @@ def test_harmonize_inputs_to_dict_list_results():
 
 def test_harmonize_inputs_to_dict_dict_input():
     res = minimize(fun=lambda x: x @ x, params=np.arange(5), algorithm="scipy_lbfgsb")
-    results = {"bla": res, "blub": res}
-    assert _harmonize_inputs_to_dict(results=results, names=None) == results
+    results = {"bla": res, om.algos.scipy_lbfgsb(): res, om.algos.scipy_neldermead: res}
+    got = _harmonize_inputs_to_dict(results=results, names=None)
+    expected = {"bla": res, "scipy_lbfgsb": res, "scipy_neldermead": res}
+    assert got == expected
 
 
 def test_harmonize_inputs_to_dict_dict_input_with_names():