Added tests. Added note on 'how_to_slice_plot' about backends.

Author: r3kste

docs/source/how_to/how_to_slice_plot.ipynb

@@ -83,6 +83,17 @@
     "fig.show()"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    ":::{note}\n",
+    "\n",
+    "For details on using other plotting backends, see [How to change the plotting backend](how_to_change_plotting_backend.ipynb).\n",
+    "\n",
+    ":::"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},

tests/optimagic/visualization/test_slice_plot.py

@@ -3,7 +3,13 @@
 
 from optimagic import mark
 from optimagic.parameters.bounds import Bounds
-from optimagic.visualization.slice_plot import slice_plot
+from optimagic.visualization.plotting_utilities import LineData, MarkerData
+from optimagic.visualization.slice_plot import (
+    _extract_slice_plot_lines_and_labels,
+    _get_plot_data,
+    _get_processed_func_and_func_eval,
+    slice_plot,
+)
 
 
 @pytest.fixture()
@@ -40,6 +46,7 @@ def sphere(params):
     {"share_x": True},
     {"share_y": False},
     {"return_dict": True},
+    {"backend": "matplotlib"},
 ]
 parametrization = [
     (func, kwargs) for func in [sphere_loglike, sphere] for kwargs in KWARGS
@@ -53,3 +60,49 @@ def test_slice_plot(fixed_inputs, func, kwargs):
         **fixed_inputs,
         **kwargs,
     )
+
+
+def test_extract_slice_plot_lines(fixed_inputs):
+    params, bounds = fixed_inputs["params"], fixed_inputs["bounds"]
+
+    func, func_eval = _get_processed_func_and_func_eval(
+        sphere, func_kwargs=None, params=params
+    )
+
+    plot_data, internal_params = _get_plot_data(
+        func=func,
+        params=params,
+        bounds=bounds,
+        func_eval=func_eval,
+        selector=None,
+        n_gridpoints=10,
+        batch_evaluator="joblib",
+        n_cores=1,
+    )
+
+    lines_list, marker_list, xlabels, ylabels = _extract_slice_plot_lines_and_labels(
+        plot_data=plot_data,
+        internal_params=internal_params,
+        func_eval=func_eval,
+        param_names={"alpha": "Alpha"},
+        color=None,
+    )
+
+    assert isinstance(lines_list, list) and len(lines_list) == len(params)
+    assert all(
+        isinstance(subplot_lines, list)
+        and len(subplot_lines) == 1
+        and isinstance(subplot_lines[0], LineData)
+        for subplot_lines in lines_list
+    )
+
+    assert isinstance(marker_list, list) and len(marker_list) == len(params)
+    assert all(isinstance(marker, MarkerData) for marker in marker_list)
+    for i, k in enumerate(params):
+        assert marker_list[i].x == params[k]
+
+    assert isinstance(xlabels, list)
+    assert xlabels == ["Alpha", "beta", "gamma", "delta"]
+
+    assert isinstance(ylabels, list)
+    assert all(ylabel == "Function Value" for ylabel in ylabels)