Fix failing test

geoarrow · Dec 23, 2024 · f757075 · f757075
1 parent 15dfdfb
commit f757075
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/rust/geoarrow/src/io/flatgeobuf/reader/sync.rs b/rust/geoarrow/src/io/flatgeobuf/reader/sync.rs
@@ -455,7 +455,7 @@ mod test {
         .unwrap();
 
         let geom_col = table.geometry_column(None).unwrap();
-        assert!(matches!(geom_col.data_type(), NativeType::Point(_, _)));
+        assert!(matches!(geom_col.data_type(), NativeType::Geometry(_)));
 
         let (batches, schema) = table.into_inner();
         assert_eq!(batches[0].num_rows(), 1);

diff --git a/rust/geoarrow/src/io/geozero/table/builder/properties.rs b/rust/geoarrow/src/io/geozero/table/builder/properties.rs
@@ -1,7 +1,7 @@
 use std::sync::Arc;
 
 use arrow_array::RecordBatch;
-use arrow_schema::{Field, Schema, SchemaBuilder};
+use arrow_schema::{Schema, SchemaBuilder};
 use chrono::{DateTime, Utc};
 use geozero::{FeatureProcessor, GeomProcessor, PropertyProcessor};
 
@@ -122,8 +122,9 @@ impl PropertiesBatchBuilder {
         let mut columns = Vec::with_capacity(self.columns.len());
 
         for (name, builder) in self.columns.into_iter() {
+            let field = builder.field().with_name(name);
             let array = builder.finish()?;
-            schema_builder.push(Field::new(name, array.data_type().clone(), true));
+            schema_builder.push(field);
             columns.push(array);
         }
 

diff --git a/rust/geoarrow/src/io/geozero/table/builder/table.rs b/rust/geoarrow/src/io/geozero/table/builder/table.rs
@@ -231,8 +231,6 @@ impl<G: GeometryArrayBuilder + GeomProcessor> GeoTableBuilder<G> {
 
         table.append_column(geom_field, geom_col.array_refs())?;
         Ok(table)
-        // TODO: 3d downcasting not yet supported
-        // table.downcast(false)
     }
 }