check schema and number of batches as well (per suggestion from @alamb)

apache · Jan 17, 2025 · d93d8f6 · d93d8f6
1 parent 1ba1d96
commit d93d8f6
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/parquet/src/arrow/arrow_writer/mod.rs b/parquet/src/arrow/arrow_writer/mod.rs
@@ -3430,7 +3430,13 @@ mod tests {
 
         // read from parquet
         let bytes = Bytes::from(parquet_bytes);
-        let result = ParquetRecordBatchReaderBuilder::try_new(bytes);
-        result.unwrap();
+        let reader = ParquetRecordBatchReaderBuilder::try_new(bytes).unwrap();
+        assert_eq!(reader.schema(), &empty_batch.schema());
+        let batches: Vec<_> = reader
+            .build()
+            .unwrap()
+            .collect::<ArrowResult<Vec<_>>>()
+            .unwrap();
+        assert_eq!(batches.len(), 0);
     }
 }