diff options
Diffstat (limited to 'openbb_platform/extensions/economy/integration/test_economy_python.py')
-rw-r--r-- | openbb_platform/extensions/economy/integration/test_economy_python.py | 173 |
1 files changed, 169 insertions, 4 deletions
diff --git a/openbb_platform/extensions/economy/integration/test_economy_python.py b/openbb_platform/extensions/economy/integration/test_economy_python.py index 75a01123c7b..eae984cd3e1 100644 --- a/openbb_platform/extensions/economy/integration/test_economy_python.py +++ b/openbb_platform/extensions/economy/integration/test_economy_python.py @@ -65,12 +65,36 @@ def test_economy_calendar(params, obb): [ ( { + "country": "spain", + "transform": "yoy", + "frequency": "annual", + "harmonized": False, + "start_date": "2020-01-01", + "end_date": "2023-06-06", + "provider": "fred", + } + ), + ( + { "country": "portugal,spain", - "units": "growth_same", + "transform": "period", "frequency": "monthly", "harmonized": True, "start_date": "2023-01-01", "end_date": "2023-06-06", + "provider": "fred", + } + ), + ( + { + "country": "portugal,spain", + "transform": "yoy", + "frequency": "quarter", + "harmonized": False, + "start_date": "2020-01-01", + "end_date": "2023-06-06", + "provider": "oecd", + "expenditure": "transport", } ), ], @@ -376,7 +400,6 @@ def test_economy_money_measures(params, obb): @parametrize( "params", [ - ({"start_date": "2023-01-01", "end_date": "2023-06-06"}), ( { "country": "united_states", @@ -508,8 +531,8 @@ def test_economy_long_term_interest_rate(params, obb): "units": None, "region_type": None, "season": None, - "aggregation_method": None, - "transform": None, + "aggregation_method": "avg", + "transform": "lin", "provider": "fred", "limit": None, } @@ -606,3 +629,145 @@ def test_economy_indicators(params, obb): assert result assert isinstance(result, OBBject) assert len(result.results) > 0 + + +@parametrize( + "params", + [ + ( + { + "date": None, + "provider": "federal_reserve", + "holding_type": "all_treasury", + "summary": False, + "monthly": False, + "cusip": None, + "wam": False, + } + ), + ( + { + "date": None, + "provider": "federal_reserve", + "holding_type": "all_agency", + "summary": False, + "monthly": False, + "cusip": None, + "wam": True, + } + ), + ], +) +@pytest.mark.integration +def test_economy_central_bank_holdings(params, obb): + """Test economy central bank holdings.""" + params = {p: v for p, v in params.items() if v} + + result = obb.economy.central_bank_holdings(**params) + assert result + assert isinstance(result, OBBject) + assert len(result.results) > 0 + + +@parametrize( + "params", + [ + ( + { + "country": "united_states,united_kingdom", + "frequency": "monthly", + "provider": "oecd", + "start_date": "2022-01-01", + "end_date": "2024-04-01", + } + ), + ], +) +@pytest.mark.integration +def test_economy_share_price_index(params, obb): + """Test economy share price index.""" + params = {p: v for p, v in params.items() if v} + + result = obb.economy.share_price_index(**params) + assert result + assert isinstance(result, OBBject) + assert len(result.results) > 0 + + +@parametrize( + "params", + [ + ( + { + "country": "united_states,united_kingdom", + "frequency": "quarter", + "provider": "oecd", + "start_date": "2022-01-01", + "end_date": "2024-04-01", + "transform": "index", + } + ), + ], +) +@pytest.mark.integration +def test_economy_house_price_index(params, obb): + """Test economy house price index.""" + params = {p: v for p, v in params.items() if v} + + result = obb.economy.house_price_index(**params) + assert result + assert isinstance(result, OBBject) + assert len(result.results) > 0 + + +@parametrize( + "params", + [ + ( + { + "country": "united_states,united_kingdom", + "frequency": "monthly", + "provider": "oecd", + "start_date": "2022-01-01", + "end_date": "2024-04-01", + } + ), + ], +) +@pytest.mark.integration +def test_economy_immediate_interest_rate(params, obb): + """Test economy immediate interest rate.""" + params = {p: v for p, v in params.items() if v} + + result = obb.economy.immediate_interest_rate(**params) + assert result + assert isinstance(result, OBBject) + assert len(result.results) > 0 + + +@parametrize( + "params", + [ + ( + { + "country": "united_states", + "item": "meats", + "region": "all_city", + "frequency": "annual", + "provider": "fred", + "start_date": "2022-01-01", + "end_date": "2024-04-01", + "transform": "pc1", + } + ), + ], +) +@pytest.mark.integration +def test_economy_retail_prices(params, obb): + """Test economy retail prices.""" + params = {p: v for p, v in params.items() if v} + + result = obb.economy.retail_prices(**params) + assert result + assert isinstance(result, OBBject) + assert len(result.results) > 0 |