mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-02 23:41:27 +00:00
d8ca7556ed
* etf tests for stockanalysis * add financedatabase etf tests * fix financedatabase etf documentation * yfinance etf tests * add etf/discovery tests * add tests to etf/screener * add etf controller tests * add etf/ta tests * remove tabulate and use rich table * etf/pred * add more etf tests, thanks Chavi * update about us website * Updating tests : etf Co-authored-by: Chavithra PARANA <chavithra@gmail.com>
81 lines
1.5 KiB
Python
81 lines
1.5 KiB
Python
# IMPORTATION STANDARD
|
|
|
|
# IMPORTATION THIRDPARTY
|
|
import pytest
|
|
|
|
# IMPORTATION INTERNAL
|
|
from gamestonk_terminal.etf import stockanalysis_model
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def vcr_config():
|
|
return {
|
|
"filter_headers": [("User-Agent", None)],
|
|
}
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
def test_get_all_names_symbols(recorder):
|
|
result = stockanalysis_model.get_all_names_symbols()
|
|
|
|
recorder.capture_list(result)
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
@pytest.mark.parametrize(
|
|
"symbol",
|
|
[
|
|
"ARKQ",
|
|
"ARKW",
|
|
],
|
|
)
|
|
def test_get_etf_overview(recorder, symbol):
|
|
result_df = stockanalysis_model.get_etf_overview(symbol)
|
|
|
|
assert not result_df.empty
|
|
recorder.capture(result_df)
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
@pytest.mark.parametrize(
|
|
"symbol",
|
|
[
|
|
"ARKQ",
|
|
"ARKW",
|
|
],
|
|
)
|
|
def test_get_etf_holdings(recorder, symbol):
|
|
result_df = stockanalysis_model.get_etf_holdings(symbol)
|
|
|
|
assert not result_df.empty
|
|
recorder.capture(result_df)
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
@pytest.mark.parametrize(
|
|
"symbols",
|
|
[
|
|
["ARKQ", "ARKW"],
|
|
["ARKK", "ARKF"],
|
|
],
|
|
)
|
|
def test_compare_etfs(recorder, symbols):
|
|
result_df = stockanalysis_model.compare_etfs(symbols)
|
|
|
|
assert not result_df.empty
|
|
recorder.capture(result_df)
|
|
|
|
|
|
@pytest.mark.vcr()
|
|
@pytest.mark.parametrize(
|
|
"name_to_search",
|
|
[
|
|
"oil",
|
|
],
|
|
)
|
|
def test_get_etfs_by_name(recorder, name_to_search):
|
|
result_df = stockanalysis_model.get_etfs_by_name(name_to_search)
|
|
|
|
assert not result_df.empty
|
|
recorder.capture(result_df)
|