def call_gdpc(self, other_args: List[str]):
"""Process gdpc command"""
parser = argparse.ArgumentParser(
add_help=False,
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
prog="gdpc",
description="""
Get real GDP per capita for United States[Source: Alpha Vantage]
""",
)
parser.add_argument(
"-s",
"--start",
help="Start year.",
dest="start",
type=int,
default=2010,
)
parser.add_argument(
"--raw",
help="Display raw data",
action="store_true",
dest="raw",
default=False,
)
ns_parser = parse_known_args_and_warn(
parser,
other_args,
export_allowed=EXPORT_BOTH_RAW_DATA_AND_FIGURES)
if ns_parser:
alphavantage_view.display_gdp_capita(
start_year=ns_parser.start,
raw=ns_parser.raw,
export=ns_parser.export,
)
def test_display_gdp_capita_no_data(mocker):
# MOCK EXPORT_DATA
mocker.patch(
target=
"gamestonk_terminal.economy.alphavantage_view.alphavantage_model.get_gdp_capita",
return_value=pd.DataFrame(),
)
alphavantage_view.display_gdp_capita(
start_year=2022,
raw=False,
export="",
)
def test_display_gdp_capita(mocker, raw):
# MOCK VISUALIZE_OUTPUT
mocker.patch(
target=
"gamestonk_terminal.helper_classes.TerminalStyle.visualize_output")
# MOCK EXPORT_DATA
mocker.patch(
target="gamestonk_terminal.economy.alphavantage_view.export_data")
alphavantage_view.display_gdp_capita(
start_year=2022,
raw=raw,
export="",
)
def test_display_gdp_capita(mocker, raw, tab):
# MOCK GTFF
mocker.patch.object(target=alphavantage_view.gtff,
attribute="USE_TABULATE_DF",
new=tab)
mocker.patch.object(target=alphavantage_view.gtff,
attribute="USE_ION",
new=True)
# MOCK ION + SHOW
mocker.patch(
target="gamestonk_terminal.stocks.options.yfinance_view.plt.ion")
mocker.patch(
target="gamestonk_terminal.stocks.options.yfinance_view.plt.show")
# MOCK EXPORT_DATA
mocker.patch(
target="gamestonk_terminal.economy.alphavantage_view.export_data")
alphavantage_view.display_gdp_capita(
start_year=2022,
raw=raw,
export="",
)