diff --git a/python/cuspatial/benchmarks/io/bench_geoseries.py b/python/cuspatial/benchmarks/io/bench_geoseries.py new file mode 100644 index 000000000..0e4bf5c15 --- /dev/null +++ b/python/cuspatial/benchmarks/io/bench_geoseries.py @@ -0,0 +1,23 @@ +# Copyright (c) 2022, NVIDIA CORPORATION. + +"""Benchmarks of GeoSeries methods.""" + +import pytest +from config import cuspatial, cupy +from utils import benchmark_with_object + + +def bench_from_geoseries_100(benchmark, gpdf_100): + benchmark(cuspatial.from_geopandas, gpdf_100["geometry"]) + + +def bench_from_geoseries_1000(benchmark, gpdf_1000): + benchmark(cuspatial.from_geopandas, gpdf_1000["geometry"]) + + +def bench_from_geoseries_10000(benchmark, gpdf_10000): + benchmark(cuspatial.from_geopandas, gpdf_10000["geometry"]) + + +def bench_from_geoseries_100000(benchmark, gpdf_100000): + benchmark(cuspatial.from_geopandas, gpdf_100000["geometry"])