summaryrefslogtreecommitdiffstats
path: root/openbb_platform/obbject_extensions/charting/integration/test_charting_api.py
diff options
context:
space:
mode:
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.py32
1 files changed, 32 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 d5df11fdc63..2c5a8cbb47e 100644
--- a/openbb_platform/obbject_extensions/charting/integration/test_charting_api.py
+++ b/openbb_platform/obbject_extensions/charting/integration/test_charting_api.py
@@ -402,3 +402,35 @@ def test_charting_technical_zlma(params, headers):
assert chart
assert not fig
assert list(chart.keys()) == ["content", "format"]
+
+
+@parametrize(
+ "params",
+ [
+ (
+ {
+ "data": "",
+ "model": "yang_zhang",
+ "chart": True,
+ }
+ )
+ ],
+)
+@pytest.mark.integration
+def test_charting_technical_cones(params, headers):
+ """Test chart ta cones."""
+ params = {p: v for p, v in params.items() if v}
+ body = json.dumps(get_equity_data())
+
+ query_str = get_querystring(params, [])
+ url = f"http://0.0.0.0:8000/api/v1/technical/cones?{query_str}"
+ result = requests.post(url, headers=headers, timeout=10, data=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"]