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))