docs(geospatial): fix flaky ci geo-literals doctests

ibis-project · Jan 31, 2024 · 417e81d · 417e81d
1 parent ac72157
commit 417e81d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ibis/expr/types/geospatial.py b/ibis/expr/types/geospatial.py
@@ -798,6 +798,8 @@ def length(self) -> ir.FloatingValue:
         --------
         >>> import ibis
         >>> ibis.options.interactive = True
+        >>> con = ibis.get_backend()
+        >>> con.load_extension("spatial")
         >>> import shapely
         >>> line = shapely.LineString([[0, 0], [1, 0], [1, 1]])
         >>> line_lit = ibis.literal(line, type="geometry")
@@ -1027,6 +1029,8 @@ def start_point(self) -> PointValue:
         --------
         >>> import ibis
         >>> ibis.options.interactive = True
+        >>> con = ibis.get_backend()
+        >>> con.load_extension("spatial")
         >>> import shapely
         >>> line = shapely.LineString([[0, 0], [1, 0], [1, 1]])
         >>> line_lit = ibis.literal(line, type="geometry")
@@ -1049,6 +1053,8 @@ def end_point(self) -> PointValue:
         --------
         >>> import ibis
         >>> ibis.options.interactive = True
+        >>> con = ibis.get_backend()
+        >>> con.load_extension("spatial")
         >>> import shapely
         >>> line = shapely.LineString([[0, 0], [1, 0], [1, 1]])
         >>> line_lit = ibis.literal(line, type="geometry")