jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-28 13:01:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Working tests

Browse Source
This commit is contained in:
Igor Radovanovic
2023-09-25 20:10:26 +02:00
parent 4dce4236ec
commit a2e425bdd2
11 changed files with 229 additions and 173 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
openbb_sdk/extensions/crypto/integration/test_crypto_api.py
View File

@@ -27,7 +27,7 @@ def headers():
"provider": "fmp",
"symbol": "BTCUSD",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -49,7 +49,7 @@ def headers():
"provider": "polygon",
"symbol": "BTCUSD",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -72,7 +72,7 @@ def headers():
"provider": "yfinance",
"symbol": "BTCUSD",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(

31
openbb_sdk/extensions/crypto/integration/test_crypto_python.py
View File

@@ -1,5 +1,4 @@
"""Test crypto extension."""
import datetime
import pytest
from openbb import obb
@@ -12,8 +11,8 @@ from openbb_core.app.model.obbject import OBBject
(
{
"symbol": "BTCUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -21,8 +20,8 @@ from openbb_core.app.model.obbject import OBBject
"interval": "1min",
"provider": "fmp",
"symbol": "BTCUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -30,8 +29,8 @@ from openbb_core.app.model.obbject import OBBject
"interval": "1day",
"provider": "fmp",
"symbol": "BTCUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -43,8 +42,8 @@ from openbb_core.app.model.obbject import OBBject
"adjusted": True,
"provider": "polygon",
"symbol": "BTCUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -56,8 +55,8 @@ from openbb_core.app.model.obbject import OBBject
"adjusted": True,
"provider": "polygon",
"symbol": "BTCUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -66,8 +65,8 @@ from openbb_core.app.model.obbject import OBBject
"period": "max",
"provider": "yfinance",
"symbol": "BTCUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -76,14 +75,14 @@ from openbb_core.app.model.obbject import OBBject
"period": "max",
"provider": "yfinance",
"symbol": "BTCUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_crypto_load(params):
result = obb.crypto.load(params)
result = obb.crypto.load(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0

88
openbb_sdk/extensions/economy/integration/test_economy_python.py
View File

@@ -1,5 +1,4 @@
"""Test economy extension."""
import datetime
import pytest
from openbb import obb
@@ -13,7 +12,7 @@ from openbb_core.app.model.obbject import OBBject
],
)
def test_economy_const(params):
result = obb.economy.const(params)
result = obb.economy.const(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -28,14 +27,14 @@ def test_economy_const(params):
"units": "growth_same",
"frequency": "monthly",
"harmonized": True,
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_economy_cpi(params):
result = obb.economy.cpi(params)
result = obb.economy.cpi(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -47,8 +46,8 @@ def test_economy_cpi(params):
(
{
"symbol": "DJI",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -56,8 +55,8 @@ def test_economy_cpi(params):
"interval": "1m",
"provider": "cboe",
"symbol": "AAVE100",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -65,8 +64,8 @@ def test_economy_cpi(params):
"interval": "1d",
"provider": "cboe",
"symbol": "AAVE100",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -74,8 +73,8 @@ def test_economy_cpi(params):
"interval": "1min",
"provider": "fmp",
"symbol": "^DJI",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -83,8 +82,8 @@ def test_economy_cpi(params):
"interval": "1day",
"provider": "fmp",
"symbol": "^DJI",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -96,8 +95,8 @@ def test_economy_cpi(params):
"multiplier": 1,
"provider": "polygon",
"symbol": "NDX",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -109,8 +108,8 @@ def test_economy_cpi(params):
"multiplier": 1,
"provider": "polygon",
"symbol": "NDX",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -121,8 +120,8 @@ def test_economy_cpi(params):
"rounding": True,
"provider": "yfinance",
"symbol": "DJI",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -133,14 +132,14 @@ def test_economy_cpi(params):
"rounding": True,
"provider": "yfinance",
"symbol": "DJI",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_economy_index(params):
result = obb.economy.index(params)
result = obb.economy.index(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -152,8 +151,8 @@ def test_economy_index(params):
(
{
"symbol": "BUKBUS",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -161,8 +160,8 @@ def test_economy_index(params):
"interval": "1m",
"provider": "cboe",
"symbol": "BUKBUS",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -170,14 +169,14 @@ def test_economy_index(params):
"interval": "1d",
"provider": "cboe",
"symbol": "BUKBUS",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_economy_european_index(params):
result = obb.economy.european_index(params)
result = obb.economy.european_index(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -190,7 +189,7 @@ def test_economy_european_index(params):
],
)
def test_economy_european_index_constituents(params):
result = obb.economy.european_index_constituents(params)
result = obb.economy.european_index_constituents(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -204,7 +203,7 @@ def test_economy_european_index_constituents(params):
],
)
def test_economy_available_indices(params):
result = obb.economy.available_indices(params)
result = obb.economy.available_indices(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -217,7 +216,7 @@ def test_economy_available_indices(params):
],
)
def test_economy_risk(params):
result = obb.economy.risk(params)
result = obb.economy.risk(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -231,14 +230,14 @@ def test_economy_risk(params):
{
"europe": True,
"provider": "cboe",
"query": "DJ",
"query": "AA",
"symbol": True,
}
),
],
)
def test_economy_index_search(params):
result = obb.economy.index_search(params)
result = obb.economy.index_search(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -251,7 +250,7 @@ def test_economy_index_search(params):
],
)
def test_economy_index_snapshots(params):
result = obb.economy.index_snapshots(params)
result = obb.economy.index_snapshots(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -264,7 +263,7 @@ def test_economy_index_snapshots(params):
],
)
def test_economy_cot_search(params):
result = obb.economy.cot_search(params)
result = obb.economy.cot_search(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -281,8 +280,8 @@ def test_economy_cot_search(params):
"legacy_format": True,
"report_type": "ALL",
"measure": "CR",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"transform": "diff",
"provider": "quandl",
}
@@ -290,7 +289,7 @@ def test_economy_cot_search(params):
],
)
def test_economy_cot(params):
result = obb.economy.cot(params)
result = obb.economy.cot(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -302,8 +301,8 @@ def test_economy_cot(params):
(
{
"series_name": "PE Ratio by Month",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"collapse": "monthly",
"transform": "diff",
}
@@ -311,7 +310,8 @@ def test_economy_cot(params):
],
)
def test_economy_sp500_multiples(params):
result = obb.economy.sp500_multiples(params)
result = obb.economy.sp500_multiples(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
assert len(result.results) > 0

18
openbb_sdk/extensions/fixedincome/integration/test_fixedincome_python.py
View File

@@ -11,7 +11,7 @@ from openbb_core.app.model.obbject import OBBject
],
)
def test_fixedincome_treasury(params):
result = obb.fixedincome.treasury(params)
result = obb.fixedincome.treasury(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -24,7 +24,7 @@ def test_fixedincome_treasury(params):
],
)
def test_fixedincome_ycrv(params):
result = obb.fixedincome.ycrv(params)
result = obb.fixedincome.ycrv(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -45,7 +45,7 @@ def test_fixedincome_ycrv(params):
],
)
def test_fixedincome_sofr(params):
result = obb.fixedincome.sofr(params)
result = obb.fixedincome.sofr(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -66,7 +66,7 @@ def test_fixedincome_sofr(params):
],
)
def test_fixedincome_estr(params):
result = obb.fixedincome.estr(params)
result = obb.fixedincome.estr(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -87,7 +87,7 @@ def test_fixedincome_estr(params):
],
)
def test_fixedincome_sonia(params):
result = obb.fixedincome.sonia(params)
result = obb.fixedincome.sonia(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -108,7 +108,7 @@ def test_fixedincome_sonia(params):
],
)
def test_fixedincome_ameribor(params):
result = obb.fixedincome.ameribor(params)
result = obb.fixedincome.ameribor(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -129,7 +129,7 @@ def test_fixedincome_ameribor(params):
],
)
def test_fixedincome_fed(params):
result = obb.fixedincome.fed(params)
result = obb.fixedincome.fed(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -143,7 +143,7 @@ def test_fixedincome_fed(params):
],
)
def test_fixedincome_projections(params):
result = obb.fixedincome.projections(params)
result = obb.fixedincome.projections(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -156,7 +156,7 @@ def test_fixedincome_projections(params):
],
)
def test_fixedincome_iorb(params):
result = obb.fixedincome.iorb(params)
result = obb.fixedincome.iorb(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0

6
openbb_sdk/extensions/forex/integration/test_forex_api.py
View File

@@ -55,7 +55,7 @@ def test_forex_pairs(params, headers):
"provider": "fmp",
"symbol": "EURUSD",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -77,7 +77,7 @@ def test_forex_pairs(params, headers):
"provider": "polygon",
"symbol": "EURUSD",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -100,7 +100,7 @@ def test_forex_pairs(params, headers):
"provider": "yfinance",
"symbol": "EURUSD",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(

32
openbb_sdk/extensions/forex/integration/test_forex_python.py
View File

@@ -25,7 +25,7 @@ from openbb_core.app.model.obbject import OBBject
],
)
def test_forex_pairs(params):
result = obb.forex.pairs(params)
result = obb.forex.pairs(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -37,8 +37,8 @@ def test_forex_pairs(params):
(
{
"symbol": "EURUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -46,8 +46,8 @@ def test_forex_pairs(params):
"interval": "1min",
"provider": "fmp",
"symbol": "EURUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -55,8 +55,8 @@ def test_forex_pairs(params):
"interval": "1day",
"provider": "fmp",
"symbol": "EURUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -68,8 +68,8 @@ def test_forex_pairs(params):
"adjusted": True,
"provider": "polygon",
"symbol": "EURUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -81,8 +81,8 @@ def test_forex_pairs(params):
"adjusted": True,
"provider": "polygon",
"symbol": "EURUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -91,8 +91,8 @@ def test_forex_pairs(params):
"period": "max",
"provider": "yfinance",
"symbol": "EURUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -101,14 +101,14 @@ def test_forex_pairs(params):
"period": "max",
"provider": "yfinance",
"symbol": "EURUSD",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_forex_load(params):
result = obb.forex.load(params)
result = obb.forex.load(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0

4
openbb_sdk/extensions/futures/integration/test_futures_python.py
View File

@@ -32,7 +32,7 @@ from openbb_core.app.model.obbject import OBBject
],
)
def test_futures_load(params):
result = obb.futures.load(params)
result = obb.futures.load(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -45,7 +45,7 @@ def test_futures_load(params):
],
)
def test_futures_curve(params):
result = obb.futures.curve(params)
result = obb.futures.curve(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0

16
openbb_sdk/extensions/news/integration/test_news_python.py
View File

@@ -1,5 +1,4 @@
"""Test news extension."""
import datetime
import pytest
from openbb import obb
@@ -13,17 +12,17 @@ from openbb_core.app.model.obbject import OBBject
(
{
"display": "full",
"date": datetime.date(2023, 1, 1),
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"updated_since": 1,
"published_since": 1,
"date": "2023-01-01",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"updated_since": 900000,
"published_since": 900000,
"sort": "created",
"order": "desc",
"isin": "US0378331005",
"cusip": "037833100",
"channels": "General",
"topics": "Elon Musk",
"topics": "car",
"authors": "Benzinga Insights",
"content_types": "Car",
"provider": "benzinga",
@@ -33,7 +32,6 @@ from openbb_core.app.model.obbject import OBBject
],
)
def test_news_globalnews(params):
result = obb.news.globalnews(params)
result = obb.news.globalnews(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0

11
openbb_sdk/extensions/stocks/integration/test_stocks_api.py
View File

@@ -552,7 +552,7 @@ def test_stocks_options_chains(params, headers):
"provider": "cboe",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -570,7 +570,7 @@ def test_stocks_options_chains(params, headers):
"provider": "fmp",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -589,6 +589,7 @@ def test_stocks_options_chains(params, headers):
"provider": "intrinio",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -599,7 +600,7 @@ def test_stocks_options_chains(params, headers):
"provider": "intrinio",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -612,7 +613,7 @@ def test_stocks_options_chains(params, headers):
"provider": "polygon",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(
@@ -645,7 +646,7 @@ def test_stocks_options_chains(params, headers):
"provider": "yfinance",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"end_date": "2023-01-02",
}
),
(

186
openbb_sdk/extensions/stocks/integration/test_stocks_python.py
View File

@@ -1,6 +1,4 @@
"""Test stocks extension."""
import datetime
import pytest
from openbb import obb
from openbb_core.app.model.obbject import OBBject
@@ -37,7 +35,7 @@ from openbb_core.app.model.obbject import OBBject
],
)
def test_stocks_fa_balance(params):
result = obb.stocks.fa.balance(params)
result = obb.stocks.fa.balance(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -50,7 +48,7 @@ def test_stocks_fa_balance(params):
],
)
def test_stocks_fa_balance_growth(params):
result = obb.stocks.fa.balance_growth(params)
result = obb.stocks.fa.balance_growth(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -61,14 +59,14 @@ def test_stocks_fa_balance_growth(params):
[
(
{
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_stocks_fa_cal(params):
result = obb.stocks.fa.cal(params)
result = obb.stocks.fa.cal(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -105,7 +103,7 @@ def test_stocks_fa_cal(params):
],
)
def test_stocks_fa_cash(params):
result = obb.stocks.fa.cash(params)
result = obb.stocks.fa.cash(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -118,7 +116,7 @@ def test_stocks_fa_cash(params):
],
)
def test_stocks_fa_cash_growth(params):
result = obb.stocks.fa.cash_growth(params)
result = obb.stocks.fa.cash_growth(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -131,7 +129,7 @@ def test_stocks_fa_cash_growth(params):
],
)
def test_stocks_fa_comp(params):
result = obb.stocks.fa.comp(params)
result = obb.stocks.fa.comp(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -142,14 +140,14 @@ def test_stocks_fa_comp(params):
[
(
{
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_stocks_fa_comsplit(params):
result = obb.stocks.fa.comsplit(params)
result = obb.stocks.fa.comsplit(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -162,7 +160,7 @@ def test_stocks_fa_comsplit(params):
],
)
def test_stocks_fa_divs(params):
result = obb.stocks.fa.divs(params)
result = obb.stocks.fa.divs(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -175,7 +173,7 @@ def test_stocks_fa_divs(params):
],
)
def test_stocks_fa_earning(params):
result = obb.stocks.fa.earning(params)
result = obb.stocks.fa.earning(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -188,7 +186,7 @@ def test_stocks_fa_earning(params):
],
)
def test_stocks_fa_emp(params):
result = obb.stocks.fa.emp(params)
result = obb.stocks.fa.emp(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -201,7 +199,7 @@ def test_stocks_fa_emp(params):
],
)
def test_stocks_fa_est(params):
result = obb.stocks.fa.est(params)
result = obb.stocks.fa.est(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -238,7 +236,7 @@ def test_stocks_fa_est(params):
],
)
def test_stocks_fa_income(params):
result = obb.stocks.fa.income(params)
result = obb.stocks.fa.income(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -251,7 +249,7 @@ def test_stocks_fa_income(params):
],
)
def test_stocks_fa_income_growth(params):
result = obb.stocks.fa.income_growth(params)
result = obb.stocks.fa.income_growth(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -272,7 +270,7 @@ def test_stocks_fa_income_growth(params):
],
)
def test_stocks_fa_ins(params):
result = obb.stocks.fa.ins(params)
result = obb.stocks.fa.ins(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -285,13 +283,13 @@ def test_stocks_fa_ins(params):
{
"symbol": "AAPL",
"include_current_quarter": True,
"date": datetime.date(2023, 1, 1),
"date": "2023-01-01",
}
),
],
)
def test_stocks_fa_ins_own(params):
result = obb.stocks.fa.ins_own(params)
result = obb.stocks.fa.ins_own(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -304,7 +302,7 @@ def test_stocks_fa_ins_own(params):
],
)
def test_stocks_fa_metrics(params):
result = obb.stocks.fa.metrics(params)
result = obb.stocks.fa.metrics(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -317,7 +315,7 @@ def test_stocks_fa_metrics(params):
],
)
def test_stocks_fa_mgmt(params):
result = obb.stocks.fa.mgmt(params)
result = obb.stocks.fa.mgmt(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -330,7 +328,7 @@ def test_stocks_fa_mgmt(params):
],
)
def test_stocks_fa_overview(params):
result = obb.stocks.fa.overview(params)
result = obb.stocks.fa.overview(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -339,11 +337,11 @@ def test_stocks_fa_overview(params):
@pytest.mark.parametrize(
"params",
[
({"symbol": "AAPL", "date": datetime.date(2023, 1, 1), "page": 1}),
({"symbol": "AAPL", "date": "2023-01-01", "page": 1}),
],
)
def test_stocks_fa_own(params):
result = obb.stocks.fa.own(params)
result = obb.stocks.fa.own(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -356,7 +354,7 @@ def test_stocks_fa_own(params):
],
)
def test_stocks_fa_pt(params):
result = obb.stocks.fa.pt(params)
result = obb.stocks.fa.pt(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -370,7 +368,7 @@ def test_stocks_fa_pt(params):
],
)
def test_stocks_fa_pta(params):
result = obb.stocks.fa.pta(params)
result = obb.stocks.fa.pta(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -383,7 +381,7 @@ def test_stocks_fa_pta(params):
],
)
def test_stocks_fa_ratios(params):
result = obb.stocks.fa.ratios(params)
result = obb.stocks.fa.ratios(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -396,7 +394,7 @@ def test_stocks_fa_ratios(params):
],
)
def test_stocks_fa_revgeo(params):
result = obb.stocks.fa.revgeo(params)
result = obb.stocks.fa.revgeo(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -409,7 +407,7 @@ def test_stocks_fa_revgeo(params):
],
)
def test_stocks_fa_revseg(params):
result = obb.stocks.fa.revseg(params)
result = obb.stocks.fa.revseg(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -422,7 +420,7 @@ def test_stocks_fa_revseg(params):
],
)
def test_stocks_fa_sec(params):
result = obb.stocks.fa.sec(params)
result = obb.stocks.fa.sec(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -435,7 +433,7 @@ def test_stocks_fa_sec(params):
],
)
def test_stocks_fa_shrs(params):
result = obb.stocks.fa.shrs(params)
result = obb.stocks.fa.shrs(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -448,7 +446,7 @@ def test_stocks_fa_shrs(params):
],
)
def test_stocks_fa_split(params):
result = obb.stocks.fa.split(params)
result = obb.stocks.fa.split(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -461,7 +459,7 @@ def test_stocks_fa_split(params):
],
)
def test_stocks_fa_transcript(params):
result = obb.stocks.fa.transcript(params)
result = obb.stocks.fa.transcript(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -474,7 +472,7 @@ def test_stocks_fa_transcript(params):
],
)
def test_stocks_ca_peers(params):
result = obb.stocks.ca.peers(params)
result = obb.stocks.ca.peers(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -484,11 +482,11 @@ def test_stocks_ca_peers(params):
"params",
[
({"symbol": "AAPL"}),
({"date": datetime.date(2023, 1, 1), "provider": "intrinio", "symbol": "AAPL"}),
({"date": "2023-01-01", "provider": "intrinio", "symbol": "AAPL"}),
],
)
def test_stocks_options_chains(params):
result = obb.stocks.options.chains(params)
result = obb.stocks.options.chains(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -500,8 +498,8 @@ def test_stocks_options_chains(params):
(
{
"symbol": "AAPL",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
@@ -514,8 +512,17 @@ def test_stocks_options_chains(params):
"outputsize": "full",
"provider": "alpha_vantage",
"symbol": "AAPL",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
{
"interval": "1m",
"provider": "cboe",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -523,8 +530,17 @@ def test_stocks_options_chains(params):
"interval": "1d",
"provider": "cboe",
"symbol": "AAPL",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
{
"interval": "1min",
"provider": "fmp",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -532,8 +548,18 @@ def test_stocks_options_chains(params):
"interval": "1day",
"provider": "fmp",
"symbol": "AAPL",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
{
"timezone": "UTC",
"interval_size": "30m",
"provider": "intrinio",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -543,8 +569,21 @@ def test_stocks_options_chains(params):
"interval_size": "60m",
"provider": "intrinio",
"symbol": "AAPL",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
{
"multiplier": 1,
"timespan": "minute",
"sort": "desc",
"limit": 49999,
"adjusted": True,
"provider": "polygon",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -556,8 +595,28 @@ def test_stocks_options_chains(params):
"adjusted": True,
"provider": "polygon",
"symbol": "AAPL",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
(
{
"interval": "1m",
"period": "max",
"prepost": True,
"actions": True,
"auto_adjust": True,
"back_adjust": True,
"progress": True,
"ignore_tz": True,
"rounding": True,
"repair": True,
"keepna": True,
"group_by": "column",
"provider": "yfinance",
"symbol": "AAPL",
"start_date": "2023-01-01",
"end_date": "2023-01-02",
}
),
(
@@ -576,14 +635,14 @@ def test_stocks_options_chains(params):
"group_by": "column",
"provider": "yfinance",
"symbol": "AAPL",
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"start_date": "2023-01-01",
"end_date": "2023-06-06",
}
),
],
)
def test_stocks_load(params):
result = obb.stocks.load(params)
result = obb.stocks.load(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -596,9 +655,9 @@ def test_stocks_load(params):
(
{
"display": "full",
"date": datetime.date(2023, 1, 1),
"start_date": datetime.date(2023, 1, 1),
"end_date": datetime.date(2023, 6, 6),
"date": "2023-01-01",
"start_date": "2023-01-01",
"end_date": "2023-06-06",
"updated_since": 1,
"published_since": 1,
"sort": "created",
@@ -616,7 +675,7 @@ def test_stocks_load(params):
),
(
{
"published_utc": datetime.date(2023, 1, 1),
"published_utc": "2023-01-01",
"order": "desc",
"provider": "polygon",
"symbols": "AAPL",
@@ -626,10 +685,9 @@ def test_stocks_load(params):
],
)
def test_stocks_news(params):
result = obb.stocks.news(params)
result = obb.stocks.news(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@pytest.mark.parametrize(
@@ -639,7 +697,7 @@ def test_stocks_news(params):
],
)
def test_stocks_multiples(params):
result = obb.stocks.multiples(params)
result = obb.stocks.multiples(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -652,7 +710,7 @@ def test_stocks_multiples(params):
],
)
def test_stocks_search(params):
result = obb.stocks.search(params)
result = obb.stocks.search(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -666,7 +724,7 @@ def test_stocks_search(params):
],
)
def test_stocks_quote(params):
result = obb.stocks.quote(params)
result = obb.stocks.quote(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0
@@ -679,7 +737,7 @@ def test_stocks_quote(params):
],
)
def test_stocks_info(params):
result = obb.stocks.info(params)
result = obb.stocks.info(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0

4
openbb_sdk/extensions/tests/utils/integration_test_generator.py
View File

@@ -109,8 +109,8 @@ def add_test_commands_to_file( # pylint: disable=W0102
{params}
],
)
def test_{test_name}(params):
result = obb.{command_name}(params)
def test_{test_name}(**params):
result = obb.{command_name}(**params)
assert result
assert isinstance(result, OBBject)
assert len(result.results) > 0