Skip to content

Commit 6cc24e4

Browse files
Update datafusion/core/tests/physical_optimizer/partition_statistics.rs
Co-authored-by: Berkay Şahin <124376117+berkaysynnada@users.noreply.github.com>
1 parent aa62950 commit 6cc24e4

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

datafusion/core/tests/physical_optimizer/partition_statistics.rs

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -553,7 +553,15 @@ mod test {
553553
vec![None],
554554
Arc::clone(&scan),
555555
scan_schema.clone(),
556-
)?);
556+
)?) as _;
557+
558+
let mut plan_string = get_plan_string(&aggregate_exec_partial);
559+
let _ = plan_string.swap_remove(1);
560+
let expected_plan = vec![
561+
"AggregateExec: mode=Partial, gby=[id@0 as id, 1 + id@0 as expr], aggr=[COUNT(c)]",
562+
//" DataSourceExec: file_groups={2 groups: [[.../datafusion/core/tests/data/test_statistics_per_partition/date=2025-03-01/j5fUeSDQo22oPyPU.parquet, .../datafusion/core/tests/data/test_statistics_per_partition/date=2025-03-02/j5fUeSDQo22oPyPU.parquet], [.../datafusion/core/tests/data/test_statistics_per_partition/date=2025-03-03/j5fUeSDQo22oPyPU.parquet, .../datafusion/core/tests/data/test_statistics_per_partition/date=2025-03-04/j5fUeSDQo22oPyPU.parquet]]}, projection=[id, date], file_type=parquet
563+
];
564+
assert_eq!(plan_string, expected_plan);
557565

558566
let p0_statistics = aggregate_exec_partial.partition_statistics(Some(0))?;
559567

0 commit comments

Comments
 (0)