add test for annotation with text and arrow

itziakos · tacaswell · commit be652cd8ea00 · 2015-01-31T14:01:42.000-05:00
diff --git a/lib/matplotlib/tests/test_text.py b/lib/matplotlib/tests/test_text.py
@@ -298,6 +298,37 @@ def test_text_annotation_get_window_extent():
     eq_(points[0, 1], -below_line)
     eq_(points[1, 1], text_bbox.height - below_line)
 
+
+def test_text_with_arrow_annotation_get_window_extent():
+    from matplotlib.figure import Figure
+    from matplotlib.text import Annotation, Text
+    from matplotlib.backends.backend_agg import RendererAgg
+
+    figure = Figure(dpi=100)
+    renderer = RendererAgg(600, 600, 100)
+    headwidth = 21
+
+    text = Text(text='test', x=0, y=0)
+    text.set_figure(figure)
+    text_bbox = text.get_window_extent(renderer=renderer)
+
+    # Text annotation with arrow
+    annotation = Annotation(
+        'test',
+        xy=(0.0, 50.0 + (headwidth / 0.72) * 0.5),
+        xytext=(50.0, 50.0), xycoords='figure pixels',
+        arrowprops={
+            'facecolor': 'black', 'width': 2,
+            'headwidth': headwidth, 'shrink': 0.0})
+    annotation.set_figure(figure)
+    annotation.draw(renderer)
+
+    bbox = annotation.get_window_extent(renderer=renderer)
+    eq_(bbox.width, text_bbox.width + 50.0)
+    expected_height = max(text_bbox.height, headwidth / 0.72)
+    assert_almost_equal(bbox.height, expected_height)
+
+
 def test_arrow_annotation_get_window_extent():
     from matplotlib.figure import Figure
     from matplotlib.text import Annotation
