Updated the FQN of the unparser CLI

snowflakedb · Jun 4, 2024 · e508b5d · e508b5d
1 parent 9c76dc9
commit e508b5d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/src/snowflake/snowpark/mock/_connection.py b/src/snowflake/snowpark/mock/_connection.py
@@ -800,6 +800,9 @@ def create_coprocessor(self):
         # It's not necessary to mock this call.
         pass
 
+    def phase1_enabled(self):
+        return False
+
 
 def _fix_pandas_df_fixed_type(table_res: TableEmulator) -> "pandas.DataFrame":
     pd_df = pandas.DataFrame()

diff --git a/tests/ast/run-unparser.sh b/tests/ast/run-unparser.sh
diff --git a/tests/ast/test_ast_driver.py b/tests/ast/test_ast_driver.py
@@ -12,8 +12,6 @@
 
 DATA_DIR = TEST_DIR / "data"
 
-UNPARSER_SCRIPT = TEST_DIR / "run-unparser.sh"
-
 @dataclass
 class TestCase:
     filename: str
@@ -60,10 +58,11 @@ def render(ast_base64: str) -> str:
     """Uses the unparser to render the AST."""
     assert pytest.unparser_jar, "A valid Unparser JAR path must be supplied either via --unparser-jar=<path> or the environment variable SNOWPARK_UNPARSER_JAR"
     res = subprocess.run([
-        UNPARSER_SCRIPT,
+        "java",
+        "-cp",
         pytest.unparser_jar,
+        "com.snowflake.snowpark.experimental.unparser.UnparserCli",
         ast_base64,
-        "--lang", "python"
     ], capture_output=True, text=True)
     return res.stdout