Skip to content

Commit d628924

Browse files
committed
[task #8213]Part of Port tests in select.rs to sqllogictest
Signed-off-by: tangruilin <[email protected]>
1 parent d6ab343 commit d628924

File tree

3 files changed

+349
-644
lines changed

3 files changed

+349
-644
lines changed

datafusion/core/tests/sql/mod.rs

Lines changed: 1 addition & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -22,16 +22,15 @@ use arrow::{
2222
util::display::array_value_to_string,
2323
};
2424

25-
use datafusion::datasource::TableProvider;
2625
use datafusion::error::Result;
2726
use datafusion::logical_expr::{Aggregate, LogicalPlan, TableScan};
27+
use datafusion::physical_plan::collect;
2828
use datafusion::physical_plan::metrics::MetricValue;
2929
use datafusion::physical_plan::ExecutionPlan;
3030
use datafusion::physical_plan::ExecutionPlanVisitor;
3131
use datafusion::prelude::*;
3232
use datafusion::test_util;
3333
use datafusion::{assert_batches_eq, assert_batches_sorted_eq};
34-
use datafusion::{datasource::MemTable, physical_plan::collect};
3534
use datafusion::{execution::context::SessionContext, physical_plan::displayable};
3635
use datafusion_common::{assert_contains, assert_not_contains};
3736
use object_store::path::Path;
@@ -322,21 +321,6 @@ async fn register_alltypes_parquet(ctx: &SessionContext) {
322321
.unwrap();
323322
}
324323

325-
/// Return a new table provider that has a single Int32 column with
326-
/// values between `seq_start` and `seq_end`
327-
pub fn table_with_sequence(
328-
seq_start: i32,
329-
seq_end: i32,
330-
) -> Result<Arc<dyn TableProvider>> {
331-
let schema = Arc::new(Schema::new(vec![Field::new("i", DataType::Int32, true)]));
332-
let arr = Arc::new(Int32Array::from((seq_start..=seq_end).collect::<Vec<_>>()));
333-
let partitions = vec![vec![RecordBatch::try_new(
334-
schema.clone(),
335-
vec![arr as ArrayRef],
336-
)?]];
337-
Ok(Arc::new(MemTable::try_new(schema, partitions)?))
338-
}
339-
340324
pub struct ExplainNormalizer {
341325
replacements: Vec<(String, String)>,
342326
}

0 commit comments

Comments
 (0)