def test_invalid_year_frequency():
time_column_name = "date"
frequency = "2Y"
df = pd.DataFrame({
time_column_name:
["2014-01-01", "2016-01-01", "2018-01-01", "2020-01-01"]
})
df[time_column_name] = pd.to_datetime(
df[time_column_name]).dt.tz_localize(tz=None)
with pytest.raises(ValueError):
assert_time_column_valid(df, time_column_name, frequency)
def test_valid_year_frequency():
time_column_name = "date"
frequency = "1Y"
df = pd.DataFrame({
time_column_name:
["2017-12-31", "2018-12-31", "2019-12-31", "2020-12-31"]
})
df[time_column_name] = pd.to_datetime(
df[time_column_name]).dt.tz_localize(tz=None)
try:
assert_time_column_valid(df, time_column_name, frequency)
except ValueError as e:
pytest.fail("Unexpected ValueError: {}".format(e))
def test_invalid_month_frequency():
time_column_name = "date"
frequency = "6M"
df = pd.DataFrame({
time_column_name: [
"2018-01-31 00:00:00",
"2018-07-31 12:30:00",
"2019-01-30 00:00:00",
"2019-07-31 00:00:00",
]
})
df[time_column_name] = pd.to_datetime(
df[time_column_name]).dt.tz_localize(tz=None)
with pytest.raises(ValueError):
assert_time_column_valid(df, time_column_name, frequency)
def test_invalid_minute_frequency():
time_column_name = "date"
frequency = "10min"
df = pd.DataFrame({
time_column_name: [
"2020-01-31 06:30:30",
"2020-01-31 06:40:31",
"2020-01-31 06:50:30",
"2020-01-31 07:00:30",
]
})
df[time_column_name] = pd.to_datetime(
df[time_column_name]).dt.tz_localize(tz=None)
with pytest.raises(ValueError):
assert_time_column_valid(df, time_column_name, frequency)
def test_valid_minute_frequency():
time_column_name = "date"
frequency = "10min"
df = pd.DataFrame({
time_column_name: [
"2020-01-31 06:30:30",
"2020-01-31 06:40:30",
"2020-01-31 06:50:30",
"2020-01-31 07:00:30",
]
})
df[time_column_name] = pd.to_datetime(
df[time_column_name]).dt.tz_localize(tz=None)
try:
assert_time_column_valid(df, time_column_name, frequency)
except ValueError as e:
pytest.fail("Unexpected ValueError: {}".format(e))
def test_valid_month_frequency():
time_column_name = "date"
frequency = "2M"
df = pd.DataFrame({
time_column_name: [
"2020-02-29 00:00:00",
"2020-04-30 00:00:00",
"2020-06-30 00:00:00",
"2020-08-31 00:00:00",
]
})
df[time_column_name] = pd.to_datetime(
df[time_column_name]).dt.tz_localize(tz=None)
try:
assert_time_column_valid(df, time_column_name, frequency)
except ValueError as e:
pytest.fail("Unexpected ValueError: {}".format(e))
def test_valid_business_day_frequency():
time_column_name = "date"
frequency = "2B"
df = pd.DataFrame({
time_column_name: [
"2020-01-01 06:30:00",
"2020-01-03 06:30:00",
"2020-01-07 06:30:00",
"2020-01-09 06:30:00",
]
})
df[time_column_name] = pd.to_datetime(
df[time_column_name]).dt.tz_localize(tz=None)
try:
assert_time_column_valid(df, time_column_name, frequency)
except ValueError as e:
pytest.fail("Unexpected ValueError: {}".format(e))
