From 258c7df9cab21a4978c100568907ac1cb7fd6ee0 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Tue, 23 Jan 2024 06:06:10 +0100
Subject: Adding upstream version 20.9.0.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 tests/dialects/test_presto.py | 36 ++++++++++++++++++++++++------------
 1 file changed, 24 insertions(+), 12 deletions(-)

(limited to 'tests/dialects/test_presto.py')

diff --git a/tests/dialects/test_presto.py b/tests/dialects/test_presto.py
index 8b5080c..387b0e0 100644
--- a/tests/dialects/test_presto.py
+++ b/tests/dialects/test_presto.py
@@ -182,12 +182,12 @@ class TestPresto(Validator):
     def test_interval_plural_to_singular(self):
         # Microseconds, weeks and quarters are not supported in Presto/Trino INTERVAL literals
         unit_to_expected = {
-            "SeCoNds": "second",
-            "minutes": "minute",
-            "hours": "hour",
-            "days": "day",
-            "months": "month",
-            "years": "year",
+            "SeCoNds": "SECOND",
+            "minutes": "MINUTE",
+            "hours": "HOUR",
+            "days": "DAY",
+            "months": "MONTH",
+            "years": "YEAR",
         }
 
         for unit, expected in unit_to_expected.items():
@@ -279,10 +279,10 @@ class TestPresto(Validator):
             },
         )
         self.validate_all(
-            "DATE_ADD('day', 1, x)",
+            "DATE_ADD('DAY', 1, x)",
             write={
-                "duckdb": "x + INTERVAL 1 day",
-                "presto": "DATE_ADD('day', 1, x)",
+                "duckdb": "x + INTERVAL 1 DAY",
+                "presto": "DATE_ADD('DAY', 1, x)",
                 "hive": "DATE_ADD(x, 1)",
                 "spark": "DATE_ADD(x, 1)",
             },
@@ -582,6 +582,8 @@ class TestPresto(Validator):
                 },
             )
 
+        self.validate_identity("SELECT a FROM test TABLESAMPLE BERNOULLI (50)")
+        self.validate_identity("SELECT a FROM test TABLESAMPLE SYSTEM (75)")
         self.validate_identity("string_agg(x, ',')", "ARRAY_JOIN(ARRAY_AGG(x), ',')")
         self.validate_identity("SELECT * FROM x OFFSET 1 LIMIT 1")
         self.validate_identity("SELECT * FROM x OFFSET 1 FETCH FIRST 1 ROWS ONLY")
@@ -600,6 +602,16 @@ class TestPresto(Validator):
             "SELECT * FROM example.testdb.customer_orders FOR TIMESTAMP AS OF CAST('2022-03-23 09:59:29.803 Europe/Vienna' AS TIMESTAMP)"
         )
 
+        self.validate_all(
+            "SELECT LAST_DAY_OF_MONTH(CAST('2008-11-25' AS DATE))",
+            read={
+                "duckdb": "SELECT LAST_DAY(CAST('2008-11-25' AS DATE))",
+            },
+            write={
+                "duckdb": "SELECT LAST_DAY(CAST('2008-11-25' AS DATE))",
+                "presto": "SELECT LAST_DAY_OF_MONTH(CAST('2008-11-25' AS DATE))",
+            },
+        )
         self.validate_all(
             "SELECT MAX_BY(a.id, a.timestamp) FROM a",
             read={
@@ -707,9 +719,9 @@ class TestPresto(Validator):
             },
         )
         self.validate_all("VALUES 1, 2, 3", write={"presto": "VALUES (1), (2), (3)"})
-        self.validate_all("INTERVAL '1 day'", write={"trino": "INTERVAL '1' day"})
-        self.validate_all("(5 * INTERVAL '7' day)", read={"": "INTERVAL '5' week"})
-        self.validate_all("(5 * INTERVAL '7' day)", read={"": "INTERVAL '5' WEEKS"})
+        self.validate_all("INTERVAL '1 day'", write={"trino": "INTERVAL '1' DAY"})
+        self.validate_all("(5 * INTERVAL '7' DAY)", read={"": "INTERVAL '5' WEEK"})
+        self.validate_all("(5 * INTERVAL '7' DAY)", read={"": "INTERVAL '5' WEEKS"})
         self.validate_all(
             "SELECT SUBSTRING(a, 1, 3), SUBSTRING(a, LENGTH(a) - (3 - 1))",
             read={
-- 
cgit v1.2.3