diff options
Diffstat (limited to 'tests/dialects/test_athena.py')
| -rw-r--r-- | tests/dialects/test_athena.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/tests/dialects/test_athena.py b/tests/dialects/test_athena.py index 3288ada..5522976 100644 --- a/tests/dialects/test_athena.py +++ b/tests/dialects/test_athena.py @@ -7,6 +7,15 @@ class TestAthena(Validator): def test_athena(self): self.validate_identity( + "CREATE TABLE IF NOT EXISTS t (name STRING) LOCATION 's3://bucket/tmp/mytable/' TBLPROPERTIES ('table_type'='iceberg', 'FORMAT'='parquet')" + ) + self.validate_identity( + "UNLOAD (SELECT name1, address1, comment1, key1 FROM table1) " + "TO 's3://amzn-s3-demo-bucket/ partitioned/' " + "WITH (format = 'TEXTFILE', partitioned_by = ARRAY['key1'])", + check_command_warning=True, + ) + self.validate_identity( """USING EXTERNAL FUNCTION some_function(input VARBINARY) RETURNS VARCHAR LAMBDA 'some-name' @@ -14,7 +23,3 @@ class TestAthena(Validator): some_function(1)""", check_command_warning=True, ) - - self.validate_identity( - "CREATE TABLE IF NOT EXISTS t (name STRING) LOCATION 's3://bucket/tmp/mytable/' TBLPROPERTIES ('table_type'='iceberg', 'FORMAT'='parquet')" - ) |