diff --git a/python/mosaic/api/predicates.py b/python/mosaic/api/predicates.py index 39f856597..9291423b8 100644 --- 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 index a4afc3a1a..bbcf44e6a 100644 --- 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))