Files
OpenBB/tests/openbb_terminal/forecast/test_nhits_model.py
Colin Delahunty 0c9a4fafc4 Nhits (#2779)
* Added nhits, updated docs

* Added nhits docs

* Updated docs image

* Added funot docs

* Fixed martin changes

* Added nhits tests

* Fixed hugo issues

Co-authored-by: martinb-bb <105685594+martinb-bb@users.noreply.github.com>
2022-10-11 12:03:28 -04:00

18 lines
458 B
Python

import pytest
from tests.openbb_terminal.forecast import conftest
try:
from openbb_terminal.forecast import nhits_model
except ImportError:
pytest.skip(allow_module_level=True)
def test_get_nhits_model(tsla_csv):
conftest.test_model(nhits_model.get_nhits_data, tsla_csv, n_epochs=1)
def test_get_nhits_model_past_covs(tsla_csv):
conftest.test_model(
nhits_model.get_nhits_data, tsla_csv, n_epochs=1, past_covariates="open"
)