"""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", [ ({"symbol": "dowjones", "provider": "fmp"}), ({"symbol": "BUKBUS", "provider": "cboe"}), ({"symbol": "^TX60", "provider": "tmx", "use_cache": False}), ], ) @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": "1d", "provider": "cboe", "symbol": "AAVE100", "start_date": "2024-01-01", "end_date": "2024-02-05", "use_cache": False, "sort": None, "limit": None, } ), ( { "interval": "1d", "provider": "cboe", "symbol": "AAVE100", "start_date": "2023-01-01", "end_date": "2023-06-06", "use_cache": False, "sort": None, "limit": None, } ), ( { "interval": "1min", "provider": "fmp", "symbol": "^DJI", "start_date": "2024-01-01", "end_date": "2024-02-05", "timeseries": 1, "sort": "desc", "limit": None, } ), ( { "interval": "1day", "provider": "fmp", "symbol": "^DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", "timeseries": 1, "sort": "desc", "limit": None, } ), ( { "interval": "1m", "sort": "desc", "limit": 49999, "provider": "polygon", "symbol": "NDX", "start_date": "2023-01-01", "end_date": "2023-06-06", } ), ( { "interval": "1d", "sort": "desc", "limit": 49999, "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", "sort": None, "limit": None, } ), ( { "interval": "1d", "provider": "intrinio", "start_date": "2023-01-01", "end_date": "2023-06-06", "symbol": "DJI", "sort": "desc", "limit": 100, } ), ], ) @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": "^DJI", "start_date": "2023-01-01", "end_date": "2023-03-03", "provider": "fmp", "sort": "desc", "interval": "1day", "limit": None, } ), ( { "interval": "1d", "provider": "cboe", "symbol": "AAVE100", "start_date": "2023-01-01", "end_date": "2023-01-05", "use_cache": True, "limit": None, "sort": None, } ), ( { "interval": "1d", "provider": "cboe", "symbol": "AAVE100", "start_date": "2024-01-01", "end_date": "2024-02-06", "use_cache": False, "limit": None, "sort": None, } ), ( { "interval": "1min", "provider": "fmp", "symbol": "^DJI", "start_date": "2024-02-01", "end_date": "2024-02-03", "timeseries": 1, "sort": "desc", "limit": None, } ), ( { "interval": "1day", "provider": "fmp", "symbol": "^DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", "timeseries": 1, "sort": "desc", "limit": None, } ), ( { "timespan": "minute", "sort": "desc", "limit": 49999, "adjusted": True, "multiplier": 1, "provider": "polygon", "symbol": "NDX", "start_date": "2023-01-01", "end_date": "2023-06-06", "interval": None, } ), ( { "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": None, } ), ( { "interval": "1d", "period": "max", "prepost": True, "rounding": True, "provider": "yfinance", "symbol": "DJI", "start_date": "2023-01-01", "end_date": "2023-06-06", "limit": None, "sort": None, } ), ( { "provider": "intrinio", "start_date": "2023-01-01", "end_date": "2023-06-06", "symbol": "$DJI", "tag": "level", "sort": "desc", "limit": 100, "type": None, "interval": 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"}), ({"provider": "tmx", "use_cache": False}), ], ) @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, "provider": "cboe", "use_cache": False, } ), ], ) @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"}), ({"provider": "tmx", "region": "ca", "use_cache": False}), ], ) @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_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 @parametrize( "params", [ ({"symbol": "^TX60", "provider": "tmx"}), ], ) @pytest.mark.integration def test_index_sectors(params, obb): result = obb.index.sectors(**params) assert result assert isinstance(result, OBBject) assert len(result.results) > 0