def test_display_ark_trades_no_tab(mocker):
mocker.patch(
target="gamestonk_terminal.feature_flags.USE_TABULATE_DF",
new=False,
)
ark_view.display_ark_trades(ticker="TSLA")
def test_display_ark_trades_export(capsys, mocker):
ark_view.export_data = mocker.Mock()
ark_view.display_ark_trades(ticker="TSLA", export="csv")
capsys.readouterr()
ark_view.export_data.assert_called_once()
def call_arktrades(self, other_args):
"""Process arktrades command"""
parser = argparse.ArgumentParser(
add_help=False,
prog="arktrades",
description="""
Get trades for ticker across all ARK funds.
""",
)
parser.add_argument(
"-n",
"--num",
help="Number of rows to show",
dest="num",
default=20,
type=int,
)
parser.add_argument(
"-s",
"--show_ticker",
action="store_true",
default=False,
help="Flag to show ticker in table",
dest="show_ticker",
)
try:
ns_parser = parse_known_args_and_warn(
parser, other_args, export_allowed=EXPORT_ONLY_RAW_DATA_ALLOWED
)
if not ns_parser:
return
ark_view.display_ark_trades(
ticker=self.ticker,
num=ns_parser.num,
export=ns_parser.export,
show_ticker=ns_parser.show_ticker,
)
except Exception as e:
print(e, "\n")
def call_arktrades(self, other_args):
"""Process arktrades command"""
parser = argparse.ArgumentParser(
add_help=False,
prog="arktrades",
description="""
Get trades for ticker across all ARK funds.
""",
)
parser.add_argument(
"-l",
"--limit",
help="Limit of rows to show",
dest="limit",
default=10,
type=check_positive,
)
parser.add_argument(
"-s",
"--show_ticker",
action="store_true",
default=False,
help="Flag to show ticker in table",
dest="show_ticker",
)
if other_args and "-" not in other_args[0][0]:
other_args.insert(0, "-l")
ns_parser = parse_known_args_and_warn(
parser, other_args, export_allowed=EXPORT_ONLY_RAW_DATA_ALLOWED
)
if ns_parser:
ark_view.display_ark_trades(
ticker=self.ticker,
num=ns_parser.limit,
export=ns_parser.export,
show_ticker=ns_parser.show_ticker,
)
def test_display_ark_trades_default():
ark_view.display_ark_trades(ticker="TSLA")
def test_display_ark_trades_invalid_ticker():
ark_view.display_ark_trades(ticker="INVALID_TICKER")