try again

googleapis · Aug 8, 2023 · b009a68 · b009a68
1 parent b60ce30
commit b009a68
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 11 deletions.
diff --git a/tests/system/test_gbq.py b/tests/system/test_gbq.py
@@ -1014,7 +1014,7 @@ def test_upload_data_with_timestamp(self, project_id):
                 "2018-07-13T05:40:45.348318",
                 "2018-08-13T05:40:45.348318",
             ],
-            dtype="datetime64[us]",
+            dtype="datetime64[ns]",
         ).dt.tz_localize("UTC")
 
         gbq.to_gbq(
@@ -1027,7 +1027,6 @@ def test_upload_data_with_timestamp(self, project_id):
         result_df = gbq.read_gbq(
             "SELECT * FROM {0}".format(self.destination_table + test_id),
             project_id=project_id,
-            dtypes={"times": "datetime64[us]"},
             credentials=self.credentials,
             dialect="legacy",
         )

diff --git a/tests/system/test_read_gbq.py b/tests/system/test_read_gbq.py
@@ -149,7 +149,7 @@ def writable_table(
                                 "2011-10-01 00:01:02.345678",
                                 "2018-04-11 23:59:59.999999",
                             ],
-                            dtype="datetime64[us]",
+                            dtype="datetime64[ns]",
                         ),
                         "float_col": [1.125, -2.375, 0.0],
                         "int64_col": pandas.Series(
@@ -175,7 +175,7 @@ def writable_table(
                                 "2011-10-01 00:01:02.345678",
                                 "2018-04-11 23:59:59.999999",
                             ],
-                            dtype="datetime64[us]",
+                            dtype="datetime64[ns]",
                         ).dt.tz_localize(datetime.timezone.utc),
                     }
                 ),
@@ -283,7 +283,7 @@ def writable_table(
                                 "2011-10-01 00:01:02.345678",
                                 None,
                             ],
-                            dtype="datetime64[us]",
+                            dtype="datetime64[ns]",
                         ),
                         "float_col": [None, -2.375, 0.0],
                         "int64_col": pandas.Series([-1, None, 0], dtype="Int64"),
@@ -303,7 +303,7 @@ def writable_table(
                                 None,
                                 "2018-04-11 23:59:59.999999",
                             ],
-                            dtype="datetime64[us]",
+                            dtype="datetime64[ns]",
                         ).dt.tz_localize(datetime.timezone.utc),
                     }
                 ),
@@ -383,7 +383,7 @@ def writable_table(
                         ),
                         "datetime_col": pandas.Series(
                             [None],
-                            dtype="datetime64[us]",
+                            dtype="datetime64[ns]",
                         ),
                         "float_col": pandas.Series([None], dtype="float64"),
                         "int64_col": pandas.Series([None], dtype="Int64"),
@@ -395,7 +395,7 @@ def writable_table(
                         ),
                         "timestamp_col": pandas.Series(
                             [None],
-                            dtype="datetime64[us]",
+                            dtype="datetime64[ns]",
                         ).dt.tz_localize(datetime.timezone.utc),
                     }
                 ),
@@ -552,9 +552,6 @@ def test_default_dtypes(
     result = read_gbq(
         query,
         use_bqstorage_api=use_bqstorage_api,
-        dtypes={"datetime_col": "datetime64[us]",
-                "timestamp_col": "datetime64[us, UTC]",
-        },
         configuration={"query": {"useQueryCache": False}},
     )
     pandas.testing.assert_frame_equal(result, expected)