diff options
Diffstat (limited to 'openbb_platform/extensions/equity/integration/test_equity_python.py')
-rw-r--r-- | openbb_platform/extensions/equity/integration/test_equity_python.py | 64 |
1 files changed, 42 insertions, 22 deletions
diff --git a/openbb_platform/extensions/equity/integration/test_equity_python.py b/openbb_platform/extensions/equity/integration/test_equity_python.py index 115d3631700..33861f3757b 100644 --- a/openbb_platform/extensions/equity/integration/test_equity_python.py +++ b/openbb_platform/extensions/equity/integration/test_equity_python.py @@ -554,7 +554,7 @@ def test_equity_calendar_ipo(params, obb): "with_ttm": False, } ), - ({"provider": "intrinio", "symbol": "AAPL", "period": "annual", "limit": 100}), + ({"provider": "intrinio", "symbol": "AAPL", "limit": 100}), ({"provider": "yfinance", "symbol": "AAPL"}), ({"provider": "finviz", "symbol": "AAPL,GOOG"}), ], @@ -590,21 +590,6 @@ def test_equity_fundamental_management(params, obb): @parametrize( "params", [ - ({"symbol": "AAPL"}), - ], -) -@pytest.mark.integration -def test_equity_fundamental_overview(params, obb): - """Test the equity fundamental overview endpoint.""" - result = obb.equity.fundamental.overview(**params) - assert result - assert isinstance(result, OBBject) - assert result.results is not None - - -@parametrize( - "params", - [ ({"symbol": "AAPL", "date": "2023-01-01", "page": 1, "provider": "fmp"}), ], ) @@ -1098,8 +1083,6 @@ def test_equity_compare_groups(params, obb): "start_date": "2023-06-01", "end_date": "2023-06-03", "interval": "1h", - "adjusted": True, - "prepost": False, } ), ( @@ -1112,8 +1095,6 @@ def test_equity_compare_groups(params, obb): "start_date": "2023-01-01", "end_date": "2023-06-06", "interval": "1d", - "adjusted": False, - "prepost": False, } ), ( @@ -1348,7 +1329,7 @@ def test_equity_fundamental_latest_attributes(params, obb): [ ({"query": "AAPL", "is_symbol": True, "provider": "cboe", "use_cache": False}), ({"query": "Apple", "provider": "sec", "use_cache": False, "is_fund": False}), - ({"query": "", "provider": "nasdaq", "use_cache": False, "is_etf": True}), + ({"query": "", "provider": "nasdaq", "is_etf": True}), ({"query": "gold", "provider": "tmx", "use_cache": False}), ({"query": "gold", "provider": "tradier", "is_symbol": False}), ( @@ -1357,7 +1338,6 @@ def test_equity_fundamental_latest_attributes(params, obb): "provider": "intrinio", "active": True, "limit": 100, - "use_cache": None, } ), ], @@ -1409,6 +1389,22 @@ def test_equity_search(params, obb): "provider": "nasdaq", } ), + ( + { + "metric": "overview", + "signal": None, + "preset": None, + "filters_dict": None, + "sector": "consumer_defensive", + "industry": "grocery_stores", + "index": "all", + "exchange": "all", + "mktcap": "all", + "recommendation": "all", + "limit": None, + "provider": "finviz", + } + ), ], ) @pytest.mark.integration @@ -1935,3 +1931,27 @@ def test_equity_compare_company_facts(params, obb): assert result assert isinstance(result, OBBject) assert len(result.results) > 0 + + +@parametrize( + "params", + [ + ( + { + "symbol": "AAPL,MSFT", + "start_date": None, + "end_date": None, + "provider": "fmp", + } + ) + ], +) +@pytest.mark.integration +def test_equity_historical_market_cap(params, obb): + """Test the equity historical market cap endpoint.""" + params = {p: v for p, v in params.items() if v} + + result = obb.equity.historical_market_cap(**params) + assert result + assert isinstance(result, OBBject) + assert len(result.results) > 0 |