ARROW-6687: [Rust] Add regression tests for the new datafusion .parquet support

This commit is contained in:
Adam Lippai committed 2019-09-28 23:06:59 +02:00
1 parent 7fb6b751f6
commit 136087fcf3
2 files changed
+16 -1

No files matched your search

+15
View File
@@ -94,6 +94,21 @@ fn parquet_query() {
assert_eq!(expected, actual);
}
#[test]
fn parquet_single_nan_schema() {
let mut ctx = ExecutionContext::new();
let testdata = env::var("PARQUET_TEST_DATA").expect("PARQUET_TEST_DATA not defined");
ctx.register_parquet("single_nan", &format!("{}/single_nan.parquet", testdata))
.unwrap();
let sql = "SELECT mycol FROM single_nan";
let relation = ctx.sql(&sql, 1024 * 1024).unwrap();
let mut results = relation.borrow_mut();
while let Some(batch) = results.next().unwrap() {
assert_eq!(1, batch.num_rows());
assert_eq!(1, batch.num_columns());
}
}
#[test]
fn csv_count_star() {
let mut ctx = ExecutionContext::new();