fix UT

AngersZhuuuu · Apr 19, 2021 · 6aa05fc · 6aa05fc
1 parent 76a746e
commit 6aa05fc
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 19 deletions.
diff --git a/sql/core/src/test/resources/sql-tests/inputs/transform.sql b/sql/core/src/test/resources/sql-tests/inputs/transform.sql
@@ -263,14 +263,17 @@ FROM script_trans
 WHERE a <= 4
 WINDOW w AS (PARTITION BY b ORDER BY a);
 
-SELECT TRANSFORM(b, MAX(a), CAST(SUM(c) AS STRING), myCol, myCol2)
-  USING 'cat' AS (a, b, c, d, e)
-FROM script_trans
-LATERAL VIEW explode(array(array(1,2,3))) myTable AS myCol
-LATERAL VIEW explode(myTable.myCol) myTable2 AS myCol2
-WHERE a <= 4
-GROUP BY b, myCol, myCol2
-HAVING max(a) > 1;
+SELECT a, b, c, CAST(d AS STriNG), e
+FROM (
+  SELECT TRANSFORM(b, MAX(a), CAST(SUM(c) AS STRING), myCol, myCol2)
+    USING 'cat' AS (a, b, c, d, e)
+  FROM script_trans
+  LATERAL VIEW explode(array(array(1,2,3))) myTable AS myCol
+  LATERAL VIEW explode(myTable.myCol) myTable2 AS myCol2
+  WHERE a <= 4
+  GROUP BY b, myCol, myCol2
+  HAVING max(a) > 1
+) tmp;
 
 FROM(
   FROM script_trans

diff --git a/sql/core/src/test/resources/sql-tests/results/transform.sql.out b/sql/core/src/test/resources/sql-tests/results/transform.sql.out
@@ -494,20 +494,23 @@ struct<a:string,b:string,c:string>
 
 
 -- !query
-SELECT TRANSFORM(b, MAX(a), CAST(SUM(c) AS STRING), myCol, myCol2)
-  USING 'cat' AS (a, b, c, d, e)
-FROM script_trans
-LATERAL VIEW explode(array(array(1,2,3))) myTable AS myCol
-LATERAL VIEW explode(myTable.myCol) myTable2 AS myCol2
-WHERE a <= 4
-GROUP BY b, myCol, myCol2
-HAVING max(a) > 1
+SELECT a, b, c, CAST(d AS STriNG), e
+FROM (
+  SELECT TRANSFORM(b, MAX(a), CAST(SUM(c) AS STRING), myCol, myCol2)
+    USING 'cat' AS (a, b, c, d, e)
+  FROM script_trans
+  LATERAL VIEW explode(array(array(1,2,3))) myTable AS myCol
+  LATERAL VIEW explode(myTable.myCol) myTable2 AS myCol2
+  WHERE a <= 4
+  GROUP BY b, myCol, myCol2
+  HAVING max(a) > 1
+) tmp
 -- !query schema
 struct<a:string,b:string,c:string,d:string,e:string>
 -- !query output
-5	4	6	[1, 2, 3]	1
-5	4	6	[1, 2, 3]	2
-5	4	6	[1, 2, 3]	3
+5	4	6	[1,2,3]	1
+5	4	6	[1,2,3]	2
+5	4	6	[1,2,3]	3
 
 
 -- !query