Skip to content

Commit 773de06

Browse files
Added tests with are writing into parquet files in memory for issue apache#15158
1 parent 8e2bfa4 commit 773de06

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

datafusion/wasmtest/src/lib.rs

+25
Original file line numberDiff line numberDiff line change
@@ -182,4 +182,29 @@ mod test {
182182
let task_ctx = ctx.task_ctx();
183183
let _ = collect(physical_plan, task_ctx).await.unwrap();
184184
}
185+
186+
#[wasm_bindgen_test(unsupported = tokio::test)]
187+
async fn test_parquet_write() {
188+
let schema = Arc::new(Schema::new(vec![
189+
Field::new("id", DataType::Int32, false),
190+
Field::new("value", DataType::Utf8, false),
191+
]));
192+
193+
let data: Vec<ArrayRef> = vec![
194+
Arc::new(Int32Array::from(vec![1])),
195+
Arc::new(StringArray::from(vec!["a"])),
196+
];
197+
198+
let batch = RecordBatch::try_new(schema.clone(), data).unwrap();
199+
let mut buffer = Vec::new();
200+
let mut writer = datafusion::parquet::arrow::ArrowWriter::try_new(
201+
&mut buffer,
202+
schema.clone(),
203+
None,
204+
)
205+
.unwrap();
206+
207+
writer.write(&batch).unwrap();
208+
writer.close().unwrap();
209+
}
185210
}

0 commit comments

Comments
 (0)