Browse Source

updated some tests

master
Gourav Kumar 9 months ago
parent
commit
2f50894b46
  1. 3
      pyfacts/pyfacts.py
  2. 1
      setup.py
  3. 2
      tests/test_pyfacts.py

3
pyfacts/pyfacts.py

@ -809,6 +809,7 @@ class TimeSeries(TimeSeriesCore):
to_frequency: Literal["W", "M", "Q", "H", "Y"],
method: Literal["sum", "mean"],
eomonth: bool = False,
ensure_coverage: bool = True,
anchor_date=Literal["start", "end"],
) -> TimeSeries:
"""Transform a time series object into a lower frequency object with an aggregation function.
@ -851,7 +852,7 @@ class TimeSeries(TimeSeriesCore):
self.start_date,
self.end_date, # + relativedelta(days=to_frequency.days),
to_frequency.symbol,
ensure_coverage=True,
ensure_coverage=ensure_coverage,
eomonth=eomonth,
)
# prev_date = dates[0]

1
setup.py

@ -2,6 +2,7 @@ from setuptools import find_packages, setup
license = open("LICENSE").read().strip()
setup(
name="pyfacts",
version="0.0.1",

2
tests/test_pyfacts.py

@ -374,7 +374,7 @@ class TestTransform:
def test_daily_to_weekly(self, create_test_data):
ts_data = create_test_data(AllFrequencies.D, num=782, skip_weekends=True)
ts = TimeSeries(ts_data, "D")
tst = ts.transform("W", "mean")
tst = ts.transform("W", "mean", ensure_coverage=False)
assert isinstance(tst, TimeSeries)
assert len(tst) == 157
assert "2017-01-30" in tst

Loading…
Cancel
Save