Sweep: Add unit tests for SSAForecasterImplementation

[sweep-ai]

Purpose

This pull request adds unit tests for the SSAForecasterImplementation class located in the fedot_ind.core.models.ts_forecasting.ssa_forecaster module. The tests cover most edge cases and ensure the correct behavior of the forecaster.

Description

The changes in this pull request include:

1. Creation of a new test file test_ssa_forecaster.py in the tests/unit/core/models directory.
2. Implementation of several test cases to cover the initialization of the SSAForecasterImplementation class with default and custom parameters.
3. Addition of tests to verify the predict() and predict_for_fit() methods of the forecaster, including handling of simple and complex time series, as well as edge cases like empty input data and missing values.
4. Implementation of a test case to ensure the correct behavior of the fit() method.

Summary

• New test file: tests/unit/core/models/test_ssa_forecaster.py
• Tests for SSAForecasterImplementation class initialization with default and custom parameters
• Tests for predict() and predict_for_fit() methods, covering simple and complex time series, as well as edge cases
• Test for fit() method

Fixes #152.

---

Tip

To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

This is an automated message generated by Sweep AI.

[pep8speaks]

Thanks for update, @sweep-ai[bot]!

There are currently no PEP 8 issues detected in this Pull Request. Cheers! 🍻

Comment last updated at 2024-06-13 11:59:52 UTC

[sweep-ai]

Sweep: PR Review

Author: sweep-ai[bot]

Sweep found 1 new issue.

Sweep has left comments on the pull request for you to review.
You may respond to any comment Sweep made your feedback will be taken into consideration if you run the review again. If Sweep made a mistake, you can resolve the comment or let Sweep know by responding to the comment.

Potential Issues

Sweep is unsure if these are issues, but they might be worth checking out.

tests/unit/core/models/test_ssa_forecaster.py

• The test_fit method does not include any assertions to verify the correctness of the fitting process.

Fedot.Industrial/tests/unit/core/models/test_ssa_forecaster.py

Line 76 in d53c7ff

target=time_series,


View Diff

---

[sweep-ai]

The test_predict_for_fit_edge_cases method does not verify the correctness of the output when handling missing values in the time series.

[technocreep]

FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_fit - TypeError: init() missing 1 required positional argument: 'data_type'
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_init_custom_params - AssertionError: 100 != 50
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_init_default_params - AttributeError: 'NoneType' object has no attribute 'get'
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_predict_complex_ts - TypeError: init() missing 1 required positional argument: 'data_type'
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_predict_edge_cases - TypeError: init() missing 1 required positional argument: 'data_type'
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_predict_for_fit_complex_ts - TypeError: init() missing 1 required positional argument: 'data_type'
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_predict_for_fit_edge_cases - TypeError: init() missing 1 required positional argument: 'data_type'
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_predict_for_fit_simple_ts - TypeError: init() missing 1 required positional argument: 'data_type'
FAILED tests/unit/core/models/test_ssa_forecaster.py::TestSSAForecasterImplementation::test_predict_simple_ts - TypeError: init() missing 1 required positional argument: 'data_type'