Fix tests

tswast · tswast · commit 7db8b1f91bb0 · 2019-04-03T15:47:51.000-07:00
diff --git a/tests/system/test_gbq.py b/tests/system/test_gbq.py
@@ -314,7 +314,8 @@ def test_should_properly_handle_timestamp_unix_epoch(self, project_id):
             {"unix_epoch": ["1970-01-01T00:00:00.000000Z"]},
             dtype="datetime64[ns]",
         )
-        expected["unix_epoch"] = expected["unix_epoch"].dt.tz_localize("UTC")
+        if expected["unix_epoch"].dt.tz is None:
+            expected["unix_epoch"] = expected["unix_epoch"].dt.tz_localize("UTC")
         tm.assert_frame_equal(df, expected)
 
     def test_should_properly_handle_arbitrary_timestamp(self, project_id):

Original file line number	Diff line number	Diff line change
`@@ -314,7 +314,8 @@ def test_should_properly_handle_timestamp_unix_epoch(self, project_id):`
`314`	`314`	`{"unix_epoch": ["1970-01-01T00:00:00.000000Z"]},`
`315`	`315`	`dtype="datetime64[ns]",`
`316`	`316`	`)`
`317`		`- expected["unix_epoch"] = expected["unix_epoch"].dt.tz_localize("UTC")`
	`317`	`+ if expected["unix_epoch"].dt.tz is None:`
	`318`	`+ expected["unix_epoch"] = expected["unix_epoch"].dt.tz_localize("UTC")`
`318`	`319`	`tm.assert_frame_equal(df, expected)`
`319`	`320`
`320`	`321`	`def test_should_properly_handle_arbitrary_timestamp(self, project_id):`