DRY up another test

python/cuspatial/cuspatial/tests/test_geoseries.py

@@ -1,7 +1,6 @@
 # Copyright (c) 2020-2024, NVIDIA CORPORATION.
 
 from enum import Enum
-from itertools import chain
 from numbers import Integral
 
 import cupy as cp
@@ -10,7 +9,6 @@
 import pandas as pd
 import pytest
 from geopandas.testing import assert_geoseries_equal
-from shapely import get_coordinates
 from shapely.affinity import rotate
 from shapely.geometry import (
  LineString,
@@ -185,14 +183,7 @@ def test_interleaved_point(gs):
  ).reset_index(drop=True),
  )
 
- polys_list = gs[gs.apply(lambda x: isinstance(x, (MultiPolygon, Polygon)))]
- polys = list(
- chain(polys_list.apply(get_coordinates))
- ) # flatten multigeometries
- coords_list = list(chain(*polys)) # flatten geometries
- xy_interleaved = list(chain(*coords_list)) # flatten coordinates
- x = xy_interleaved[::2]
- y = xy_interleaved[1::2]
+ xy, x, y = geometry_to_coords(gs, (MultiPolygon, Polygon))
 
  cudf.testing.assert_series_equal(
  cugs.polygons.x.reset_index(drop=True),