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="", )