fixed Polygon to pass a test.

NOAA-ORR-ERD · Jan 20, 2018 · 0c735a5 · 0c735a5
1 parent 071591d
commit 0c735a5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/py_gnome/gnome/utilities/geometry/polygons.py b/py_gnome/gnome/utilities/geometry/polygons.py
@@ -75,9 +75,11 @@ def __array_wrap__(self, out_arr, context=None):
 
     def __getitem__(self, index):
         """
-        Override __getitem__ to return a simple (2, ) ndarray, rather than a
+        Override __getitem__ to return a ndarray, rather than a
         Polygon object
         """
+        if type(index) is slice:
+            return Polygon(np.ndarray.__getitem__(self, index), self.metadata)
         return np.asarray(np.ndarray.__getitem__(self, index))
 
     def __eq__(self, other):

diff --git a/py_gnome/tests/unit_tests/test_utilities/test_geometry/test_polygons.py b/py_gnome/tests/unit_tests/test_utilities/test_geometry/test_polygons.py
@@ -60,6 +60,7 @@ def test_index2(self):
     def test_slice(self):
         P = Polygon([(1, 2), (3, 4), (5, 6)])
         print P[:2]
+        print Polygon([(1, 2), (3, 4)])[:]
         assert P[:2] == Polygon([(1, 2), (3, 4)])
 
     def test_metadata(self):