cleanup test

astromancer · astromancer · commit 9f12372b3982 · 2021-03-03T18:52:32.000+02:00
diff --git a/lib/matplotlib/tests/test_patches.py b/lib/matplotlib/tests/test_patches.py
@@ -5,7 +5,8 @@
 from numpy.testing import assert_almost_equal, assert_array_equal
 import pytest
 
-from matplotlib.patches import Patch, Polygon, Rectangle, FancyArrowPatch
+from matplotlib.patches import (Annulus, Patch, Polygon, Rectangle, 
+                                FancyArrowPatch)
 from matplotlib.testing.decorators import image_comparison, check_figures_equal
 from matplotlib.transforms import Bbox
 import matplotlib.pyplot as plt
@@ -582,6 +583,18 @@ def test_rotated_arcs():
         ax.set_aspect("equal")
 
 
+@image_comparison(baseline_images=['annulus'], extensions=['png'])
+def test_annulus():
+
+    fig, ax = plt.subplots()
+    cir = Annulus((0.5, 0.5), 0.2, 0.05, fc='g')        # circular annulus
+    ell = Annulus((0.5, 0.5), (0.5, 0.3), 0.1, 45,      # elliptical
+                  fc='m', ec='b', alpha=0.5, hatch='xxx')
+    ax.add_patch(cir)
+    ax.add_patch(ell)
+    ax.set_aspect('equal')
+
+
 def test_degenerate_polygon():
     point = [0, 0]
     correct_extents = Bbox([point, point]).extents
@@ -628,14 +641,4 @@ def test_default_joinstyle():
     assert patch.get_joinstyle() == 'miter'
 
 
-@image_comparison(baseline_images=['annulus'], extensions=['png'])
-def test_annulus():
-    from matplotlib.patches import Annulus
 
-    fig, ax = plt.subplots()
-    cir = Annulus((0.5, 0.5), 0.2, 0.05, fc='g')        # circular annulus
-    ell = Annulus((0.5, 0.5), (0.5, 0.3), 0.1, 45,      # elliptical
-                  fc='m', ec='b', alpha=0.5, hatch='xxx')
-    ax.add_patch(cir)
-    ax.add_patch(ell)
-    ax.set_aspect('equal')
