diff options
Diffstat (limited to 'tests/dialects/test_spark.py')
| -rw-r--r-- | tests/dialects/test_spark.py | 25 |
1 files changed, 21 insertions, 4 deletions
diff --git a/tests/dialects/test_spark.py b/tests/dialects/test_spark.py index bcfd984..7c8ca1b 100644 --- a/tests/dialects/test_spark.py +++ b/tests/dialects/test_spark.py @@ -5,6 +5,9 @@ class TestSpark(Validator): dialect = "spark" def test_ddl(self): + self.validate_identity("CREATE TABLE foo (col VARCHAR(50))") + self.validate_identity("CREATE TABLE foo (col STRUCT<struct_col_a: VARCHAR((50))>)") + self.validate_all( "CREATE TABLE db.example_table (col_a struct<struct_col_a:int, struct_col_b:string>)", write={ @@ -223,6 +226,20 @@ TBLPROPERTIES ( self.validate_identity("SPLIT(str, pattern, lim)") self.validate_all( + "SELECT * FROM ((VALUES 1))", write={"spark": "SELECT * FROM (VALUES (1))"} + ) + self.validate_all( + "SELECT CAST(STRUCT('fooo') AS STRUCT<a: VARCHAR(2)>)", + write={"spark": "SELECT CAST(STRUCT('fooo') AS STRUCT<a: STRING>)"}, + ) + self.validate_all( + "SELECT CAST(123456 AS VARCHAR(3))", + write={ + "": "SELECT TRY_CAST(123456 AS TEXT)", + "spark": "SELECT CAST(123456 AS STRING)", + }, + ) + self.validate_all( "SELECT piv.Q1 FROM (SELECT * FROM produce PIVOT(SUM(sales) FOR quarter IN ('Q1', 'Q2'))) AS piv", read={ "snowflake": "SELECT piv.Q1 FROM produce PIVOT(SUM(sales) FOR quarter IN ('Q1', 'Q2')) piv", @@ -368,10 +385,10 @@ TBLPROPERTIES ( self.validate_all( "SELECT LEFT(x, 2), RIGHT(x, 2)", write={ - "duckdb": "SELECT SUBSTRING(x, 1, 2), SUBSTRING(x, LENGTH(x) - 2 + 1, 2)", - "presto": "SELECT SUBSTRING(x, 1, 2), SUBSTRING(x, LENGTH(x) - 2 + 1, 2)", - "hive": "SELECT SUBSTRING(x, 1, 2), SUBSTRING(x, LENGTH(x) - 2 + 1, 2)", - "spark": "SELECT SUBSTRING(x, 1, 2), SUBSTRING(x, LENGTH(x) - 2 + 1, 2)", + "duckdb": "SELECT LEFT(x, 2), RIGHT(x, 2)", + "presto": "SELECT SUBSTRING(x, 1, 2), SUBSTRING(x, LENGTH(x) - (2 - 1))", + "hive": "SELECT SUBSTRING(x, 1, 2), SUBSTRING(x, LENGTH(x) - (2 - 1))", + "spark": "SELECT LEFT(x, 2), RIGHT(x, 2)", }, ) self.validate_all( |