File tree 1 file changed +25
-0
lines changed
1 file changed +25
-0
lines changed Original file line number Diff line number Diff line change @@ -182,4 +182,29 @@ mod test {
182
182
let task_ctx = ctx. task_ctx ( ) ;
183
183
let _ = collect ( physical_plan, task_ctx) . await . unwrap ( ) ;
184
184
}
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
+ }
185
210
}
You can’t perform that action at this time.
0 commit comments