Skip to content

Commit

Permalink
test: added test for strip=True
Browse files Browse the repository at this point in the history
  • Loading branch information
madtoinou committed Jul 17, 2023
1 parent 6b055d6 commit c8ce8bd
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions darts/tests/dataprocessing/transformers/test_midas.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,6 +109,23 @@ def test_not_complete_monthly_to_quarterly(self):
"back into into a monthly TimeSeries with missing values.",
)

# verify that the result is identical when strip=True
midas = MIDAS(low_freq="QS", strip=True)
quarterly_not_complete_ts_midas = midas.fit_transform(
self.monthly_not_complete_ts
)
self.assertEqual(
quarterly_not_complete_ts_midas,
self.quarterly_not_complete_ts,
)
inversed_quarterly_not_complete_ts_midas = midas.inverse_transform(
quarterly_not_complete_ts_midas
)
self.assertEqual(
self.monthly_not_complete_ts,
inversed_quarterly_not_complete_ts_midas,
)

def test_multivariate_monthly_to_quarterly(self):
"""
Check that multivariate monthly to quarterly is properly transformed
Expand Down

0 comments on commit c8ce8bd

Please sign in to comment.