Case statements for the RANGE type. (GoogleCloudPlatform#415)

* Case statements for the RANGE type. * added test cases for range data type * see if using different quotes gets the test to pass * separated tests
leerho · May 21, 2024 · c1c515c · c1c515c
1 parent 0fc5414
commit c1c515c
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/udfs/community/test_cases.js b/udfs/community/test_cases.js
@@ -346,6 +346,24 @@ generate_udf_test("typeof", [
         expected_output: `"NUMERIC"`
     },
 ]);
+generate_udf_test("typeof", [
+    {
+        inputs: [`RANGE<DATE> "[null, 2024-04-25)"`],
+        expected_output: `"RANGE<DATE>"`
+    },
+]);
+generate_udf_test("typeof", [
+    {
+        inputs: [`RANGE<DATETIME> "[2021-01-01, 2024-04-25)"`],
+        expected_output: `"RANGE<DATETIME>"`
+    },
+]);
+generate_udf_test("typeof", [
+    {
+        inputs: [`RANGE(current_timestamp() - interval 5 day, current_timestamp())`],
+        expected_output: `"RANGE<TIMESTAMP>"`
+    },
+]);
 generate_udf_test("url_decode", [
     {
         inputs: [

diff --git a/udfs/community/typeof.sqlx b/udfs/community/typeof.sqlx
@@ -32,6 +32,9 @@ AS ( (
         WHEN literal LIKE '[%' THEN 'ARRAY'
         WHEN REGEXP_CONTAINS(literal, r'^(STRUCT)?\(') THEN 'STRUCT'
         WHEN literal LIKE 'ST_%' THEN 'GEOGRAPHY'
+        WHEN literal LIKE 'RANGE<DATE>%' THEN 'RANGE<DATE>'
+        WHEN literal LIKE 'RANGE<DATETIME>%' THEN 'RANGE<DATETIME>'
+        WHEN literal LIKE 'RANGE<TIMESTAMP>%' THEN 'RANGE<TIMESTAMP>'
         WHEN literal = 'NULL' THEN 'NULL'
       ELSE
       'UNKNOWN'