fixed bug with label of Arrow2DSeries/Arrow3DSeries

Author: Davide-sd

doc/source/changelog.rst

@@ -2,6 +2,13 @@
  Changelog
 ==========
 
+v4.0.3
+======
+
+* Fixed bug with label of ``tick_formatter_multiples_of``.
+* Fixed bug with label of ``Arrow2DSeries/Arrow3DSeries``.
+
+
 v4.0.2
 ======
 

spb/series/vectors.py

@@ -626,7 +626,13 @@ def get_label(self, use_latex=False, wrapper="$%s$"):
         """
         if use_latex is False:
             return self._label_str
-        return self._get_wrapped_label(self._label_latex, wrapper)
+
+        # if the default label is shown, then it'll start with \left(...
+        label = self._label_latex
+        if label[0] == "\\":
+            return self._get_wrapped_label(self._label_latex, wrapper)
+        # if the user provided the label
+        return label
 
     def get_data(self):
         """

tests/backends/test_matplotlib.py

@@ -2079,7 +2079,7 @@ def test_arrow_2d():
     assert isinstance(ax, Axes)
     assert len(ax.patches) == 1
     assert len(ax.get_legend().legend_handles) == 1
-    assert ax.get_legend().legend_handles[0].get_label() == "$test$"
+    assert ax.get_legend().legend_handles[0].get_label() == "test"
     assert ax.get_legend().legend_handles[0].get_color() == "r"
     p.backend.update_interactive({a: 4, b: 5})
     p.backend.close()
@@ -2145,7 +2145,7 @@ def test_arrow_3d():
     assert isinstance(ax, mpl_toolkits.mplot3d.axes3d.Axes3D)
     assert len(ax.patches) == 1
     assert len(ax.get_legend().legend_handles) == 1
-    assert ax.get_legend().legend_handles[0].get_label() == "$test$"
+    assert ax.get_legend().legend_handles[0].get_label() == "test"
     assert ax.get_legend().legend_handles[0].get_color() == "r"
     # only way to test if it renders what it's supposed to
     assert np.allclose(ax.patches[0]._xyz, [1, 2, 3])

tests/test_series.py

@@ -2588,6 +2588,20 @@ def test_series_labels():
     assert s2.get_label(False) == "test"
     assert s2.get_label(True) == "test"
 
+    s1 = Arrow2DSeries((0, 1), (2, 3))
+    assert s1.get_label(False) == "(0.0, 1.0) -> (2.0, 4.0)"
+    assert s1.get_label(True) == r"$\left(0.0, 1.0\right) \rightarrow \left(2.0, 4.0\right)$"
+    s2 = Arrow2DSeries((0, 1), (2, 3), "test")
+    assert s2.get_label(False) == "test"
+    assert s2.get_label(True) == "test"
+
+    s1 = Arrow3DSeries((0, 1, 2), (3, 4, 5))
+    assert s1.get_label(False) == "(0.0, 1.0, 2.0) -> (3.0, 5.0, 7.0)"
+    assert s1.get_label(True) == r"$\left(0.0, 1.0, 2.0\right) \rightarrow \left(3.0, 5.0, 7.0\right)$"
+    s2 = Arrow3DSeries((0, 1, 2), (3, 4, 5), "test")
+    assert s2.get_label(False) == "test"
+    assert s2.get_label(True) == "test"
+
 
 @pytest.mark.parametrize(
     "use_cm", [None, True, False]