Merge pull request #11788 from ydb-platform/mergelibs-241120-1113

Library import 241120-1113

1 files changed, 80 insertions, 0 deletions

diff --git a/yql/essentials/tests/sql/suites/agg_phases/count_opt.sql b/yql/essentials/tests/sql/suites/agg_phases/count_opt.sql
new file mode 100644
index 0000000000..97df3a4e5a
--- /dev/null
+++ b/yql/essentials/tests/sql/suites/agg_phases/count_opt.sql
@@ -0,0 +1,80 @@
+/* syntax version 1 */
+/* postgres can not */
+
+$t = SELECT * FROM AS_TABLE([<|key: 1, value: just(2)|>,<|key: 1, value: just(3)|>]);
+
+$p = 
+SELECT
+    key,count(value) as a
+FROM $t
+GROUP BY
+    key
+    with combine
+    ;
+
+$p = PROCESS $p;
+--select FormatType(TypeOf($p));
+select * from $p;
+
+$p = SELECT
+    key,count(a) as a
+FROM $p
+GROUP BY
+    key
+    with combinestate
+    ;
+    
+$p = PROCESS $p;
+--select FormatType(TypeOf($p));
+select * from $p;
+
+$p = SELECT
+    key,count(a) as a
+FROM $p
+GROUP BY
+    key
+    with mergestate
+    ;
+    
+$p = PROCESS $p;
+--select FormatType(TypeOf($p));
+select * from $p;
+
+    
+$p1 = SELECT
+    key,count(a) as a
+FROM $p
+GROUP BY
+    key
+    with mergefinalize
+    ;
+    
+$p1 = PROCESS $p1;
+--select FormatType(TypeOf($p1));
+select * from $p1;
+
+$p2 = SELECT
+    key,count(a) as a
+FROM (select key, just(a) as a from $p)
+GROUP BY
+    key
+    with mergemanyfinalize
+    ;
+    
+$p2 = PROCESS $p2;
+--select FormatType(TypeOf($p2));
+select * from $p2;
+
+$p3 = 
+SELECT
+    key,count(value) as a
+FROM $t
+GROUP BY
+    key
+    with finalize
+    ;
+
+$p3 = PROCESS $p3;
+--select FormatType(TypeOf($p));
+select * from $p3;
+