Add text wrapping tests

dstansby · dstansby · commit 3be4aeaf3086 · 2019-08-10T21:29:30.000+01:00
diff --git a/lib/matplotlib/tests/test_text.py b/lib/matplotlib/tests/test_text.py
@@ -592,3 +592,17 @@ def test_large_subscript_title():
     tt.set_position((x, 1.01))
     ax.set_title('Old Way', loc='left')
     ax.set_xticklabels('')
+
+
+def test_long_word_wrap():
+    fig = plt.figure(figsize=(6, 4))
+    text = fig.text(9.5, 8, 'Alonglineoftexttowrap', wrap=True)
+    fig.canvas.draw()
+    assert text._get_wrapped_text() == 'Alonglineoftexttowrap'
+
+
+def test_wrap_no_wrap():
+    fig = plt.figure(figsize=(6, 4))
+    text = fig.text(0, 0, 'non wrapped text', wrap=True)
+    fig.canvas.draw()
+    assert text._get_wrapped_text() == 'non wrapped text'
