update

Author: xiedeyantu

datafusion/optimizer/src/unions_to_filter.rs

@@ -383,6 +383,10 @@ mod tests {
     struct VolatileTestUdf;
 
     impl ScalarUDFImpl for VolatileTestUdf {
+        fn as_any(&self) -> &dyn std::any::Any {
+            self
+        }
+
         fn name(&self) -> &str {
             "volatile_test"
         }

datafusion/substrait/tests/cases/roundtrip_logical_plan.rs

@@ -985,13 +985,20 @@ async fn roundtrip_is_not_unknown() -> Result<()> {
 
 #[tokio::test]
 async fn roundtrip_union() -> Result<()> {
-    roundtrip("SELECT a, e FROM data UNION SELECT a, e FROM data").await
+    assert_expected_plan(
+        "SELECT a, e FROM data UNION SELECT a, e FROM data",
+        "Aggregate: groupBy=[[data.a, data.e]], aggr=[[]]\n  TableScan: data projection=[a, e]",
+        false,
+    )
+    .await
 }
 
 #[tokio::test]
 async fn roundtrip_union2() -> Result<()> {
-    roundtrip(
+    assert_expected_plan(
         "SELECT a, b FROM data UNION SELECT a, b FROM data UNION SELECT a, b FROM data",
+        "Aggregate: groupBy=[[data.a, data.b]], aggr=[[]]\n  TableScan: data projection=[a, b]",
+        false,
     )
     .await
 }