"""Test economy extension.""" import pytest from extensions.tests.conftest import parametrize from openbb_core.app.model.obbject import OBBject @pytest.fixture(scope="session") def obb(pytestconfig): # pylint: disable=inconsistent-return-statements """Fixture to setup obb.""" if pytestconfig.getoption("markexpr") != "not integration": import openbb # pylint: disable=import-outside-toplevel return openbb.obb # pylint: disable=redefined-outer-name @parametrize( "params", [ ({"index": "dowjones", "provider": "fmp"}), ({"index": "BUKBUS", "provider": "cboe"}), ], ) @pytest.mark.integration def test_index_constituents(params, obb): result = obb.index.constituents(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0 @parametrize( "params", [ ( { "interval": "1m", "provider": "cboe", "symbol": "AAVE100", "start_date": "2023-01-01", "end_date": "2023-06-06", "use_cache": False, } ), ( { "interval": "1d", "provider": "cboe", "symbol": "AAVE100", "start_date": "2023-01-01", "end_date": "2023-06-06", "use_cache": False, } ), ( { "interval": "1min", "provider": "fmp", "symbol": "^DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", "timeseries": 1, "sort": "desc", } ), ( { "interval": "1day", "provider": "fmp", "symbol": "^DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", "timeseries": 1, "sort": "desc", } ), ( { "timespan": "minute", "sort": "desc", "limit": 49999, "adjusted": True, "multiplier": 1, "provider": "polygon", "symbol": "NDX", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "timespan": "day", "sort": "desc", "limit": 49999, "adjusted": True, "multiplier": 1, "provider": "polygon", "symbol": "NDX", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "interval": "1d", "period": "max", "prepost": True, "rounding": True, "provider": "yfinance", "symbol": "DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "provider": "intrinio", "start_date": "2023-01-01", "end_date": "2023-06-06", "symbol": "$DJI", "tag": "level", "sort": "desc", "limit": 100, "type": None, } ), ], ) @pytest.mark.integration def test_index_price_historical(params, obb): result = obb.index.price.historical(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0 @parametrize( "params", [ ( { "symbol": "AAVE100", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "interval": "1m", "provider": "cboe", "symbol": "AAVE100", "start_date": "2023-01-01", "end_date": "2023-06-06", "use_cache": True, } ), ( { "interval": "1d", "provider": "cboe", "symbol": "AAVE100", "start_date": "2023-01-01", "end_date": "2023-06-06", "use_cache": False, } ), ( { "interval": "1min", "provider": "fmp", "symbol": "^DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", "timeseries": 1, "sort": "desc", } ), ( { "interval": "1day", "provider": "fmp", "symbol": "^DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", "timeseries": 1, "sort": "desc", } ), ( { "timespan": "minute", "sort": "desc", "limit": 49999, "adjusted": True, "multiplier": 1, "provider": "polygon", "symbol": "NDX", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "timespan": "day", "sort": "desc", "limit": 49999, "adjusted": True, "multiplier": 1, "provider": "polygon", "symbol": "NDX", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "interval": "1d", "period": "max", "prepost": True, "rounding": True, "provider": "yfinance", "symbol": "DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "provider": "intrinio", "start_date": "2023-01-01", "end_date": "2023-06-06", "symbol": "$DJI", "tag": "level", "sort": "desc", "limit": 100, "type": None, } ), ], ) @pytest.mark.integration def test_index_market(params, obb): result = obb.index.market(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0 @parametrize( "params", [ ({}), ({"provider": "cboe", "use_cache": False}), ({"provider": "fmp"}), ({"provider": "yfinance"}), ], ) @pytest.mark.integration def test_index_available(params, obb): result = obb.index.available(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0 @parametrize( "params", [ ({"query": "D", "is_symbol": True, "use_cache": False, "provider": "cboe"}), ], ) @pytest.mark.integration def test_index_search(params, obb): result = obb.index.search(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0 @parametrize( "params", [ ({"region": "us", "provider": "cboe"}), ], ) @pytest.mark.integration def test_index_snapshots(params, obb): result = obb.index.snapshots(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0 @parametrize( "params", [ ( { "series_name": "PE Ratio by Month", "start_date": "2023-01-01", "end_date": "2023-06-06", "collapse": "monthly", "transform": "diff", "provider": "nasdaq", } ), ], ) @pytest.mark.integration def test_index_sp500_multiples(params, obb): result = obb.index.sp500_multiples(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0