From aba23d0a144be8a60114d176a16a93771bf58e06 Mon Sep 17 00:00:00 2001 From: rhoboro Date: Tue, 13 Mar 2018 15:09:11 +0900 Subject: [PATCH] add a test for extact timestamp --- pandas_gbq/tests/test_gbq.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/pandas_gbq/tests/test_gbq.py b/pandas_gbq/tests/test_gbq.py index eb252253..a7ac8339 100644 --- a/pandas_gbq/tests/test_gbq.py +++ b/pandas_gbq/tests/test_gbq.py @@ -1389,6 +1389,31 @@ def test_upload_data_with_missing_schema_fields_raises_error(self): private_key=self.credentials, table_schema=test_schema) + def test_upload_data_with_timestamp(self): + df = tm.makeMixedDataFrame() + test_id = "21" + test_size = 6 + df = DataFrame(np.random.randn(test_size, 4), index=range(test_size), + columns=list('ABCD')) + test_timestamp = datetime.now(pytz.timezone('UTC')) + df['times'] = test_timestamp + + gbq.to_gbq( + df, self.destination_table + test_id, + _get_project_id(), + private_key=self.credentials) + + result_df = gbq.read_gbq("SELECT * FROM {0}".format( + self.destination_table + test_id), + project_id=_get_project_id(), + private_key=self.credentials) + + assert len(result_df) == test_size + + expected = df['times'].astype('M8[ns]').sort_values() + result = result_df['times'].astype('M8[ns]').sort_values() + tm.assert_numpy_array_equal(expected.values, result.values) + def test_list_dataset(self): dataset_id = self.dataset_prefix + "1" assert dataset_id in self.dataset.datasets()