def test_display_rsi_strategy(mocker):
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)
# MOCK VISUALIZE_OUTPUT
mocker.patch(
target=
"gamestonk_terminal.helper_classes.TerminalStyle.visualize_output")
ticker = "PM"
start = datetime.strptime("2020-12-01", "%Y-%m-%d")
end = datetime.strptime("2020-12-02", "%Y-%m-%d")
df_stock = stocks_helper.load_ticker(ticker=ticker,
start_date=start,
end_date=end)
bt_view.display_rsi_strategy(
ticker=ticker,
df_stock=df_stock,
periods=2,
low_rsi=2,
high_rsi=2,
spy_bt=True,
no_bench=False,
shortable=True,
export=False,
)
def test_display_rsi_strategy(mocker):
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)
mocker.patch.object(target=bt_view.gtff, attribute="USE_ION", new=True)
mocker.patch(
target="gamestonk_terminal.stocks.backtesting.bt_view.plt.ion")
mocker.patch(
target="gamestonk_terminal.stocks.backtesting.bt_view.plt.show")
ticker = "PM"
start = datetime.strptime("2020-12-01", "%Y-%m-%d")
end = datetime.strptime("2020-12-02", "%Y-%m-%d")
df_stock = stocks_helper.load_ticker(ticker=ticker,
start_date=start,
end_date=end)
bt_view.display_rsi_strategy(
ticker=ticker,
df_stock=df_stock,
periods=2,
low_rsi=2,
high_rsi=2,
spy_bt=True,
no_bench=False,
shortable=True,
export=False,
)
def call_rsi(self, other_args: List[str]):
"""Call RSI Strategy"""
parser = argparse.ArgumentParser(
add_help=False,
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
prog="rsi",
description=
"""Strategy that buys when the stock is less than a threshold
and shorts when it exceeds a threshold.""",
)
parser.add_argument(
"-p",
"--periods",
dest="periods",
help="Number of periods for RSI calculation",
type=check_positive,
default=14,
)
parser.add_argument(
"-u",
"--high",
default=70,
dest="high",
type=check_positive,
help="High (upper) RSI Level",
)
parser.add_argument(
"-l",
"--low",
default=30,
dest="low",
type=check_positive,
help="Low RSI Level",
)
parser.add_argument(
"--spy",
action="store_true",
default=False,
help="Flag to add spy hold comparison",
dest="spy",
)
parser.add_argument(
"--no_bench",
action="store_true",
default=False,
help="Flag to not show buy and hold comparison",
dest="no_bench",
)
parser.add_argument(
"--no_short",
action="store_false",
default=True,
dest="shortable",
help="Flag that disables the short sell",
)
ns_parser = parse_known_args_and_warn(
parser, other_args, export_allowed=EXPORT_ONLY_RAW_DATA_ALLOWED)
if ns_parser:
if ns_parser.high < ns_parser.low:
console.print("Low RSI value is higher than Low RSI value\n")
bt_view.display_rsi_strategy(
ticker=self.ticker,
df_stock=self.stock,
periods=ns_parser.periods,
low_rsi=ns_parser.low,
high_rsi=ns_parser.high,
spy_bt=ns_parser.spy,
no_bench=ns_parser.no_bench,
shortable=ns_parser.shortable,
export=ns_parser.export,
)
def call_rsi(self, other_args: List[str]):
"""Call RSI Strategy"""
parser = argparse.ArgumentParser(
add_help=False,
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
prog="rsi",
description=
"""Strategy that buys when the stock is less than a threshold
and shorts when it exceeds a threshold.""",
)
parser.add_argument(
"-p",
"--periods",
dest="periods",
help="Number of periods for RSI calculation",
type=check_positive,
default=14,
)
parser.add_argument(
"-u",
"--high",
default=70,
dest="high",
type=check_positive,
help="High (upper) RSI Level",
)
parser.add_argument(
"-l",
"--low",
default=30,
dest="low",
type=check_positive,
help="Low RSI Level",
)
parser.add_argument(
"--spy",
action="store_true",
default=False,
help="Flag to add spy hold comparison",
dest="spy",
)
parser.add_argument(
"--no_bench",
action="store_true",
default=False,
help="Flag to not show buy and hold comparison",
dest="no_bench",
)
parser.add_argument(
"--no_short",
action="store_false",
default=True,
dest="shortable",
help="Flag that disables the short sell",
)
parser.add_argument(
"--export",
choices=["csv", "json", "xlsx"],
default="",
type=str,
dest="export",
help="Export dataframe data to csv,json,xlsx file",
)
try:
ns_parser = parse_known_args_and_warn(parser, other_args)
if not ns_parser:
return
if ns_parser.high < ns_parser.low:
print("Low RSI value is higher than Low RSI value\n")
return
bt_view.display_rsi_strategy(
ticker=self.ticker,
df_stock=self.stock,
periods=ns_parser.periods,
low_rsi=ns_parser.low,
high_rsi=ns_parser.high,
spy_bt=ns_parser.spy,
no_bench=ns_parser.no_bench,
shortable=ns_parser.shortable,
export=ns_parser.export,
)
except Exception as e:
print(e, "\n")