Skip to content

Commit f55bfd3

Browse files
committed
fix tests
1 parent 7ef8fd6 commit f55bfd3

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

datafusion/core/tests/physical_optimizer/filter_pushdown/mod.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1819,6 +1819,8 @@ async fn test_topk_dynamic_filter_pushdown_integration() {
18191819
let mut cfg = SessionConfig::new();
18201820
cfg.options_mut().execution.parquet.pushdown_filters = true;
18211821
cfg.options_mut().execution.parquet.max_row_group_size = 128;
1822+
// Always pushdown filters into row filters for this test
1823+
cfg.options_mut().execution.parquet.filter_effectiveness_threshold = 0.0;
18221824
let ctx = SessionContext::new_with_config(cfg);
18231825
ctx.register_object_store(
18241826
ObjectStoreUrl::parse("memory://").unwrap().as_ref(),

datafusion/core/tests/sql/explain_analyze.rs

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -863,7 +863,7 @@ async fn parquet_explain_analyze() {
863863
.to_string();
864864

865865
// should contain aggregated stats
866-
assert_contains!(&formatted, "output_rows=8");
866+
assert_contains!(&formatted, "output_rows=5");
867867
assert_contains!(
868868
&formatted,
869869
"row_groups_pruned_bloom_filter=1 total \u{2192} 1 matched"
@@ -996,11 +996,7 @@ async fn parquet_recursive_projection_pushdown() -> Result<()> {
996996
@r"
997997
SortExec: expr=[id@0 ASC NULLS LAST], preserve_partitioning=[false]
998998
RecursiveQueryExec: name=number_series, is_distinct=false
999-
CoalescePartitionsExec
1000-
ProjectionExec: expr=[id@0 as id, 1 as level]
1001-
FilterExec: id@0 = 1
1002-
RepartitionExec: partitioning=RoundRobinBatch(NUM_CORES), input_partitions=1
1003-
DataSourceExec: file_groups={1 group: [[TMP_DIR/hierarchy.parquet]]}, projection=[id], file_type=parquet, predicate=id@0 = 1, pruning_predicate=id_null_count@2 != row_count@3 AND id_min@0 <= 1 AND 1 <= id_max@1, required_guarantees=[id in (1)]
999+
DataSourceExec: file_groups={1 group: [[TMP_DIR/hierarchy.parquet]]}, projection=[id, 1 as level], file_type=parquet, predicate=id@0 = 1, pruning_predicate=id_null_count@2 != row_count@3 AND id_min@0 <= 1 AND 1 <= id_max@1, required_guarantees=[id in (1)]
10041000
CoalescePartitionsExec
10051001
ProjectionExec: expr=[id@0 + 1 as ns.id + Int64(1), level@1 + 1 as ns.level + Int64(1)]
10061002
FilterExec: id@0 < 10

0 commit comments

Comments
 (0)