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
