def test_as_utc_timestamp_equality(self): # Arrange timestamp1 = datetime(1970, 1, 1, 0, 0, 0, 0) timestamp2 = UNIX_EPOCH timestamp3 = pd.Timestamp(1970, 1, 1, 0, 0, 0, 0) timestamp4 = pd.Timestamp(1970, 1, 1, 0, 0, 0, 0).tz_localize("UTC") # Act timestamp1_converted = as_utc_timestamp(timestamp1) timestamp2_converted = as_utc_timestamp(timestamp2) timestamp3_converted = as_utc_timestamp(timestamp3) timestamp4_converted = as_utc_timestamp(timestamp4) # Assert assert timestamp1_converted == timestamp2_converted assert timestamp2_converted == timestamp3_converted assert timestamp3_converted == timestamp4_converted
def test_as_utc_timestamp_given_tz_aware_pandas(self): # Arrange timestamp = pd.Timestamp(2013, 2, 1, 0, 0, 0, 0).tz_localize("UTC") # Act result = as_utc_timestamp(timestamp) # Assert assert result == pd.Timestamp("2013-02-01 00:00:00+00:00") assert result.tz == pytz.utc
def test_as_utc_timestamp_given_tz_aware_datetime(self): # Arrange timestamp = datetime(2013, 2, 1, 0, 0, 0, 0, tzinfo=pytz.utc) # Act result = as_utc_timestamp(timestamp) # Assert assert result == pd.Timestamp("2013-02-01 00:00:00+00:00") assert result.tz == pytz.utc
def test_as_utc_timestamp_given_tz_naive_pandas_timestamp(self): # Arrange timestamp = pd.Timestamp(2013, 2, 1, 0, 0, 0, 0) # Act result = as_utc_timestamp(timestamp) # Assert assert result == pd.Timestamp("2013-02-01 00:00:00+00:00") assert result.tz == pytz.utc
def test_as_utc_timestamp_given_tz_naive_pandas_timestamp(self): # Arrange timestamp = pd.Timestamp(2013, 2, 1, 0, 0, 0, 0) # Act result = as_utc_timestamp(timestamp) # Assert self.assertEqual(pd.Timestamp("2013-02-01 00:00:00+00:00"), result) self.assertEqual(pytz.utc, result.tz)
def test_as_utc_timestamp_given_tz_naive_datetime(self): # Arrange timestamp = datetime(2013, 2, 1, 0, 0, 0, 0) # Act result = as_utc_timestamp(timestamp) # Assert assert pd.Timestamp("2013-02-01 00:00:00+00:00") == result assert pytz.utc == result.tz