uniform json golden sql tests

Author: chelsea-lin

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

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

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

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

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

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

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

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

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

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

tests/unit/core/compile/sqlglot/expressions/test_unary_compiler.py

@@ -47,39 +47,32 @@ def test_array_slice_with_start_and_stop(repeated_types_df: bpd.DataFrame, snaps
 # JSON Ops
 def test_json_extract(json_types_df: bpd.DataFrame, snapshot):
     result = bbq.json_extract(json_types_df["json_col"], "$")
-    expected_sql = "JSON_EXTRACT(`bfcol_1`, '$') AS `bfcol_4`"
-    assert expected_sql in result.to_frame().sql
     snapshot.assert_match(result.to_frame().sql, "out.sql")
 
 
-def test_json_extract_array(json_types_df: bpd.DataFrame):
+def test_json_extract_array(json_types_df: bpd.DataFrame, snapshot):
     result = bbq.json_extract_array(json_types_df["json_col"], "$")
-    expected_sql = "JSON_EXTRACT_ARRAY(`bfcol_1`, '$') AS `bfcol_4`"
-    assert expected_sql in result.to_frame().sql
+    snapshot.assert_match(result.to_frame().sql, "out.sql")
 
 
-def test_json_extract_string_array(json_types_df: bpd.DataFrame):
+def test_json_extract_string_array(json_types_df: bpd.DataFrame, snapshot):
     result = bbq.json_extract_string_array(json_types_df["json_col"], "$")
-    expected_sql = "JSON_EXTRACT_STRING_ARRAY(`bfcol_1`, '$') AS `bfcol_4`"
-    assert expected_sql in result.to_frame().sql
+    snapshot.assert_match(result.to_frame().sql, "out.sql")
 
 
-def test_json_query(json_types_df: bpd.DataFrame):
+def test_json_query(json_types_df: bpd.DataFrame, snapshot):
     result = bbq.json_query(json_types_df["json_col"], "$")
-    expected_sql = "JSON_QUERY(`bfcol_1`, '$') AS `bfcol_4`"
-    assert expected_sql in result.to_frame().sql
+    snapshot.assert_match(result.to_frame().sql, "out.sql")
 
 
-def test_json_query_array(json_types_df: bpd.DataFrame):
+def test_json_query_array(json_types_df: bpd.DataFrame, snapshot):
     result = bbq.json_query_array(json_types_df["json_col"], "$")
-    expected_sql = "JSON_QUERY_ARRAY(`bfcol_1`, '$') AS `bfcol_4`"
-    assert expected_sql in result.to_frame().sql
+    snapshot.assert_match(result.to_frame().sql, "out.sql")
 
 
-def test_json_value(json_types_df: bpd.DataFrame):
+def test_json_value(json_types_df: bpd.DataFrame, snapshot):
     result = bbq.json_value(json_types_df["json_col"], "$")
-    expected_sql = "JSON_VALUE(`bfcol_1`, '$') AS `bfcol_4`"
-    assert expected_sql in result.to_frame().sql
+    snapshot.assert_match(result.to_frame().sql, "out.sql")
 
 
 def test_parse_json(scalar_types_df: bpd.DataFrame, snapshot):