refactor: sqlglot scalar unit tests expressions (#1916)

The sqlglot scalar unit tests are being converted to use ArrayValue for operations, since some scalar functions do not have a matching pandas API.

Author: chelsea-lin

tests/unit/core/compile/sqlglot/expressions/snapshots/test_binary_compiler/test_add_numeric/out.sql

@@ -1,16 +1,13 @@
 WITH `bfcte_0` AS (
   SELECT
-    `int64_col` AS `bfcol_0`,
-    `rowindex` AS `bfcol_1`
+    `int64_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`scalar_types`
 ), `bfcte_1` AS (
   SELECT
     *,
-    `bfcol_1` AS `bfcol_4`,
-    `bfcol_0` + `bfcol_0` AS `bfcol_5`
+    `bfcol_0` + `bfcol_0` AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_4` AS `rowindex`,
-  `bfcol_5` AS `int64_col`
+  `bfcol_1` AS `int64_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_binary_compiler/test_add_numeric_w_scalar/out.sql

@@ -1,16 +1,13 @@
 WITH `bfcte_0` AS (
   SELECT
-    `int64_col` AS `bfcol_0`,
-    `rowindex` AS `bfcol_1`
+    `int64_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`scalar_types`
 ), `bfcte_1` AS (
   SELECT
     *,
-    `bfcol_1` AS `bfcol_4`,
-    `bfcol_0` + 1 AS `bfcol_5`
+    `bfcol_0` + 1 AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_4` AS `rowindex`,
-  `bfcol_5` AS `int64_col`
+  `bfcol_1` AS `int64_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_binary_compiler/test_add_string/out.sql

@@ -1,16 +1,13 @@
 WITH `bfcte_0` AS (
   SELECT
-    `rowindex` AS `bfcol_0`,
-    `string_col` AS `bfcol_1`
+    `string_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`scalar_types`
 ), `bfcte_1` AS (
   SELECT
     *,
-    `bfcol_0` AS `bfcol_4`,
-    CONCAT(`bfcol_1`, 'a') AS `bfcol_5`
+    CONCAT(`bfcol_0`, 'a') AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_4` AS `rowindex`,
-  `bfcol_5` AS `string_col`
+  `bfcol_1` AS `string_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_binary_compiler/test_json_set/out.sql

@@ -1,20 +1,13 @@
 WITH `bfcte_0` AS (
   SELECT
-    `rowindex` AS `bfcol_0`,
-    `json_col` AS `bfcol_1`
+    `json_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`json_types`
 ), `bfcte_1` AS (
   SELECT
     *,
-    JSON_SET(`bfcol_1`, '$.a', 100) AS `bfcol_4`
+    JSON_SET(`bfcol_0`, '$.a', 100) AS `bfcol_1`
   FROM `bfcte_0`
-), `bfcte_2` AS (
-  SELECT
-    *,
-    JSON_SET(`bfcol_4`, '$.b', 'hi') AS `bfcol_7`
-  FROM `bfcte_1`
 )
 SELECT
-  `bfcol_0` AS `rowindex`,
-  `bfcol_7` AS `json_col`
-FROM `bfcte_2`
+  `bfcol_1` AS `json_col`
+FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_unary_compiler/test_array_index/out.sql

@@ -1,15 +1,13 @@
 WITH `bfcte_0` AS (
   SELECT
-    `rowindex` AS `bfcol_0`,
-    `string_list_col` AS `bfcol_1`
+    `string_list_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`repeated_types`
 ), `bfcte_1` AS (
   SELECT
     *,
-    `bfcol_1`[SAFE_OFFSET(1)] AS `bfcol_4`
+    `bfcol_0`[SAFE_OFFSET(1)] AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_0` AS `rowindex`,
-  `bfcol_4` AS `string_list_col`
+  `bfcol_1` AS `string_list_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_unary_compiler/test_array_slice_with_only_start/out.sql

@@ -1,21 +1,19 @@
 WITH `bfcte_0` AS (
   SELECT
-    `rowindex` AS `bfcol_0`,
-    `string_list_col` AS `bfcol_1`
+    `string_list_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`repeated_types`
 ), `bfcte_1` AS (
   SELECT
     *,
     ARRAY(
       SELECT
         el
-      FROM UNNEST(`bfcol_1`) AS el WITH OFFSET AS slice_idx
+      FROM UNNEST(`bfcol_0`) AS el WITH OFFSET AS slice_idx
       WHERE
         slice_idx >= 1
-    ) AS `bfcol_4`
+    ) AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_0` AS `rowindex`,
-  `bfcol_4` AS `string_list_col`
+  `bfcol_1` AS `string_list_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_unary_compiler/test_array_slice_with_start_and_stop/out.sql

@@ -1,21 +1,19 @@
 WITH `bfcte_0` AS (
   SELECT
-    `rowindex` AS `bfcol_0`,
-    `string_list_col` AS `bfcol_1`
+    `string_list_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`repeated_types`
 ), `bfcte_1` AS (
   SELECT
     *,
     ARRAY(
       SELECT
         el
-      FROM UNNEST(`bfcol_1`) AS el WITH OFFSET AS slice_idx
+      FROM UNNEST(`bfcol_0`) AS el WITH OFFSET AS slice_idx
       WHERE
         slice_idx >= 1 AND slice_idx < 5
-    ) AS `bfcol_4`
+    ) AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_0` AS `rowindex`,
-  `bfcol_4` AS `string_list_col`
+  `bfcol_1` AS `string_list_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_unary_compiler/test_array_to_string/out.sql

@@ -1,15 +1,13 @@
 WITH `bfcte_0` AS (
   SELECT
-    `rowindex` AS `bfcol_0`,
-    `string_list_col` AS `bfcol_1`
+    `string_list_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`repeated_types`
 ), `bfcte_1` AS (
   SELECT
     *,
-    ARRAY_TO_STRING(`bfcol_1`, '.') AS `bfcol_4`
+    ARRAY_TO_STRING(`bfcol_0`, '.') AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_0` AS `rowindex`,
-  `bfcol_4` AS `string_list_col`
+  `bfcol_1` AS `string_list_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_unary_compiler/test_json_extract/out.sql

@@ -1,15 +1,13 @@
 WITH `bfcte_0` AS (
   SELECT
-    `rowindex` AS `bfcol_0`,
-    `json_col` AS `bfcol_1`
+    `json_col` AS `bfcol_0`
   FROM `bigframes-dev`.`sqlglot_test`.`json_types`
 ), `bfcte_1` AS (
   SELECT
     *,
-    JSON_EXTRACT(`bfcol_1`, '$') AS `bfcol_4`
+    JSON_EXTRACT(`bfcol_0`, '$') AS `bfcol_1`
   FROM `bfcte_0`
 )
 SELECT
-  `bfcol_0` AS `rowindex`,
-  `bfcol_4` AS `json_col`
+  `bfcol_1` AS `json_col`
 FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/snapshots/test_unary_compiler/test_json_extract_array/out.sql

@@ -0,0 +1,13 @@
+WITH `bfcte_0` AS (
+  SELECT
+    `json_col` AS `bfcol_0`
+  FROM `bigframes-dev`.`sqlglot_test`.`json_types`
+), `bfcte_1` AS (
+  SELECT
+    *,
+    JSON_EXTRACT_ARRAY(`bfcol_0`, '$') AS `bfcol_1`
+  FROM `bfcte_0`
+)
+SELECT
+  `bfcol_1` AS `json_col`
+FROM `bfcte_1`