updated inference tests

etna-team · Jul 25, 2024 · a3b90ac · a3b90ac
1 parent 915afe6
commit a3b90ac
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/tests/test_transforms/test_inference/test_inverse_transform.py b/tests/test_transforms/test_inference/test_inverse_transform.py
@@ -23,6 +23,7 @@
 from etna.transforms import EventTransform
 from etna.transforms import ExogShiftTransform
 from etna.transforms import FilterFeaturesTransform
+from etna.transforms import FourierDecomposeTransform
 from etna.transforms import FourierTransform
 from etna.transforms import GaleShapleyFeatureSelectionTransform
 from etna.transforms import HolidayTransform
@@ -141,6 +142,7 @@ def _test_inverse_transform_train(self, ts, transform, expected_changes):
                 "regular_ts",
                 {},
             ),
+            (FourierDecomposeTransform(in_column="target", k=5, residuals=True), "regular_ts", {}),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -602,6 +604,7 @@ def test_inverse_transform_train_fail_resample(self, transform, dataset_name, ex
                 "regular_ts",
                 {},
             ),
+            (FourierDecomposeTransform(in_column="target", k=5, residuals=True), "regular_ts", {}),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -1093,6 +1096,7 @@ def _test_inverse_transform_train_subset_segments(self, ts, transform, segments)
                 ),
                 "regular_ts",
             ),
+            (FourierDecomposeTransform(in_column="target", k=5, residuals=True), "regular_ts"),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -1378,6 +1382,8 @@ def _test_inverse_transform_future_subset_segments(self, ts, transform, segments
                 ),
                 "regular_ts",
             ),
+            (FourierDecomposeTransform(in_column="target", k=5, residuals=True), "regular_ts"),
+            (FourierDecomposeTransform(in_column="positive", k=5, residuals=True), "ts_with_exog"),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -2589,6 +2595,7 @@ def _test_inverse_transform_future_with_target(
                 "regular_ts",
                 {},
             ),
+            # (FourierDecomposeTransform(in_column="target", k=5, residuals=True), "regular_ts", {}),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -3096,6 +3103,8 @@ def _test_inverse_transform_future_without_target(
                 "regular_ts",
                 {},
             ),
+            (FourierDecomposeTransform(in_column="target", k=5, residuals=True), "regular_ts", {}),
+            (FourierDecomposeTransform(in_column="positive", k=5, residuals=True), "ts_with_exog", {}),
             # embeddings
             (
                 EmbeddingSegmentTransform(

diff --git a/tests/test_transforms/test_inference/test_transform.py b/tests/test_transforms/test_inference/test_transform.py
@@ -23,6 +23,7 @@
 from etna.transforms import EventTransform
 from etna.transforms import ExogShiftTransform
 from etna.transforms import FilterFeaturesTransform
+from etna.transforms import FourierDecomposeTransform
 from etna.transforms import FourierTransform
 from etna.transforms import GaleShapleyFeatureSelectionTransform
 from etna.transforms import HolidayTransform
@@ -127,6 +128,11 @@ def _test_transform_train(self, ts, transform, expected_changes):
                 "regular_ts",
                 {"create": {"res"}},
             ),
+            (
+                FourierDecomposeTransform(in_column="target", k=2, residuals=True),
+                "regular_ts",
+                {"create": {"target_dft_0", "target_dft_1", "target_dft_residuals"}},
+            ),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -544,6 +550,11 @@ def test_transform_train_datetime_timestamp(self, transform, dataset_name, expec
                 "regular_ts",
                 {"create": {"res"}},
             ),
+            (
+                FourierDecomposeTransform(in_column="target", k=2, residuals=True),
+                "regular_ts",
+                {"create": {"target_dft_0", "target_dft_1", "target_dft_residuals"}},
+            ),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -1049,6 +1060,7 @@ def _test_transform_train_subset_segments(self, ts, transform, segments):
                 ),
                 "regular_ts",
             ),
+            (FourierDecomposeTransform(in_column="target", k=2, residuals=True), "regular_ts"),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -1316,6 +1328,8 @@ def _test_transform_future_subset_segments(self, ts, transform, segments, horizo
                 ),
                 "regular_ts",
             ),
+            (FourierDecomposeTransform(in_column="target", k=2, residuals=True), "regular_ts"),
+            (FourierDecomposeTransform(in_column="positive", k=2, residuals=True), "ts_with_exog"),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -2420,6 +2434,7 @@ def _test_transform_future_with_target(self, ts, transform, expected_changes, ga
                 "regular_ts",
                 {"create": {"res"}},
             ),
+            # (FourierDecomposeTransform(in_column="target", k=2, residuals=True), "regular_ts",             {"create": {"target_dft_0", "target_dft_1", "target_dft_residuals"}}),
             # embeddings
             (
                 EmbeddingSegmentTransform(
@@ -2859,6 +2874,16 @@ def _test_transform_future_without_target(self, ts, transform, expected_changes,
                 "regular_ts",
                 {"create": {"res"}},
             ),
+            (
+                FourierDecomposeTransform(in_column="target", k=2, residuals=True),
+                "regular_ts",
+                {"create": {"target_dft_0", "target_dft_1", "target_dft_residuals"}},
+            ),
+            (
+                FourierDecomposeTransform(in_column="positive", k=2, residuals=True),
+                "ts_with_exog",
+                {"create": {"positive_dft_0", "positive_dft_1", "positive_dft_residuals"}},
+            ),
             # embeddings
             (
                 EmbeddingSegmentTransform(