more

Author: alamb

datafusion/core/tests/physical_optimizer/enforce_sorting.rs

@@ -176,13 +176,8 @@ impl EnforceSortingTest {
 
         // return a string with both input and optimized plan
         format!(
-            "let expected_input = [\n\
-            {expected_input:#?}\n\
-            ];\n
-            let expected_optimized = [\n\
-            ---------------\n\
-            ];\
-            {expected_optimized:#?}",
+            "let expected_input = {expected_input:#?}\n\
+            let expected_optimized = {expected_optimized:#?}",
         )
     }
 }
@@ -295,17 +290,12 @@ async fn test_remove_unnecessary_sort5() -> Result<()> {
     };
     insta::assert_snapshot!(test.run(), @r#"
     let expected_input = [
-    [
         "SortExec: expr=[a@2 ASC], preserve_partitioning=[false]",
         "  HashJoinExec: mode=Partitioned, join_type=Inner, on=[(col_a@0, c@2)]",
         "    DataSourceExec: partitions=1, partition_sizes=[0]",
         "    DataSourceExec: file_groups={1 group: [[x]]}, projection=[a, b, c, d, e], output_ordering=[a@0 ASC], file_type=parquet",
     ]
-    ];
-
-                let expected_optimized = [
-    ---------------
-    ];[
+    let expected_optimized = [
         "HashJoinExec: mode=Partitioned, join_type=Inner, on=[(col_a@0, c@2)]",
         "  DataSourceExec: partitions=1, partition_sizes=[0]",
         "  DataSourceExec: file_groups={1 group: [[x]]}, projection=[a, b, c, d, e], output_ordering=[a@0 ASC], file_type=parquet",