include st_within in python import and scala test

databrickslabs · Feb 8, 2024 · d22a7af · d22a7af
1 parent acc0b2a
commit d22a7af
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/mosaic/api/predicates.py b/python/mosaic/api/predicates.py
@@ -9,7 +9,7 @@
 ######################
 
 
-__all__ = ["st_intersects", "st_contains"]
+__all__ = ["st_intersects", "st_contains", "st_within"]
 
 
 def st_intersects(left_geom: ColumnOrName, right_geom: ColumnOrName) -> Column:

diff --git a/src/test/scala/com/databricks/labs/mosaic/functions/MosaicContextBehaviors.scala b/src/test/scala/com/databricks/labs/mosaic/functions/MosaicContextBehaviors.scala
@@ -105,6 +105,7 @@ trait MosaicContextBehaviors extends MosaicSpatialQueryTest {
         noException should be thrownBy getFunc("st_perimeter").apply(Seq(multiPolygon.expr))
         noException should be thrownBy getFunc("st_distance").apply(Seq(multiPolygon.expr, pointWkt))
         noException should be thrownBy getFunc("st_contains").apply(Seq(multiPolygon.expr, pointWkt))
+        noException should be thrownBy getFunc("st_within").apply(Seq(multiPolygon.expr, pointWkt))
         noException should be thrownBy getFunc("st_translate").apply(Seq(multiPolygon.expr, xLit, yLit))
         noException should be thrownBy getFunc("st_scale").apply(Seq(multiPolygon.expr, xLit, yLit))
         noException should be thrownBy getFunc("st_rotate").apply(Seq(multiPolygon.expr, xLit, yLit))