jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-29 13:32:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f68929fa1efaa5e9abb7911029e287c975b37fd3
OpenBB/tests/openbb_terminal/stocks/technical_analysis/test_ta_controller.py
Colin Delahunty 87458c965d Stocks Suggestor Repair (#3477) 
* broken edition

* Fixed issue

* Update StockController

* Update choices generation

* Update parent_classes

* Update choices generation

* Update choices generation

* Update crypto_controller

* Update DefiController

* Update crypto/disc controller

* Update crypto/dd

* Update choices generation

* Update crypto : dd + nft

* Update crypto/onchain

* Update crypto_controller

* Linting

* Update crypto/ov

* Linting

* Update crypto/ov

* Update crypto/qa

* Update crypto/tools

* Update crypto/ta + dd

* Linting

* Update alt

* Tests fix crypto

* Update future

* Update futures

* Refactored through Options

* Tests + Linting

* finished stocks refactor

* Update stocks/sia

* finished stocks refactor

* Added fixes

* Added fix

* Tests

* Fix conditions

* Fix portfolio

* Prepare for merge

Co-authored-by: Chavithra PARANA <chavithra@gmail.com>
Co-authored-by: northern-64bit <75195383+northern-64bit@users.noreply.github.com>
2022-11-21 17:54:26 +01:00

794 lines
20 KiB
Python
Raw Blame History

# IMPORTATION STANDARD
import os
from datetime import datetime
# IMPORTATION THIRDPARTY
import pandas as pd
import pytest
# IMPORTATION INTERNAL
from openbb_terminal.stocks.technical_analysis import ta_controller
from openbb_terminal.stocks import stocks_helper
# pylint: disable=E1101
# pylint: disable=W0603
# pylint: disable=E1111
EMPTY_DF = pd.DataFrame()
MOCK_STOCK_DF = pd.read_csv(
"tests/openbb_terminal/stocks/technical_analysis/csv/test_ta_controller/stock_df.csv",
index_col=0,
)
@pytest.fixture(scope="module")
def vcr_config():
return {
"filter_headers": [("User-Agent", None)],
"filter_query_parameters": [
("period1", "MOCK_PERIOD_1"),
("period2", "MOCK_PERIOD_2"),
("date", "MOCK_DATE"),
],
}
@pytest.mark.vcr(record_mode="none")
@pytest.mark.parametrize(
"queue, expected",
[
(["load", "help"], ["help"]),
(["quit", "help"], ["help"]),
],
)
def test_menu_with_queue(expected, mocker, queue):
path_controller = "openbb_terminal.stocks.technical_analysis.ta_controller"
# MOCK SWITCH
mocker.patch(
target=f"{path_controller}.TechnicalAnalysisController.switch",
return_value=["quit"],
)
result_menu = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=EMPTY_DF,
queue=queue,
).menu()
assert result_menu == expected
@pytest.mark.vcr(record_mode="none")
def test_menu_without_queue_completion(mocker):
path_controller = "openbb_terminal.stocks.technical_analysis.ta_controller"
# ENABLE AUTO-COMPLETION : HELPER_FUNCS.MENU
mocker.patch(
target="openbb_terminal.feature_flags.USE_PROMPT_TOOLKIT",
new=True,
)
mocker.patch(
target="openbb_terminal.parent_classes.session",
)
mocker.patch(
target="openbb_terminal.parent_classes.session.prompt",
return_value="quit",
)
# DISABLE AUTO-COMPLETION : CONTROLLER.COMPLETER
# DISABLE AUTO-COMPLETION
mocker.patch.object(
target=ta_controller.obbff,
attribute="USE_PROMPT_TOOLKIT",
new=True,
)
mocker.patch(
target=f"{path_controller}.session",
)
mocker.patch(
target=f"{path_controller}.session.prompt",
return_value="quit",
)
result_menu = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=EMPTY_DF,
queue=None,
).menu()
assert result_menu == ["help"]
@pytest.mark.vcr(record_mode="none")
@pytest.mark.parametrize(
"mock_input",
["help", "homee help", "home help", "mock"],
)
def test_menu_without_queue_sys_exit(mock_input, mocker):
path_controller = "openbb_terminal.stocks.technical_analysis.ta_controller"
# DISABLE AUTO-COMPLETION
mocker.patch.object(
target=ta_controller.obbff,
attribute="USE_PROMPT_TOOLKIT",
new=False,
)
mocker.patch(
target=f"{path_controller}.session",
return_value=None,
)
# MOCK USER INPUT
mocker.patch("builtins.input", return_value=mock_input)
# MOCK SWITCH
class SystemExitSideEffect:
def __init__(self):
self.first_call = True
def __call__(self, *args, **kwargs):
if self.first_call:
self.first_call = False
raise SystemExit()
return ["quit"]
mock_switch = mocker.Mock(side_effect=SystemExitSideEffect())
mocker.patch(
target=f"{path_controller}.TechnicalAnalysisController.switch",
new=mock_switch,
)
result_menu = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=EMPTY_DF,
queue=None,
).menu()
assert result_menu == ["help"]
@pytest.mark.vcr(record_mode="none")
@pytest.mark.record_stdout
@pytest.mark.parametrize(
"start",
[
datetime.strptime("2021-12-01", "%Y-%m-%d"),
None,
],
)
def test_print_help(start):
controller = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=start,
interval="MOCK_INTERVAL",
stock=EMPTY_DF,
queue=None,
)
controller.print_help()
@pytest.mark.vcr(record_mode="none")
@pytest.mark.parametrize(
"an_input, expected_queue",
[
("", []),
(
"/help",
[
"home",
"help",
],
),
("help/help", ["help", "help"]),
("q", ["quit"]),
("h", []),
(
"r",
[
"quit",
"quit",
"reset",
"stocks",
"load MOCK_TICKER",
"ta",
],
),
],
)
def test_switch(an_input, expected_queue):
controller = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=MOCK_STOCK_DF,
queue=None,
)
queue = controller.switch(an_input=an_input)
assert queue == expected_queue
@pytest.mark.vcr(record_mode="none")
def test_call_cls(mocker):
mocker.patch("os.system")
controller = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=EMPTY_DF,
queue=None,
)
controller.call_cls([])
assert controller.queue == []
os.system.assert_called_once_with("cls||clear")
@pytest.mark.vcr(record_mode="none")
@pytest.mark.parametrize(
"func, queue, expected_queue",
[
(
"call_exit",
[],
["quit", "quit", "quit"],
),
("call_exit", ["help"], ["quit", "quit", "quit", "help"]),
("call_home", [], ["quit", "quit"]),
("call_help", [], []),
("call_quit", [], ["quit"]),
("call_quit", ["help"], ["quit", "help"]),
(
"call_reset",
[],
[
"quit",
"quit",
"reset",
"stocks",
"load MOCK_TICKER",
"ta",
],
),
(
"call_reset",
["help"],
[
"quit",
"quit",
"reset",
"stocks",
"load MOCK_TICKER",
"ta",
"help",
],
),
],
)
def test_call_func_expect_queue(expected_queue, func, queue):
controller = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=EMPTY_DF,
queue=queue,
)
result = getattr(controller, func)([])
assert result is None
assert controller.queue == expected_queue
@pytest.mark.vcr(record_mode="none")
@pytest.mark.parametrize(
"tested_func, other_args, mocked_func, called_args, called_kwargs",
[
(
"call_view",
[],
"finviz_view.view",
[
"MOCK_TICKER",
],
dict(),
),
(
"call_summary",
[],
"finbrain_view.technical_summary_report",
[
"MOCK_TICKER",
],
dict(),
),
(
"call_recom",
[
"--screener=cfd",
"--exchange=MOCK_EXCHANGE",
"--interval=1m",
"--export=csv",
],
"tradingview_view.print_recommendation",
[],
dict(
symbol="MOCK_TICKER",
screener="cfd",
exchange="MOCK_EXCHANGE",
interval="1m",
export="csv",
),
),
(
"call_ema",
[
"1,2",
"--offset=2",
"--export=csv",
],
"overlap_view.view_ma",
[],
dict(
ma_type="EMA",
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
window=[1, 2],
offset=2,
export="csv",
),
),
(
"call_sma",
[
"1,2",
"--offset=2",
"--export=csv",
],
"overlap_view.view_ma",
[],
dict(
ma_type="SMA",
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
window=[1, 2],
offset=2,
export="csv",
),
),
(
"call_wma",
[
"1,2",
"--offset=2",
"--export=csv",
],
"overlap_view.view_ma",
[],
dict(
ma_type="WMA",
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
window=[1, 2],
offset=2,
export="csv",
),
),
(
"call_hma",
[
"1,2",
"--offset=2",
"--export=csv",
],
"overlap_view.view_ma",
[],
dict(
ma_type="HMA",
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
window=[1, 2],
offset=2,
export="csv",
),
),
(
"call_vwap",
[
"--offset=2",
"--export=csv",
],
"overlap_view.view_vwap",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
interval="MOCK_INTERVAL",
start_date=None,
end_date=None,
offset=2,
export="csv",
),
),
(
"call_zlma",
[
"1,2",
"--offset=2",
"--export=csv",
],
"overlap_view.view_ma",
[],
dict(
ma_type="ZLMA",
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
window=[1, 2],
offset=2,
export="csv",
),
),
(
"call_cci",
[
"--window=1",
"--scalar=2",
"--export=csv",
],
"momentum_view.display_cci",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
window=14,
scalar=2,
export="csv",
),
),
(
"call_macd",
[
"--fast=1",
"--slow=2",
"--signal=3",
"--export=csv",
],
"momentum_view.display_macd",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
n_fast=1,
n_slow=2,
n_signal=3,
export="csv",
),
),
(
"call_rsi",
[
"1",
"--scalar=2",
"--drift=3",
"--export=csv",
],
"momentum_view.display_rsi",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
window=1,
scalar=2,
drift=3,
export="csv",
),
),
(
"call_rsp",
[
"-t",
"--export=csv",
],
"rsp_view.display_rsp",
[],
dict(
s_ticker="MOCK_TICKER",
tickers_show=True,
export="csv",
),
),
(
"call_stoch",
[
"--fastkperiod=1",
"--slowdperiod=2",
"--slowkperiod=3",
"--export=csv",
],
"momentum_view.display_stoch",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
fastkperiod=1,
slowdperiod=2,
slowkperiod=3,
export="csv",
),
),
(
"call_fisher",
[
"1",
"--export=csv",
],
"momentum_view.display_fisher",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
window=1,
export="csv",
),
),
(
"call_cg",
[
"1",
"--export=csv",
],
"momentum_view.display_cg",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF["Adj Close"],
window=1,
export="csv",
),
),
(
"call_adx",
[
"1",
"--scalar=2",
"--drift=3",
"--export=csv",
],
"trend_indicators_view.display_adx",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
window=1,
scalar=2,
drift=3,
export="csv",
),
),
(
"call_aroon",
[
"1",
"--scalar=2",
"--export=csv",
],
"trend_indicators_view.display_aroon",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
window=1,
scalar=2,
export="csv",
),
),
(
"call_bbands",
[
"1",
"--std=2",
"--mamode=ema",
"--export=csv",
],
"volatility_view.display_bbands",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
window=1,
n_std=2,
mamode="ema",
export="csv",
),
),
(
"call_atr",
["--mamode=sma", "--export=csv"],
"volatility_view.display_atr",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
mamode="sma",
window=14,
offset=0,
export="csv",
),
),
(
"call_donchian",
[
"--length_upper=1",
"--length_lower=2",
"--export=csv",
],
"volatility_view.display_donchian",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
upper_length=1,
lower_length=2,
export="csv",
),
),
(
"call_kc",
[
"1",
"--scalar=2",
"--mamode=sma",
"--offset=3",
"--export=csv",
],
"volatility_view.view_kc",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
window=1,
scalar=2,
mamode="sma",
offset=3,
export="csv",
),
),
(
"call_ad",
[
"--open",
"--export=csv",
],
"volume_view.display_ad",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
use_open=True,
export="csv",
),
),
(
"call_adosc",
[
"--open",
"--fast=1",
"--slow=2",
"--export=csv",
],
"volume_view.display_adosc",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
use_open=True,
fast=1,
slow=2,
export="csv",
),
),
(
"call_obv",
[
"--export=csv",
],
"volume_view.display_obv",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
export="csv",
),
),
(
"call_fib",
[
"1",
"--start=2021-12-01",
"--end=2021-12-02",
"--export=csv",
],
"custom_indicators_view.fibonacci_retracement",
[],
dict(
symbol="MOCK_TICKER",
data=MOCK_STOCK_DF,
limit=1,
start_date=datetime.strptime("2021-12-01", "%Y-%m-%d"),
end_date=datetime.strptime("2021-12-02", "%Y-%m-%d"),
export="csv",
),
),
],
)
def test_call_func(
tested_func, mocked_func, other_args, called_args, called_kwargs, mocker
):
path_controller = "openbb_terminal.stocks.technical_analysis.ta_controller"
if mocked_func:
mock = mocker.Mock()
mocker.patch(
target=f"{path_controller}.{mocked_func}",
new=mock,
)
controller = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=MOCK_STOCK_DF,
queue=None,
)
controller.screen_tickers = ["PM"]
getattr(controller, tested_func)(other_args)
if called_args or called_kwargs:
mock.assert_called_once_with(*called_args, **called_kwargs)
else:
mock.assert_called_once()
else:
controller = ta_controller.TechnicalAnalysisController(
ticker="MOCK_TICKER",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="MOCK_INTERVAL",
stock=EMPTY_DF,
queue=None,
)
controller.screen_tickers = ["PM"]
getattr(controller, tested_func)(other_args)
@pytest.mark.vcr
def test_call_load(mocker):
# FORCE SINGLE THREADING
yf_download = stocks_helper.yf.download
def mock_yf_download(*args, **kwargs):
kwargs["threads"] = False
return yf_download(*args, **kwargs)
mocker.patch("yfinance.download", side_effect=mock_yf_download)
controller = ta_controller.TechnicalAnalysisController(
ticker="TSLA",
start=datetime.strptime("2021-12-01", "%Y-%m-%d"),
interval="1440",
stock=pd.DataFrame(),
queue=None,
)
other_args = [
"TSLA",
"--start=2021-12-17",
"--end=2021-12-18",
"--source=YahooFinance",
]
controller.call_load(other_args=other_args)
Reference in New Issue View Git Blame Copy Permalink