fix area test

geoarrow · Jul 1, 2024 · cfe82aa · cfe82aa
1 parent 40e4736
commit cfe82aa
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/core/tests/algorithm/test_area.py b/python/core/tests/algorithm/test_area.py
@@ -13,8 +13,9 @@ def test_area():
     assert isinstance(gdf, gpd.GeoDataFrame)
 
     table = gars.from_geopandas(gdf)
-    ga_area = gars.geometry_col(table).area()
-    assert ga_area.num_chunks() == 1
+    ga_area = gars.area(gars.geometry_col(table))
+    pa_arr = pa.chunked_array(ga_area)
+    assert pa_arr.num_chunks == 1
 
-    pa_area = pa.array(ga_area.chunk(0))
+    pa_area = pa_arr.chunk(0)
     assert np.allclose(shapely_area, pa_area)