Merge branch 'databrickslabs:main' into main

a0x8o · Feb 13, 2024 · 8e57422 · 8e57422
2 parents 67aec7f + 398ef5c
commit 8e57422
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(pointWkt, multiPolygon.expr))
         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))