Added further checks and internal corrections

handley-lab · Jul 26, 2023 · d16a80c · d16a80c
1 parent e9dce13
commit d16a80c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/anesthetic/plot.py b/anesthetic/plot.py
@@ -613,7 +613,9 @@ def make_1d_axes(params, ncol=None, labels=None,
             "make_1d_axes(..., labels=tex)  # anesthetic 2.0"
             )
     fig = fig_kw.pop('fig') if 'fig' in fig_kw else plt.figure(**fig_kw)
-    axes = AxesSeries(index=np.atleast_1d(params),
+    if np.array(params).ndim == 0:
+        params = [params]
+    axes = AxesSeries(index=params,
                       fig=fig,
                       ncol=ncol,
                       labels=labels,

diff --git a/tests/test_plot.py b/tests/test_plot.py
@@ -122,14 +122,22 @@ def test_make_1d_axes():
 
 @pytest.mark.parametrize('params', [[0, 1, 2, 3],
                                     [0, 2, 3],
-                                    [1, 2, 3]])
+                                    [1, 2, 3],
+                                    ['x0', 'x1', 'x2'],
+                                    ['1', '2', '3'],
+                                    [0, 'x1', 'x2'],
+                                    ])
 def test_make_Nd_axes_integers(params):
     fig, axes = make_1d_axes(params)
     assert isinstance(fig, Figure)
     assert isinstance(axes, AxesSeries)
+    assert list(axes.index) == params
+
     fig, axes = make_2d_axes(params)
     assert isinstance(fig, Figure)
     assert isinstance(axes, AxesDataFrame)
+    assert list(axes.index) == params
+    assert list(axes.columns) == params
 
 
 def test_make_2d_axes_inputs_outputs():