Labellers testing module

CHANGELOG.md

@@ -14,6 +14,7 @@
 * Update default Bokeh markers for rcparams ([2104](https://github.com/arviz-devs/arviz/pull/2104))
 * Correctly (re)order dimensions for `bfmi` and `plot_energy` ([2126](https://github.com/arviz-devs/arviz/pull/2126))
 * Fix bug with the dimension order dependency ([2103](https://github.com/arviz-devs/arviz/pull/2103))
+* Add testing module for labeller classes ([2095](https://github.com/arviz-devs/arviz/pull/2095))
 
 ### Deprecation
 * Removed `fill_last`, `contour` and `plot_kwargs` arguments from `plot_pair` function ([2085](https://github.com/arviz-devs/arviz/pull/2085))

arviz/tests/base_tests/test_labels.py

@@ -0,0 +1,68 @@
+"""Tests for labeller classes."""
+import pytest
+
+from ...labels import (
+    BaseLabeller,
+    DimCoordLabeller,
+    DimIdxLabeller,
+    IdxLabeller,
+    MapLabeller,
+    NoModelLabeller,
+    NoVarLabeller,
+)
+
+
+class Data:
+    def __init__(self):
+        self.sel = {
+            "instrument": "a",
+            "experiment": 3,
+        }
+        self.isel = {
+            "instrument": 0,
+            "experiment": 4,
+        }
+
+
+@pytest.fixture
+def multidim_sels():
+    return Data()
+
+
+class Labellers:
+    def __init__(self):
+        self.labellers = {
+            "BaseLabeller": BaseLabeller(),
+            "DimCoordLabeller": DimCoordLabeller(),
+            "IdxLabeller": IdxLabeller(),
+            "DimIdxLabeller": DimIdxLabeller(),
+            "MapLabeller": MapLabeller(),
+            "NoVarLabeller": NoVarLabeller(),
+            "NoModelLabeller": NoModelLabeller(),
+        }
+
+
+@pytest.fixture
+def labellers():
+    return Labellers()
+
+
+@pytest.mark.parametrize(
+    "args",
+    [
+        ("BaseLabeller", "theta\na, 3"),
+        ("DimCoordLabeller", "theta\ninstrument: a, experiment: 3"),
+        ("IdxLabeller", "theta\n0, 4"),
+        ("DimIdxLabeller", "theta\ninstrument#0, experiment#4"),
+        ("MapLabeller", "theta\na, 3"),
+        ("NoVarLabeller", "a, 3"),
+        ("NoModelLabeller", "theta\na, 3"),
+    ],
+)
+class TestLabellers:
+    # pylint: disable=redefined-outer-name
+    def test_make_label_vert(self, args, multidim_sels, labellers):
+        name, expected_label = args
+        labeller_arg = labellers.labellers[name]
+        label = labeller_arg.make_label_vert("theta", multidim_sels.sel, multidim_sels.isel)
+        assert label == expected_label