fix python tests

Canner · Nov 14, 2024 · 6d7e9d0 · 6d7e9d0
1 parent 2b49890
commit 6d7e9d0
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 13 deletions.
diff --git a/ibis-server/tests/routers/v3/connector/bigquery/test_functions.py b/ibis-server/tests/routers/v3/connector/bigquery/test_functions.py
@@ -6,8 +6,8 @@
 
 from app.config import get_config
 from app.main import app
-from tests.routers.v3.connector.bigquery.conftest import base_url, function_list_path
 from tests.conftest import DATAFUSION_FUNCTION_COUNT
+from tests.routers.v3.connector.bigquery.conftest import base_url, function_list_path
 
 manifest = {
     "catalog": "my_catalog",

diff --git a/ibis-server/tests/routers/v3/connector/bigquery/test_query.py b/ibis-server/tests/routers/v3/connector/bigquery/test_query.py
@@ -42,17 +42,17 @@
                 {
                     "name": "timestamptz",
                     "expression": "cast('2024-01-01T23:59:59' as timestamp with time zone)",
-                    "type": "timestamp",
+                    "type": "timestamptz",
                 },
                 {
                     "name": "dst_utc_minus_5",
                     "expression": "cast('2024-01-15 23:00:00 America/New_York' as timestamp with time zone)",
-                    "type": "timestamp",
+                    "type": "timestamptz",
                 },
                 {
                     "name": "dst_utc_minus_4",
                     "expression": "cast('2024-07-15 23:00:00 America/New_York' as timestamp with time zone)",
-                    "type": "timestamp",
+                    "type": "timestamptz",
                 },
             ],
             "primaryKey": "o_orderkey",
@@ -83,9 +83,9 @@ def test_query(manifest_str, connection_info):
         assert len(result["data"]) == 1
         assert result["data"][0] == [
             "2024-01-01 23:59:59.000000",
-            "2024-01-01 23:59:59.000000",
-            "2024-01-16 04:00:00.000000",  # utc-5
-            "2024-07-16 03:00:00.000000",  # utc-4
+            "2024-01-01 23:59:59.000000 UTC",
+            "2024-01-16 04:00:00.000000 UTC",  # utc-5
+            "2024-07-16 03:00:00.000000 UTC",  # utc-4
             "36485_1202",
             1202,
             "1992-06-06",

diff --git a/ibis-server/tests/routers/v3/connector/postgres/test_query.py b/ibis-server/tests/routers/v3/connector/postgres/test_query.py
@@ -42,17 +42,17 @@
                 {
                     "name": "timestamptz",
                     "expression": "cast('2024-01-01T23:59:59' as timestamp with time zone)",
-                    "type": "timestamp",
+                    "type": "timestamptz",
                 },
                 {
                     "name": "dst_utc_minus_5",
                     "expression": "cast('2024-01-15 23:00:00 America/New_York' as timestamp with time zone)",
-                    "type": "timestamp",
+                    "type": "timestamptz",
                 },
                 {
                     "name": "dst_utc_minus_4",
                     "expression": "cast('2024-07-15 23:00:00 America/New_York' as timestamp with time zone)",
-                    "type": "timestamp",
+                    "type": "timestamptz",
                 },
             ],
             "primaryKey": "o_orderkey",
@@ -83,9 +83,9 @@ def test_query(manifest_str, connection_info):
         assert len(result["data"]) == 1
         assert result["data"][0] == [
             "2024-01-01 23:59:59.000000",
-            "2024-01-01 23:59:59.000000",
-            "2024-01-16 04:00:00.000000",  # utc-5
-            "2024-07-16 03:00:00.000000",  # utc-4
+            "2024-01-01 23:59:59.000000 UTC",
+            "2024-01-16 04:00:00.000000 UTC",  # utc-5
+            "2024-07-16 03:00:00.000000 UTC",  # utc-4
             "1_370",
             370,
             "1996-01-02",