Fixed bug: mathtext rendered width not being calculated correctly (#25692)

lganic · timhoffm · web-flow · commit e31d8ed80cb3 · 2023-04-15T14:31:38.000-07:00
* Text now accounts for mathtext width when wrapping

* Added test case for mathtext wrapping

* Update lib/matplotlib/text.py

Co-authored-by: Tim Hoffmann &lt;2836374+timhoffm@users.noreply.github.com&gt;

* removed unnecessary variable

---------

Co-authored-by: Tim Hoffmann &lt;2836374+timhoffm@users.noreply.github.com&gt;
diff --git a/lib/matplotlib/tests/test_text.py b/lib/matplotlib/tests/test_text.py
@@ -701,6 +701,15 @@ def test_wrap():
                                         'times.')
 
 
+def test_mathwrap():
+    fig = plt.figure(figsize=(6, 4))
+    s = r'This is a very $\overline{\mathrm{long}}$ line of Mathtext.'
+    text = fig.text(0, 0.5, s, size=40, wrap=True)
+    fig.canvas.draw()
+    assert text._get_wrapped_text() == ('This is a very $\\overline{\\mathrm{long}}$\n'
+                                        'line of Mathtext.')
+
+
 def test_get_window_extent_wrapped():
     # Test that a long title that wraps to two lines has the same vertical
     # extent as an explicit two line title.
diff --git a/lib/matplotlib/text.py b/lib/matplotlib/text.py
@@ -642,10 +642,11 @@ def _get_rendered_text_width(self, text):
         """
         Return the width of a given text string, in pixels.
         """
+
         w, h, d = self._renderer.get_text_width_height_descent(
             text,
             self.get_fontproperties(),
-            False)
+            cbook.is_math_text(text))
         return math.ceil(w)
 
     def _get_wrapped_text(self):
