added tests

tinkoff-ai · Jun 13, 2023 · 7e8834c · 7e8834c
1 parent 3222cf9
commit 7e8834c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_commands/test_utils.py b/tests/test_commands/test_utils.py
@@ -2,7 +2,9 @@
 
 import pytest
 
+from etna.commands.backtest_command import ADDITIONAL_BACKTEST_PARAMETERS
 from etna.commands.forecast_command import ADDITIONAL_FORECAST_PARAMETERS
+from etna.commands.forecast_command import ADDITIONAL_PIPELINE_PARAMETERS
 from etna.commands.utils import _estimate_n_folds
 from etna.commands.utils import _max_n_folds_backtest
 from etna.commands.utils import _max_n_folds_forecast
@@ -277,6 +279,16 @@ def test_estimate_max_n_folds_backtest_with_transforms(
             ADDITIONAL_FORECAST_PARAMETERS,
             {"prediction_interval": True},
         ),
+        (
+            {"n_folds": 2, "n_jobs": 4, "estimate_n_folds": True},
+            ADDITIONAL_BACKTEST_PARAMETERS,
+            {"n_folds": 2, "n_jobs": 4},
+        ),
+        (
+            {"_target_": "etna.pipeline.Pipeline", "horizon": 4, "context_size": 1},
+            ADDITIONAL_PIPELINE_PARAMETERS,
+            {"_target_": "etna.pipeline.Pipeline", "horizon": 4},
+        ),
     ),
 )
 def test_remove_params(params, to_remove, expected):