Add roundtrip_expr_api test case

apache · Jun 21, 2024 · 2939e80 · 2939e80
1 parent 1b9f54f
commit 2939e80
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/datafusion/proto/tests/cases/roundtrip_logical_plan.rs b/datafusion/proto/tests/cases/roundtrip_logical_plan.rs
@@ -60,7 +60,7 @@ use datafusion_expr::{
     WindowFunctionDefinition, WindowUDF, WindowUDFImpl,
 };
 use datafusion_functions_aggregate::expr_fn::{
-    bit_and, bit_or, bit_xor, bool_and, bool_or,
+    array_agg, bit_and, bit_or, bit_xor, bool_and, bool_or
 };
 use datafusion_functions_aggregate::string_agg::string_agg;
 use datafusion_proto::bytes::{
@@ -675,6 +675,7 @@ async fn roundtrip_expr_api() -> Result<()> {
         string_agg(col("a").cast_to(&DataType::Utf8, &schema)?, lit("|")),
         bool_and(lit(true)),
         bool_or(lit(true)),
+        array_agg(lit(1))
     ];
 
     // ensure expressions created with the expr api can be round tripped