mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-08 10:22:09 +00:00
b71abcfbf4
* quick econ fix * black * keys and feature flags * terminal name 👀 * some more replacements * some more replacements * edit pyproject * gst -> openbb * add example portfolios back to git * Update api from gst * sorry. skipping some tests * another round of names * another round of test edits * Missed some .gst refs and update timezone * water mark stuff * Fixing Names in terminal.spec and name of GTFF_DEFAULTS to OBBFF_DEFAULTS * fix more GST to OpenBB Terminal * Logging : merge conflicts with main * Revert wrong files Co-authored-by: Andrew <andrew.kenreich@gmail.com> Co-authored-by: DidierRLopes <dro.lopes@campus.fct.unl.pt> Co-authored-by: Chavithra PARANA <chavithra@gmail.com>
108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
# IMPORTATION THIRDPARTY
|
|
import pytest
|
|
|
|
# IMPORTATION INTERNAL
|
|
from openbb_terminal.econometrics import (
|
|
regression_model,
|
|
econometrics_model,
|
|
regression_view,
|
|
)
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
@pytest.mark.record_stdout
|
|
@pytest.mark.parametrize(
|
|
"regression_variables, data, datasets, show_regression, lags",
|
|
[
|
|
(
|
|
["TOTEMP-longley", "GNP-longley", "ARMED-longley", "POP-longley"],
|
|
{
|
|
"longley": econometrics_model.load(
|
|
"longley", ["csv", "xlsx"], {}, {"longley": "longley"}
|
|
)
|
|
},
|
|
{
|
|
"TOTEMP-longley": {"TOTEMP": None, "longley": None},
|
|
"GNP-longley": {"GNP": None, "longley": None},
|
|
"ARMED-longley": {"ARMED": None, "longley": None},
|
|
"POP-longley": {"POP": None, "longley": None},
|
|
},
|
|
False,
|
|
1,
|
|
),
|
|
(
|
|
["TOTEMP-longley", "GNP-longley", "ARMED-longley", "POP-longley"],
|
|
{
|
|
"longley": econometrics_model.load(
|
|
"longley", ["csv", "xlsx"], {}, {"longley": "longley"}
|
|
)
|
|
},
|
|
{
|
|
"TOTEMP-longley": {"TOTEMP": None, "longley": None},
|
|
"GNP-longley": {"GNP": None, "longley": None},
|
|
"ARMED-longley": {"ARMED": None, "longley": None},
|
|
"POP-longley": {"POP": None, "longley": None},
|
|
},
|
|
False,
|
|
2,
|
|
),
|
|
],
|
|
)
|
|
def test_display_bgod(regression_variables, data, datasets, show_regression, lags):
|
|
_, _, _, model = regression_model.get_ols(
|
|
regression_variables=regression_variables,
|
|
data=data,
|
|
datasets=datasets,
|
|
show_regression=show_regression,
|
|
)
|
|
|
|
regression_view.display_bgod(model=model, lags=lags)
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
@pytest.mark.record_stdout
|
|
@pytest.mark.parametrize(
|
|
"regression_variables, data, datasets, show_regression",
|
|
[
|
|
(
|
|
["TOTEMP-longley", "ARMED-longley", "POP-longley"],
|
|
{
|
|
"longley": econometrics_model.load(
|
|
"longley", ["csv", "xlsx"], {}, {"longley": "longley"}
|
|
)
|
|
},
|
|
{
|
|
"TOTEMP-longley": {"TOTEMP": None, "longley": None},
|
|
"GNP-longley": {"GNP": None, "longley": None},
|
|
"ARMED-longley": {"ARMED": None, "longley": None},
|
|
"POP-longley": {"POP": None, "longley": None},
|
|
},
|
|
False,
|
|
),
|
|
(
|
|
["GNP-longley", "ARMED-longley", "POP-longley"],
|
|
{
|
|
"longley": econometrics_model.load(
|
|
"longley", ["csv", "xlsx"], {}, {"longley": "longley"}
|
|
)
|
|
},
|
|
{
|
|
"TOTEMP-longley": {"TOTEMP": None, "longley": None},
|
|
"GNP-longley": {"GNP": None, "longley": None},
|
|
"ARMED-longley": {"ARMED": None, "longley": None},
|
|
"POP-longley": {"POP": None, "longley": None},
|
|
},
|
|
False,
|
|
),
|
|
],
|
|
)
|
|
def test_display_bpag(regression_variables, data, datasets, show_regression):
|
|
_, _, _, model = regression_model.get_ols(
|
|
regression_variables=regression_variables,
|
|
data=data,
|
|
datasets=datasets,
|
|
show_regression=show_regression,
|
|
)
|
|
|
|
regression_view.display_bpag(model=model)
|