Update tests

geoarrow · Sep 6, 2024 · 5da2330 · 5da2330
1 parent 59d8a4f
commit 5da2330
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/python/geoarrow-compute/pyproject.toml b/python/geoarrow-compute/pyproject.toml
@@ -15,5 +15,5 @@ dynamic = ["version"]
 
 [tool.maturin]
 features = ["pyo3/extension-module"]
-module-name = "geoarrow.rust.core._compute"
+module-name = "geoarrow.rust.compute._compute"
 python-source = "python"
diff --git a/python/tests/algorithm/test_affine_ops.py b/python/tests/algorithm/test_affine_ops.py
@@ -1,3 +1,4 @@
+import geoarrow.rust.compute as grc
 import geoarrow.rust.core as gars
 import geodatasets
 import geopandas as gpd
@@ -19,7 +20,7 @@ def test_affine_function():
     yoff = 20
 
     affine = Affine.translation(xoff, yoff)
-    translated = gars.affine_transform(geom, affine)
+    translated = grc.affine_transform(geom, affine)
     orig_geom = geom.chunk(0)[0]
     translated_geom = translated.chunk(0)[0]
 

diff --git a/python/tests/algorithm/test_area.py b/python/tests/algorithm/test_area.py
@@ -1,3 +1,4 @@
+import geoarrow.rust.compute as grc
 import geoarrow.rust.core as gars
 import geodatasets
 import geopandas as gpd
@@ -13,7 +14,7 @@ def test_area():
     assert isinstance(gdf, gpd.GeoDataFrame)
 
     table = gars.from_geopandas(gdf)
-    ga_area = gars.area(gars.geometry_col(table))
+    ga_area = grc.area(gars.geometry_col(table))
     pa_arr = pa.chunked_array(ga_area)
     assert pa_arr.num_chunks == 1