(pts: Vec<&str>, ids: Vec<i32>)
| 791 | } |
| 792 | |
| 793 | fn make_partitioned_batch(pts: Vec<&str>, ids: Vec<i32>) -> RecordBatch { |
| 794 | let schema = Arc::new(ArrowSchema::new(vec![ |
| 795 | ArrowField::new("pt", ArrowDataType::Utf8, false), |
| 796 | ArrowField::new("id", ArrowDataType::Int32, false), |
| 797 | ])); |
| 798 | RecordBatch::try_new( |
| 799 | schema, |
| 800 | vec![ |
| 801 | Arc::new(arrow_array::StringArray::from(pts)), |
| 802 | Arc::new(Int32Array::from(ids)), |
| 803 | ], |
| 804 | ) |
| 805 | .unwrap() |
| 806 | } |
| 807 | |
| 808 | #[tokio::test] |
| 809 | async fn test_write_and_commit() { |
no outgoing calls