diff --git a/tests/test_pyfacts.py b/tests/test_pyfacts.py index aab530a..2d04b36 100644 --- a/tests/test_pyfacts.py +++ b/tests/test_pyfacts.py @@ -30,7 +30,7 @@ class TestDateSeries: def test_monthly(self): start_date = datetime.datetime(2020, 1, 1) end_date = datetime.datetime(2020, 12, 31) - d = create_date_series(start_date, end_date, frequency="M") + d = create_date_series(start_date, end_date, frequency="M", ensure_coverage=False) assert len(d) == 12 d = create_date_series(start_date, end_date, frequency="M", eomonth=True) @@ -327,7 +327,7 @@ class TestExpand: ts_data = create_test_data(AllFrequencies.M, num=6) ts = TimeSeries(ts_data, "M") expanded_ts = ts.expand("W", "ffill") - assert len(expanded_ts) == 22 + assert len(expanded_ts) == 23 assert expanded_ts.frequency.name == "weekly" assert expanded_ts.iloc[0][1] == expanded_ts.iloc[1][1] @@ -346,7 +346,7 @@ class TestShrink: ts = TimeSeries(ts_data, "D") shrunk_ts_w = ts.shrink("W", "ffill") shrunk_ts_m = ts.shrink("M", "ffill") - assert len(shrunk_ts_w) == 143 + assert len(shrunk_ts_w) == 144 assert len(shrunk_ts_m) == 34 def test_weekly_to_smaller(self, create_test_data): @@ -376,7 +376,7 @@ class TestTransform: ts = TimeSeries(ts_data, "D") tst = ts.transform("W", "mean") assert isinstance(tst, TimeSeries) - assert len(tst) == 156 + assert len(tst) == 157 assert "2017-01-30" in tst assert tst.iloc[4] == (datetime.datetime(2017, 1, 30), 1021.19)