refactor: fix test_string_astype_json

Author: chelsea-lin

bigframes/core/compile/sqlglot/expressions/generic_ops.py

@@ -252,7 +252,7 @@ def _cast_to_json(expr: TypedExpr, op: ops.AsTypeOp) -> sge.Expression:
     sg_expr = expr.expr
 
     if from_type == dtypes.STRING_DTYPE:
-        func_name = "PARSE_JSON_IN_SAFE" if op.safe else "PARSE_JSON"
+        func_name = "SAFE.PARSE_JSON" if op.safe else "PARSE_JSON"
         return sge.func(func_name, sg_expr)
     if from_type in (dtypes.INT_DTYPE, dtypes.BOOL_DTYPE, dtypes.FLOAT_DTYPE):
         sg_expr = sge.Cast(this=sg_expr, to="STRING")

tests/unit/core/compile/sqlglot/expressions/snapshots/test_generic_ops/test_astype_json/out.sql

@@ -13,7 +13,7 @@ WITH `bfcte_0` AS (
     PARSE_JSON(CAST(`bool_col` AS STRING)) AS `bfcol_6`,
     PARSE_JSON(`string_col`) AS `bfcol_7`,
     PARSE_JSON(CAST(`bool_col` AS STRING)) AS `bfcol_8`,
-    PARSE_JSON_IN_SAFE(`string_col`) AS `bfcol_9`
+    SAFE.PARSE_JSON(`string_col`) AS `bfcol_9`
   FROM `bfcte_0`
 )
 SELECT