Add test

apache · Jun 5, 2022 · 6b6f148 · 6b6f148
1 parent b928309
commit 6b6f148
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/arrow/src/ipc/reader.rs b/arrow/src/ipc/reader.rs
@@ -1942,4 +1942,17 @@ mod tests {
         let output_batch = roundtrip_ipc_stream(&input_batch);
         assert_eq!(input_batch, output_batch);
     }
+
+    #[test]
+    fn test_no_columns_batch() {
+        let schema = Arc::new(Schema::new(vec![]));
+        let options = RecordBatchOptions {
+            match_field_names: true,
+            row_count: Some(10),
+        };
+        let input_batch =
+            RecordBatch::try_new_with_options(schema, vec![], &options).unwrap();
+        let output_batch = roundtrip_ipc_stream(&input_batch);
+        assert_eq!(input_batch, output_batch);
+    }
 }