def test__make_aggregate_index_localization(start, end): length = '30min' label = 'ending' test_data = { 0: pd.DataFrame(range(1), index=pd.DatetimeIndex([start])), 1: pd.DataFrame(range(1), index=pd.DatetimeIndex([end])), } with pytest.raises(TypeError): utils._make_aggregate_index(test_data, length, label, 'UTC')
def test__make_aggregate_index_instant(): length = '30min' label = 'instant' test_data = { 0: pd.DataFrame(range(6), index=pd.date_range( '20190101T0100Z', freq='10min', periods=6)) # end 32 } with pytest.raises(ValueError): utils._make_aggregate_index(test_data, length, label, 'UTC')
def test__make_aggregate_index_invalid_length(): length = '33min' label = 'beginning' test_data = { 0: pd.DataFrame(range(6), index=pd.date_range( '20190101T0158Z', freq='7min', periods=6)) # end 32 } with pytest.raises(ValueError): utils._make_aggregate_index(test_data, length, label, 'UTC')
def test__make_aggregate_index_offset_left(length, label, expected): test_data = { 0: pd.DataFrame(range(6), index=pd.date_range( '20191004T0658Z', freq='7min', periods=6)) # end 32 } out = utils._make_aggregate_index(test_data, length, label, 'UTC') pdt.assert_index_equal(out, expected)
def test__make_aggregate_index(length, label, expected): test_data = { 0: pd.DataFrame(range(5), index=pd.date_range( '20191004T0700Z', freq='7min', periods=5)), # end 35 1: pd.DataFrame(range(4), index=pd.date_range( '20191004T0015-0700', freq='10min', periods=4))} # end 45 out = utils._make_aggregate_index(test_data, length, label, 'UTC') pdt.assert_index_equal(out, expected)
def test__make_aggregate_index_tz(): length = '30min' label = 'beginning' test_data = { 0: pd.DataFrame(range(6), index=pd.date_range( '20190101T1600Z', freq='5min', periods=6)) # end 30 } expected = pd.DatetimeIndex(['20190101T0900'], tz='America/Denver') out = utils._make_aggregate_index(test_data, length, label, 'America/Denver') pdt.assert_index_equal(out, expected)