fix

duckdb · Apr 17, 2024 · af1996e · af1996e
1 parent 426a4fb
commit af1996e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/vtab/arrow.rs b/src/vtab/arrow.rs
@@ -532,7 +532,7 @@ mod test {
     use crate::{Connection, Result};
     use arrow::{
         array::{
-            Array, AsArray, Date32Array, Date64Array, Float64Array, Int32Array, PrimitiveArray, StringArray,
+            Array, ArrayRef, AsArray, Date32Array, Date64Array, Float64Array, Int32Array, PrimitiveArray, StringArray,
             StructArray, Time32SecondArray, Time64MicrosecondArray, TimestampMicrosecondArray,
             TimestampMillisecondArray, TimestampNanosecondArray, TimestampSecondArray,
         },
@@ -614,6 +614,8 @@ mod test {
         let mut stmt = db.prepare("SELECT s FROM t1")?;
         let rbs: Vec<RecordBatch> = stmt.query_arrow([])?.collect();
         assert_eq!(rbs.iter().map(|op| op.num_rows()).sum::<usize>(), 2);
+
+        Ok(())
     }
 
     fn check_rust_primitive_array_roundtrip<T1, T2>(