Passing tests

.gitignore

@@ -76,9 +76,11 @@ cache/
 # lightning logs
 lightning_logs/
 
-# user script to install all extensions and providers
-openbb_platform/platform/core/tests/app/model/charts/test_chart_settings.py
+# testing artifacts
 */mocked_path
 
 # CA certificates
 *.pem
+
+# Platform auto generated files
+openbb_platform/openbb/package/*

openbb_platform/extensions/charting/integration/test_charting_api.py

@@ -63,8 +63,13 @@ def test_chart_stocks_load(params, headers):
     result = requests.get(url, headers=headers, timeout=10)
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -82,8 +87,12 @@ def test_chart_stocks_multiples(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -101,8 +110,12 @@ def test_chart_stocks_news(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -132,8 +145,12 @@ def test_chart_ta_adx(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -152,8 +169,12 @@ def test_chart_ta_aroon(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -183,8 +204,12 @@ def test_chart_ta_ema(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -214,8 +239,12 @@ def test_chart_ta_hma(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -246,8 +275,12 @@ def test_chart_ta_macd(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -278,8 +311,12 @@ def test_chart_ta_rsi(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -309,8 +346,12 @@ def test_chart_ta_sma(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -340,8 +381,12 @@ def test_chart_ta_wma(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]
 
 
 @pytest.mark.parametrize(
@@ -371,5 +416,9 @@ def test_chart_ta_zlma(params, headers):
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
 
-    assert result.json()["chart"]
-    assert list(result.json()["chart"].keys()) == ["content", "format"]
+    chart = result.json()["chart"]
+    fig = chart.pop("fig", {})
+
+    assert chart
+    assert not fig
+    assert list(chart.keys()) == ["content", "format"]

openbb_platform/extensions/charting/openbb_charting/core/openbb_figure.py

@@ -1222,7 +1222,9 @@ def to_html(self, *args, **kwargs) -> str:
 
         return super().to_html(*args, **kwargs)
 
-    def to_plotly_json(self, ndarray: bool = False) -> Dict[str, Any]:
+    def to_plotly_json(
+        self, ndarray: bool = False, np_nan: bool = False
+    ) -> Dict[str, Any]:
         """
         Convert figure to a JSON representation as a Python dict
 
@@ -1231,9 +1233,13 @@ def to_plotly_json(self, ndarray: bool = False) -> Dict[str, Any]:
         ----------
         ndarray : `bool`, optional
             If the plotly json should contain np.ndarray, by default False
+
+        np_nan : `bool`, optional
+            If the plotly json should contain np.nan, by default False
         """
 
         def remove_ndarrays(data):
+            """Remove ndarrays from the plotly json."""
             if isinstance(data, dict):
                 for key, val in data.items():
                     if isinstance(val, np.ndarray):
@@ -1249,9 +1255,28 @@ def remove_ndarrays(data):
 
             return data
 
+        def remove_nan(data):
+            """Remove nan from the plotly json."""
+            if isinstance(data, dict):
+                for key, val in data.items():
+                    if isinstance(val, float) and np.isnan(val):
+                        data[key] = None
+                    elif isinstance(val, (dict, list)):
+                        data[key] = remove_nan(val)
+            elif isinstance(data, list):
+                for i, val in enumerate(data):
+                    if isinstance(val, float) and np.isnan(val):
+                        data[i] = None
+                    elif isinstance(val, (dict, list)):
+                        data[i] = remove_nan(val)
+
+            return data
+
         plotly_json = super().to_plotly_json()
         if not ndarray:
             plotly_json = remove_ndarrays(plotly_json)
+        if not np_nan:
+            plotly_json = remove_nan(plotly_json)
 
         return plotly_json
 

openbb_platform/extensions/crypto/integration/test_crypto_api.py

@@ -18,7 +18,14 @@ def headers():
 @pytest.mark.parametrize(
     "params",
     [
-        ({"symbol": "BTCUSD", "start_date": "2023-01-01", "end_date": "2023-06-06"}),
+        (
+            {
+                "symbol": "BTC-USD",
+                "start_date": "2023-01-01",
+                "end_date": "2023-01-06",
+                "provider": "polygon",
+            }
+        ),
         (
             {
                 "interval": "1min",
@@ -31,7 +38,7 @@ def headers():
         ),
         (
             {
-                "interval": "1day",
+                "interval": "15min",
                 "provider": "fmp",
                 "symbol": "BTCUSD",
                 "start_date": "2023-01-01",
@@ -72,7 +79,7 @@ def headers():
                 "provider": "yfinance",
                 "symbol": "BTCUSD",
                 "start_date": "2023-01-01",
-                "end_date": "2023-01-02",
+                "end_date": "2023-01-08",
             }
         ),
         (

openbb_platform/extensions/crypto/integration/test_crypto_python.py

@@ -7,7 +7,6 @@
 @pytest.fixture(scope="session")
 def obb(pytestconfig):
     """Fixture to setup obb."""
-
     if pytestconfig.getoption("markexpr") != "not integration":
         import openbb
 
@@ -36,11 +35,11 @@ def obb(pytestconfig):
         ),
         (
             {
-                "interval": "1day",
+                "interval": "15min",
                 "provider": "fmp",
                 "symbol": "BTCUSD",
                 "start_date": "2023-01-01",
-                "end_date": "2023-06-06",
+                "end_date": "2023-01-03",
                 "timeseries": 1,
             }
         ),
@@ -70,16 +69,6 @@ def obb(pytestconfig):
                 "end_date": "2023-01-02",
             }
         ),
-        (
-            {
-                "interval": "5m",
-                "period": "max",
-                "provider": "yfinance",
-                "symbol": "BTCUSD",
-                "start_date": "2023-01-01",
-                "end_date": "2023-01-02",
-            }
-        ),
         (
             {
                 "interval": "1d",