Skip to content

Commit

Permalink
Added fixture
Browse files Browse the repository at this point in the history
  • Loading branch information
christopherbunn committed Aug 7, 2023
1 parent 459528c commit b765260
Showing 1 changed file with 2 additions and 20 deletions.
22 changes: 2 additions & 20 deletions evalml/tests/component_tests/test_varmax_regressor.py
Original file line number Diff line number Diff line change
Expand Up @@ -181,30 +181,12 @@ def test_fit_predict_sk_failure(
def test_different_time_units_out_of_sample(

Check warning on line 181 in evalml/tests/component_tests/test_varmax_regressor.py

View check run for this annotation

Codecov / codecov/patch

evalml/tests/component_tests/test_varmax_regressor.py#L179-L181

Added lines #L179 - L181 were not covered by tests
freq_str,
freq_num,
multiseries_ts_data_unstacked,
ts_multiseries_data,
):
from sktime.forecasting.base import ForecastingHorizon
from sktime.forecasting.varmax import VARMAX

Check warning on line 187 in evalml/tests/component_tests/test_varmax_regressor.py

View check run for this annotation

Codecov / codecov/patch

evalml/tests/component_tests/test_varmax_regressor.py#L186-L187

Added lines #L186 - L187 were not covered by tests

datetime_ = pd.date_range("1/1/1870", periods=20, freq=freq_num + freq_str)

X = pd.DataFrame(
range(20),
index=datetime_,
)
y = pd.DataFrame(
{
"target_0": pd.Series(
np.sin(np.linspace(-8 * np.pi, 8 * np.pi, 20)),
index=datetime_,
),
"target_1": pd.Series(
np.sin(np.linspace(-4 * np.pi, 4 * np.pi, 20)),
index=datetime_,
),
},
)

X, _, y = ts_multiseries_data(freq=freq_num + freq_str, datetime_feature=False)
fh_ = ForecastingHorizon([i + 1 for i in range(len(y[15:]))], is_relative=True)

Check warning on line 190 in evalml/tests/component_tests/test_varmax_regressor.py

View check run for this annotation

Codecov / codecov/patch

evalml/tests/component_tests/test_varmax_regressor.py#L189-L190

Added lines #L189 - L190 were not covered by tests

a_clf = VARMAX(maxiter=10)
Expand Down

0 comments on commit b765260

Please sign in to comment.