Deal properly with scalar PixCoords

astropy · Nov 30, 2016 · bf3f349 · bf3f349
1 parent 8b232cf
commit bf3f349
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/regions/shapes/polygon.py b/regions/shapes/polygon.py
@@ -40,8 +40,13 @@ def area(self):
         raise NotImplementedError
 
     def contains(self, pixcoord):
-        return points_in_polygon(pixcoord.x.astype(float),
-                                 pixcoord.y.astype(float),
+        if pixcoord.isscalar:
+            x = np.array([pixcoord.x], dtype=float)
+            y = np.array([pixcoord.y], dtype=float)
+        else:
+            x = pixcoord.x.astype(float)
+            y = pixcoord.y.astype(float)
+        return points_in_polygon(x, y,
                                  self.vertices.x.astype(float),
                                  self.vertices.y.astype(float)).astype(bool)