def test_map_sp500_view(mocker):
# MOCK EXPORT_DATA
mock_open = mocker.Mock()
mocker.patch(
target="gamestonk_terminal.economy.finviz_view.webbrowser.open",
new=mock_open)
finviz_view.map_sp500_view(period="1w", map_type="sp500")
mock_open.assert_called_once()
def call_map(self, other_args: List[str]):
"""Process map command"""
parser = argparse.ArgumentParser(
add_help=False,
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
prog="map",
description="""
Performance index stocks map categorized by sectors and industries.
Size represents market cap. Opens web-browser. [Source: Finviz]
""",
)
parser.add_argument(
"-p",
"--period",
action="store",
dest="s_period",
type=str,
default="1d",
choices=["1d", "1w", "1m", "3m", "6m", "1y"],
help="Performance period.",
)
parser.add_argument(
"-t",
"--type",
action="store",
dest="s_type",
type=str,
default="sp500",
choices=["sp500", "world", "full", "etf"],
help="Map filter type.",
)
try:
ns_parser = parse_known_args_and_warn(parser, other_args)
if not ns_parser:
return
finviz_view.map_sp500_view(
period=ns_parser.s_period,
map_type=ns_parser.s_type,
)
except Exception as e:
print(e, "\n")