From 6a22086850fc960715b618e82f4c2e43a4529146 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Mon, 20 Feb 2023 09:50:35 +0100 Subject: Merging upstream version 11.2.0. Signed-off-by: Daniel Baumann --- tests/dataframe/unit/test_session.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'tests/dataframe/unit/test_session.py') diff --git a/tests/dataframe/unit/test_session.py b/tests/dataframe/unit/test_session.py index 5213667..f5b79fd 100644 --- a/tests/dataframe/unit/test_session.py +++ b/tests/dataframe/unit/test_session.py @@ -80,6 +80,15 @@ class TestDataframeSession(DataFrameSQLValidator): df.sql(pretty=False), ) + @mock.patch("sqlglot.schema", MappingSchema()) + def test_select_quoted(self): + sqlglot.schema.add_table('"TEST"', {"name": "string"}) + + self.assertEqual( + SparkSession().table('"TEST"').select(F.col("name")).sql(dialect="snowflake")[0], + '''SELECT "TEST"."name" AS "name" FROM "TEST" AS "TEST"''', + ) + @mock.patch("sqlglot.schema", MappingSchema()) def test_sql_with_aggs(self): # TODO: Do exact matches once CTE names are deterministic -- cgit v1.2.3