Skip to content

Commit fa65c68

Browse files
Tangruilinalamb
andauthored
[task #8213]Port tests in select.rs to sqllogictest (#8967)
* [task #8213]Part of Port tests in select.rs to sqllogictest Signed-off-by: tangruilin <[email protected]> * test large strings as well * Restore parameter tests * make test deterministic --------- Signed-off-by: tangruilin <[email protected]> Co-authored-by: Andrew Lamb <[email protected]>
1 parent b97daf7 commit fa65c68

File tree

3 files changed

+410
-632
lines changed

3 files changed

+410
-632
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;
@@ -249,21 +248,6 @@ async fn register_alltypes_parquet(ctx: &SessionContext) {
249248
.unwrap();
250249
}
251250

252-
/// Return a new table provider that has a single Int32 column with
253-
/// values between `seq_start` and `seq_end`
254-
pub fn table_with_sequence(
255-
seq_start: i32,
256-
seq_end: i32,
257-
) -> Result<Arc<dyn TableProvider>> {
258-
let schema = Arc::new(Schema::new(vec![Field::new("i", DataType::Int32, true)]));
259-
let arr = Arc::new(Int32Array::from((seq_start..=seq_end).collect::<Vec<_>>()));
260-
let partitions = vec![vec![RecordBatch::try_new(
261-
schema.clone(),
262-
vec![arr as ArrayRef],
263-
)?]];
264-
Ok(Arc::new(MemTable::try_new(schema, partitions)?))
265-
}
266-
267251
pub struct ExplainNormalizer {
268252
replacements: Vec<(String, String)>,
269253
}

0 commit comments

Comments
 (0)