diff options
Diffstat (limited to 'openbb_platform/obbject_extensions/charting/integration/test_charting_api.py')
-rw-r--r-- | openbb_platform/obbject_extensions/charting/integration/test_charting_api.py | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/openbb_platform/obbject_extensions/charting/integration/test_charting_api.py b/openbb_platform/obbject_extensions/charting/integration/test_charting_api.py index 5066d5a40b7..fc9f2b0b646 100644 --- a/openbb_platform/obbject_extensions/charting/integration/test_charting_api.py +++ b/openbb_platform/obbject_extensions/charting/integration/test_charting_api.py @@ -719,3 +719,42 @@ def test_charting_etf_holdings(params, headers): assert chart assert not fig assert list(chart.keys()) == ["content", "format"] + + +@parametrize( + "params", + [ + ( + { + "provider": "econdb", + "country": "united_kingdom", + "date": None, + "chart": True, + } + ), + ( + { + "provider": "fred", + "date": "2023-05-10,2024-05-10", + "chart": True, + } + ), + ], +) +@pytest.mark.integration +def test_charting_fixedincome_government_yield_curve(params, headers): + """Test chart fixedincome government yield curve.""" + params = {p: v for p, v in params.items() if v} + body = (json.dumps({"extra_params": {"chart_params": {"title": "test chart"}}}),) + query_str = get_querystring(params, []) + url = f"http://0.0.0.0:8000/api/v1/fixedincome/government/yield_curve?{query_str}" + result = requests.get(url, headers=headers, timeout=10, json=body) + assert isinstance(result, requests.Response) + assert result.status_code == 200 + + chart = result.json()["chart"] + fig = chart.pop("fig", {}) + + assert chart + assert not fig + assert list(chart.keys()) == ["content", "format"] |