diff options
Diffstat (limited to 'openbb_platform/extensions')
3 files changed, 46 insertions, 49 deletions
diff --git a/openbb_platform/extensions/equity/integration/test_equity_api.py b/openbb_platform/extensions/equity/integration/test_equity_api.py index 949a30a46a0..544bb8052b1 100644 --- a/openbb_platform/extensions/equity/integration/test_equity_api.py +++ b/openbb_platform/extensions/equity/integration/test_equity_api.py @@ -144,6 +144,14 @@ def test_equity_calendar_splits(params, headers): ({"start_date": "2023-11-09", "end_date": "2023-11-10", "provider": "fmp"}), ({"start_date": "2023-11-09", "end_date": "2023-11-10", "provider": "nasdaq"}), ({"start_date": "2023-11-09", "end_date": "2023-11-10", "provider": "tmx"}), + ( + { + "start_date": None, + "end_date": None, + "provider": "seeking_alpha", + "country": "us", + } + ), ], ) @pytest.mark.integration @@ -397,7 +405,14 @@ def test_equity_estimates_historical(params, headers): "calendar_period": None, "provider": "intrinio", } - ) + ), + ( + { + "symbol": "AAPL,BAM:CA", + "period": "annual", + "provider": "seeking_alpha", + } + ), ], ) @pytest.mark.integration @@ -434,6 +449,13 @@ def test_equity_estimates_forward_sales(params, headers): "provider": "fmp", } ), + ( + { + "symbol": "AAPL,BAM:CA", + "period": "annual", + "provider": "seeking_alpha", + } + ), ], ) @pytest.mark.integration @@ -1654,24 +1676,6 @@ def test_equity_discovery_top_retail(params, headers): @parametrize( "params", - [({"provider": "seeking_alpha"})], -) -@pytest.mark.integration -def test_equity_discovery_upcoming_release_days(params, headers): - """Test the equity discovery upcoming release days endpoint.""" - params = {p: v for p, v in params.items() if v} - - query_str = get_querystring(params, []) - url = ( - f"http://0.0.0.0:8000/api/v1/equity/discovery/upcoming_release_days?{query_str}" - ) - result = requests.get(url, headers=headers, timeout=30) - assert isinstance(result, requests.Response) - assert result.status_code == 200 - - -@parametrize( - "params", [ ( { diff --git a/openbb_platform/extensions/equity/integration/test_equity_python.py b/openbb_platform/extensions/equity/integration/test_equity_python.py index 249057fc6ca..cb836f59019 100644 --- a/openbb_platform/extensions/equity/integration/test_equity_python.py +++ b/openbb_platform/extensions/equity/integration/test_equity_python.py @@ -132,6 +132,14 @@ def test_equity_calendar_splits(params, obb): ({"start_date": "2023-11-09", "end_date": "2023-11-10", "provider": "fmp"}), ({"start_date": "2023-11-09", "end_date": "2023-11-10", "provider": "nasdaq"}), ({"start_date": "2023-11-09", "end_date": "2023-11-10", "provider": "tmx"}), + ( + { + "start_date": None, + "end_date": None, + "provider": "seeking_alpha", + "country": "us", + } + ), ], ) @pytest.mark.integration @@ -705,7 +713,14 @@ def test_equity_estimates_consensus(params, obb): "calendar_period": None, "provider": "intrinio", } - ) + ), + ( + { + "symbol": "AAPL,BAM:CA", + "period": "annual", + "provider": "seeking_alpha", + } + ), ], ) @pytest.mark.integration @@ -739,6 +754,13 @@ def test_equity_estimates_forward_sales(params, obb): "provider": "fmp", } ), + ( + { + "symbol": "AAPL,BAM:CA", + "period": "annual", + "provider": "seeking_alpha", + } + ), ], ) @pytest.mark.integration @@ -1572,21 +1594,6 @@ def test_equity_discovery_top_retail(params, obb): @parametrize( "params", - [({"provider": "seeking_alpha"})], -) -@pytest.mark.integration -def test_equity_discovery_upcoming_release_days(params, obb): - """Test the equity discovery upcoming release days endpoint.""" - params = {p: v for p, v in params.items() if v} - - result = obb.equity.discovery.upcoming_release_days(**params) - assert result - assert isinstance(result, OBBject) - assert len(result.results) > 0 - - -@parametrize( - "params", [ ( { diff --git a/openbb_platform/extensions/equity/openbb_equity/discovery/discovery_router.py b/openbb_platform/extensions/equity/openbb_equity/discovery/discovery_router.py index b8fbbc1a7be..652227975a6 100644 --- a/openbb_platform/extensions/equity/openbb_equity/discovery/discovery_router.py +++ b/openbb_platform/extensions/equity/openbb_equity/discovery/discovery_router.py @@ -153,20 +153,6 @@ async def top_retail( @router.command( - model="UpcomingReleaseDays", - examples=[APIEx(parameters={"provider": "seeking_alpha"})], -) -async def upcoming_release_days( - cc: CommandContext, - provider_choices: ProviderChoices, - standard_params: StandardParams, - extra_params: ExtraParams, -) -> OBBject: - """Get upcoming earnings release dates.""" - return await OBBject.from_query(Query(**locals())) - - -@router.command( model="DiscoveryFilings", examples=[ APIEx(parameters={"provider": "fmp"}), |