BUG: catch empty GeometryIndex in repr (#80)

xarray-contrib · Aug 27, 2024 · 66b541b · 66b541b
1 parent ba7ef28
commit 66b541b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/xvec/index.py b/xvec/index.py
@@ -277,6 +277,8 @@ def _repr_inline_(self, max_width: int) -> str:
     def __repr__(self) -> str:
         srs = _format_crs(self.crs)
         shape = self._index.index.shape[0]
+        if shape == 0:
+            return f"GeometryIndex([], crs={srs})"
         if shape < 10:
             wkts = [repr(g) for g in self._index.index]
         else:

diff --git a/xvec/tests/test_index.py b/xvec/tests/test_index.py
@@ -193,3 +193,8 @@ def test_repr(geom_dataset, geom_dataset_no_crs):
     long = xr.concat([geom_dataset] * 10, dim="geom")
     actual = repr(long.xindexes["geom"])
     expected = "GeometryIndex(\n    [<POINT (1 2)>\n     <POINT (3 4)>\n     <POINT (1 2)>\n     <POINT (3 4)>\n     ...\n     <POINT (1 2)>\n     <POINT (3 4)>\n     <POINT (1 2)>\n     <POINT (3 4)>],\n    crs=EPSG:26915)"
+
+    empty = geom_dataset.xvec.query("geom", shapely.Point(0, 0), predicate="intersects")
+    actual = repr(empty.xindexes["geom"])
+    expected = "GeometryIndex([], crs=EPSG:26915)"
+    assert actual == expected