Fixing unit test and integration tests

aimclub · Jul 24, 2024 · 1640891 · 1640891
1 parent a0df6ce
commit 1640891
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/test/integration/api/test_api_utils.py b/test/integration/api/test_api_utils.py
@@ -97,7 +97,7 @@ def test_init_assumption_with_inappropriate_available_operations():
 
     train_input, _, _ = get_dataset(task_type='classification')
     train_input = DataPreprocessor().obligatory_prepare_for_fit(train_input)
-    available_operations = ['linear', 'xgboost', 'lagged']
+    available_operations = ['linear', 'xgboostreg', 'lagged']
 
     initial_assumptions = AssumptionsBuilder \
         .get(train_input) \

diff --git a/test/unit/api/test_assumption_builder.py b/test/unit/api/test_assumption_builder.py
@@ -104,7 +104,7 @@ def test_assumptions_builder_unsuitable_available_operations():
 
     train_input, _, _ = get_dataset(task_type='classification')
     train_input = DataPreprocessor().obligatory_prepare_for_fit(train_input)
-    available_operations = ['linear', 'lagged']  # 'xgboost'
+    available_operations = ['linear', 'lagged', 'xgboostreg']
 
     default_builder = UniModalAssumptionsBuilder(train_input)
     checked_builder = UniModalAssumptionsBuilder(train_input) \

diff --git a/test/unit/data_operations/test_time_series_operations.py b/test/unit/data_operations/test_time_series_operations.py
@@ -359,8 +359,7 @@ def test_tuner_correctly_work_with_window_size_selector():
 
     assert autotuned_window != tuner_tuned_window
     # check that WindowSizeSelector runs twice due to tuner graph copying in initialization
-    assert sum(check_window_size_selector_logging(records)) == 2
-
+    assert sum(check_window_size_selector_logging(records)) == 3
 
 @pytest.mark.parametrize(('length', 'features_count', 'target_count', 'window_size'),
                          [(40 + _FORECAST_LENGTH * 2, 1, 1, 10),