Skip to main content

Browse the archive

Revision afd2fac5a595b966a2722cfbc9d49948b07d2948 authored by Aaron Smith on 03 November 2022, 13:26:31 UTC, committed by GitHub on 03 November 2022, 13:26:31 UTC 
Python 3.11 Support (#524)
 
* First pass at Python 3.11 support

* Update 'whats new'

* Update build_requirements for different python versions

* Fix docs

* Empty commit for CI

* Use lowest supported numpy (==) for building instead of max compatible (~=)

* Make min numpy 1.21.3

* Use lowest supported version of scipy when building

* More scipy hackery

* https://github.com/scipy/scipy/blob/26a77da3a4ca126a943a331c1aa6ef3915b0d501/pyproject.toml#L41

* More scipy hackery

* Typos

* More typos
1 parent f999f87
Raw File
test_estimators.py 
# -*- coding: utf-8 -*-

from sklearn.base import clone
from pmdarima.arima import ARIMA, AutoARIMA
from pmdarima.pipeline import Pipeline
from pmdarima.datasets import load_wineind
from pmdarima.preprocessing import FourierFeaturizer
import pytest

y = load_wineind()


@pytest.mark.parametrize(
    'est', [
        ARIMA(order=(2, 1, 1)),
        AutoARIMA(seasonal=False, maxiter=3),
        Pipeline([
            ("fourier", FourierFeaturizer(m=12)),
            ("arima", AutoARIMA(seasonal=False, stepwise=True,
                                suppress_warnings=True, d=1, max_p=2, max_q=0,
                                start_q=0, start_p=1,
                                maxiter=3, error_action='ignore'))
        ])
    ]
)
def test_clonable(est):
    # fit it, then clone it
    est.fit(y)
    est2 = clone(est)
    assert isinstance(est2, est.__class__)
    assert est is not est2
back to top