try 4

Author: alamb

datafusion/core/tests/physical_optimizer/enforce_sorting.rs

@@ -171,15 +171,18 @@ impl EnforceSortingTest {
             .indent(true)
             .to_string();
 
+        let expected_input: Vec<&str> = input_plan_string.trim().lines().collect();
+        let expected_optimized: Vec<&str> =optimized_plan_string.trim().lines().collect();
+
         // return a string with both input and optimized plan
         format!(
             "let expected_input = [\n\
-            {input_plan_string}\n\
+            {expected_input:#?}\n\
             ];\n
             let expected_optimized = [\n\
             ---------------\n\
             ];\
-            {optimized_plan_string}",
+            {expected_optimized:#?}",
         )
     }
 }
@@ -290,21 +293,24 @@ async fn test_remove_unnecessary_sort5() -> Result<()> {
         plan: physical_plan,
         repartition_sorts: true,
     };
-    insta::assert_snapshot!(test.run(), @r"
+    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
-
+    [
+        "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 = [
     ---------------
-    ];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
-    ");
+    ];[
+        "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",
+    ]
+    "#);
     Ok(())
 }