TST: Update test_as_mpl_selector to accept rotated regions

Author: dhomeier

CHANGES.rst

@@ -9,6 +9,9 @@ New Features
 
 - Added the DS9 'boxcircle' point symbol. [#387]
 
+- Support rotation of the ``as_mpl_selector`` widgets for rectangular
+  and ellipse regions. [#390]
+
 Bug Fixes
 ---------
 

regions/shapes/tests/test_ellipse.py

@@ -115,12 +115,17 @@ def update_mask(reg):
         # works with rotated ellipses, the following exception check can
         # be removed as well as the ``angle=0 * u.deg`` in the call to
         # copy() below.
-        with pytest.raises(NotImplementedError,
-                           match=('Cannot create matplotlib selector for '
-                                  'rotated ellipse.')):
-            self.reg.as_mpl_selector(ax)
+        import matplotlib.widgets
+        if not hasattr(matplotlib.widgets.EllipseSelector, '_rotation'):
+            with pytest.raises(NotImplementedError,
+                               match=('Cannot create matplotlib selector for rotated ellipse.')):
+                self.reg.as_mpl_selector(ax)
 
-        region = self.reg.copy(angle=0 * u.deg)
+            angle = 0 * u.deg
+        else:
+            angle = self.reg.angle
+
+        region = self.reg.copy(angle=angle)
 
         selector = region.as_mpl_selector(ax, callback=update_mask, sync=sync)  # noqa
 
@@ -162,7 +167,7 @@ def update_mask(reg):
             assert_allclose(region.center.y, 4)
             assert_allclose(region.width, 4)
             assert_allclose(region.height, 3)
-            assert_quantity_allclose(region.angle, 0 * u.deg)
+            assert_quantity_allclose(region.angle, angle)
 
             assert_equal(mask, 0)
 

regions/shapes/tests/test_rectangle.py

@@ -118,12 +118,17 @@ def update_mask(reg):
         # this works with rotated rectangles, the following exception
         # check can be removed as well as the ``angle=0 * u.deg`` in the
         # call to copy() below.
-        with pytest.raises(NotImplementedError,
-                           match=('Cannot create matplotlib selector for '
-                                  'rotated rectangle.')):
-            self.reg.as_mpl_selector(ax)
+        import matplotlib.widgets
+        if not hasattr(matplotlib.widgets.RectangleSelector, '_rotation'):
+            with pytest.raises(NotImplementedError,
+                               match=('Cannot create matplotlib selector for rotated rectangle.')):
+                self.reg.as_mpl_selector(ax)
 
-        region = self.reg.copy(angle=0 * u.deg)
+            angle = 0 * u.deg
+        else:
+            angle = self.reg.angle
+
+        region = self.reg.copy(angle=angle)
 
         selector = region.as_mpl_selector(ax, callback=update_mask, sync=sync)  # noqa
 
@@ -162,7 +167,7 @@ def update_mask(reg):
             assert_allclose(region.center.y, 4)
             assert_allclose(region.width, 4)
             assert_allclose(region.height, 3)
-            assert_quantity_allclose(region.angle, 0 * u.deg)
+            assert_quantity_allclose(region.angle, angle)
 
             assert_equal(mask, 0)