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