geoarrow_array/test/
geometrycollection.rs

1use geoarrow_schema::{CoordType, Dimension, GeometryCollectionType};
2use geoarrow_test::raw;
3
4use crate::array::GeometryCollectionArray;
5use crate::builder::GeometryCollectionBuilder;
6
7pub fn array(
8    coord_type: CoordType,
9    dim: Dimension,
10    _prefer_multi: bool,
11) -> GeometryCollectionArray {
12    let typ = GeometryCollectionType::new(dim, Default::default()).with_coord_type(coord_type);
13    let geoms = match dim {
14        Dimension::XY => raw::geometrycollection::xy::geoms(),
15        Dimension::XYZ => raw::geometrycollection::xyz::geoms(),
16        Dimension::XYM => raw::geometrycollection::xym::geoms(),
17        Dimension::XYZM => raw::geometrycollection::xyzm::geoms(),
18    };
19
20    GeometryCollectionBuilder::from_nullable_geometry_collections(&geoms, typ)
21        .unwrap()
22        .finish()
23}