mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-01 15:01:18 +00:00
8df31f8de4
* add USD Liquidity Index to Economy menu * mypy * add screenshots to Terminal Economy Intro * add test to test_fred_model * add Broad US Dollar Index to list of comparables * docs pages update * fix test_nasdaq_model * ask obb indexing * return just the calculated series by default * a grid for Squares * raw table output --------- Co-authored-by: James Maslek <jmaslek11@gmail.com> Co-authored-by: martinb-bb <105685594+martinb-bb@users.noreply.github.com>
133 lines
3.3 KiB
Python
133 lines
3.3 KiB
Python
# IMPORTATION STANDARD
|
|
|
|
# IMPORTATION THIRDPARTY
|
|
import pandas as pd
|
|
import pytest
|
|
|
|
# IMPORTATION INTERNAL
|
|
from openbb_terminal.economy import fred_model
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_headers": [("User-Agent", None)],
|
|
"filter_query_parameters": [
|
|
("api_key", "MOCK_API_KEY"),
|
|
],
|
|
}
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.parametrize(
|
|
"func, kwargs_dict",
|
|
[("check_series_id", {"series_id": "DGS5"})],
|
|
)
|
|
def test_output(func, kwargs_dict, recorder):
|
|
payload = getattr(fred_model, func)(**kwargs_dict)
|
|
assert len(payload) > 0
|
|
assert "seriess" in payload
|
|
recorder.capture(payload["seriess"])
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.parametrize(
|
|
"func, kwargs_dict",
|
|
[
|
|
("get_series_notes", {"search_query": "finance"}),
|
|
("get_series_data", {"series_id": "DGS10", "start_date": "2020-01-01"}),
|
|
],
|
|
)
|
|
def test_check_output(func, kwargs_dict, recorder):
|
|
result_df = getattr(fred_model, func)(**kwargs_dict)
|
|
|
|
assert not result_df.empty
|
|
recorder.capture(result_df)
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.parametrize(
|
|
"func, kwargs_dict",
|
|
[
|
|
("check_series_id", {"series_id": "DGS5"}),
|
|
],
|
|
)
|
|
def test_invalid_response_status(func, kwargs_dict, mocker):
|
|
# MOCK GET
|
|
attrs = {
|
|
"status_code": 400,
|
|
"json.return_value": {"error_message": "mock error message"},
|
|
}
|
|
|
|
mock_response = mocker.Mock(**attrs)
|
|
|
|
mocker.patch(
|
|
target="openbb_terminal.helper_funcs.requests.get",
|
|
new=mocker.Mock(return_value=mock_response),
|
|
)
|
|
|
|
payload = getattr(fred_model, func)(**kwargs_dict)
|
|
assert len(payload) == 0
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.parametrize(
|
|
"func, kwargs_dict",
|
|
[("get_series_data", {"series_id": "DGS10", "start_date": "2020-01-01"})],
|
|
)
|
|
def test_load_data(func, kwargs_dict, recorder):
|
|
result_df = getattr(fred_model, func)(**kwargs_dict)
|
|
|
|
assert not result_df.empty
|
|
recorder.capture(result_df)
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.parametrize(
|
|
"func, kwargs_dict",
|
|
[
|
|
("get_cpi", {"countries": ["united kingdom"]}),
|
|
("get_cpi", {"countries": ["united kingdom", "united_states"]}),
|
|
("get_cpi", {"countries": ["united kingdom", "united_states", "belgium"]}),
|
|
(
|
|
"get_cpi",
|
|
{"countries": ["united kingdom", "united_states"], "frequency": "monthly"},
|
|
),
|
|
(
|
|
"get_cpi",
|
|
{
|
|
"countries": ["united kingdom", "united_states"],
|
|
"units": "growth_previous",
|
|
},
|
|
),
|
|
(
|
|
"get_cpi",
|
|
{"countries": ["united kingdom", "united_states"], "harmonized": True},
|
|
),
|
|
(
|
|
"get_cpi",
|
|
{
|
|
"countries": ["united kingdom", "united_states"],
|
|
"units": "growth_previous",
|
|
},
|
|
),
|
|
],
|
|
)
|
|
def test_get_cpi(func, kwargs_dict, recorder):
|
|
result_df = getattr(fred_model, func)(**kwargs_dict)
|
|
|
|
assert isinstance(result_df, pd.DataFrame)
|
|
recorder.capture(result_df)
|
|
|
|
|
|
@pytest.mark.vcr
|
|
def test_EQUITY_INDICES(recorder):
|
|
assert isinstance(fred_model.EQUITY_INDICES, dict)
|
|
recorder.capture(fred_model.EQUITY_INDICES)
|
|
|
|
|
|
@pytest.mark.vcr
|
|
@pytest.mark.record_stdout
|
|
def test_get_usd_liquidity_BAD_SYMBOL():
|
|
fred_model.get_usd_liquidity("BAD_SYMBOL")
|