see if this patch fixes and if so then move this to a follow up PR

tests/contrib/test_viz.py

@@ -1,5 +1,6 @@
 import json
 import sys
+from unittest.mock import patch
 
 import matplotlib
 import matplotlib.pyplot as plt
@@ -181,3 +182,25 @@ def test_plot_results_components_data_structure(datadir):
         brazil.plot_results(
             data["testmus"], data["results"], test_size=0.05, ax=ax, components=True
         )
+
+
+def test_plot_results_wrong_axis_labels(datadir):
+    """
+    If the returned labels are different from the expected, then the hardcoded
+    values in label_part will be wrong, causing `next` to fail on the iterator
+    for label_idx.
+    """
+    data = json.load(datadir.joinpath("hypotest_results.json").open(encoding="utf-8"))
+
+    fig = Figure()
+    ax = fig.subplots()
+
+    with patch(
+        "matplotlib.axes._axes.Axes.get_legend_handles_labels"
+    ) as mock_get_legend_handles_labels:
+        mock_get_legend_handles_labels.return_value = None, ["fail"]
+
+        with pytest.raises(StopIteration):
+            brazil.plot_results(data["testmus"], data["results"], test_size=0.05, ax=ax)
+
+        mock_get_legend_handles_labels.assert_called()