Merge pull request #530 from databrickslabs/within-minor

include `st_within` in python import and scala test (minor)
databrickslabs · Feb 12, 2024 · 398ef5c · 398ef5c
2 parents acc0b2a + 9e072f1
commit 398ef5c
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))