def test_with_utc_index_given_tz_aware_different_timezone_dataframe(self):
# Arrange
data1 = pd.DataFrame(
{"timestamp": ["2019-05-21 12:00:00", "2019-05-21 12:15:00"]}
)
data1.set_index("timestamp")
data1.index = pd.to_datetime(data1.index)
data2 = pd.DataFrame(
{
"timestamp": [
datetime(1970, 1, 1, 0, 0, 0, 0),
datetime(1970, 1, 1, 0, 0, 0, 0),
]
}
)
data2.set_index("timestamp")
data2.index = pd.to_datetime(data2.index, utc=True)
# Act
result1 = as_utc_index(data1)
result2 = as_utc_index(data2)
# Assert
assert result1.index[0] == result2.index[0]
assert result1.index.tz == result2.index.tz
def test_as_utc_index_given_empty_dataframe_returns_empty_dataframe(self):
# Arrange
data = pd.DataFrame()
# Act
result = as_utc_index(data)
# Assert
assert result.empty
def test_with_utc_index_given_tz_aware_dataframe(self):
# Arrange
data = pd.DataFrame({"timestamp": ["2019-05-21T12:00:00+00:00",
"2019-05-21T12:15:00+00:00"]})
data.set_index("timestamp")
data.index = pd.to_datetime(data.index, utc=True)
# Act
result = as_utc_index(data)
# Assert
self.assertEqual(pytz.utc, result.index.tz)
def setup(self):
# Fixture Setup
news_csv_path = os.path.join(PACKAGE_ROOT, "data", "news_events.csv")
self.news_data = as_utc_index(
pd.read_csv(news_csv_path, parse_dates=True, index_col=0))
