Modify test.

viirya · Nov 17, 2021 · ebf67d3 · ebf67d3
1 parent fe8445d
commit ebf67d3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/datafusion/src/execution/context.rs b/datafusion/src/execution/context.rs
@@ -3114,7 +3114,7 @@ mod tests {
     }
 
     #[tokio::test]
-    async fn ctx_sql_should_optimize_plan() -> Result<()> {
+    async fn ctx_sql_should_not_optimize_plan() -> Result<()> {
         let mut ctx = ExecutionContext::new();
         let plan1 = ctx
             .create_logical_plan("SELECT * FROM (SELECT 1) AS one WHERE TRUE AND TRUE")?;
@@ -3125,11 +3125,16 @@ mod tests {
             .sql("SELECT * FROM (SELECT 1) AS one WHERE TRUE AND TRUE")
             .await?;
 
-        assert_eq!(
+        assert_ne!(
             format!("{:?}", opt_plan1),
             format!("{:?}", plan2.to_logical_plan())
         );
 
+        assert_eq!(
+            format!("{:?}", opt_plan1),
+            format!("{:?}", ctx.optimize(&plan2.to_logical_plan())?)
+        );
+
         Ok(())
     }