fix #14 by excepting also IndexError

CHANGES.md

@@ -1,6 +1,7 @@
 # v0.x #
 - added support for saving more subsets and loading a certain one in/from a netcdf grid file
 - fix #15 by setting correct shape for derived cell grids
+- fix issue #14 of gpi2rowcol input types
 
 # v0.1 #
 - Initial version pulled out of pytesmo

pygeogrids/grids.py

@@ -414,7 +414,7 @@ def gpi2rowcol(self, gpi):
         try:
             gpi[0]
             iterable = True
-        except TypeError:
+        except (TypeError, IndexError):
             iterable = False
         gpi = np.atleast_1d(gpi)
         if len(self.shape) == 2:

tests/test_grid.py

@@ -144,6 +144,17 @@ def test_gpi2rowcol(self):
         assert row == row_should
         assert column == column_should
 
+    def test_gpi2rowcol_np_int(self):
+        """
+        test if gpi to row column lookup works correctly
+        """
+        gpi = np.array([200])[0]
+        row_should = 1
+        column_should = 200 - 144
+        row, column = self.grid.gpi2rowcol(gpi)
+        assert row == row_should
+        assert column == column_should
+
     def test_gpi2rowcol_iterable(self):
         """
         test if gpi to row column lookup works correctly