def test_screener(mocker, tab): mocker.patch.object(target=helper_funcs.gtff, attribute="USE_TABULATE_DF", new=tab) finviz_compare_view.screener( similar=["TSLA", "GM"], data_type="overview", export="", )
def call_technical(self, other_args: List[str]): """Process technical command""" parser = argparse.ArgumentParser( add_help=False, formatter_class=argparse.ArgumentDefaultsHelpFormatter, prog="technical", description=""" Prints screener data of similar companies. [Source: Finviz] """, ) 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 finviz_compare_view.screener( ticker=self.ticker, similar=self.similar, data_type="technical", export=ns_parser.export, ) except Exception as e: print(e, "\n")
def test_screener_empty(mocker): target = "gamestonk_terminal.stocks.comparison_analysis.finviz_compare_model.get_comparison_data" mocker.patch(target=target, return_value=pd.DataFrame()) finviz_compare_view.screener( similar=["TSLA", "GM"], data_type="overview", export="", )
def call_technical(self, other_args: List[str]): """Process technical command""" parser = argparse.ArgumentParser( add_help=False, formatter_class=argparse.ArgumentDefaultsHelpFormatter, prog="technical", description=""" Prints screener data of similar companies. [Source: Finviz] """, ) ns_parser = parse_known_args_and_warn(parser, other_args, EXPORT_ONLY_RAW_DATA_ALLOWED) if ns_parser: if self.similar and len(self.similar) > 1: finviz_compare_view.screener( similar=self.similar, data_type="technical", export=ns_parser.export, ) else: console.print( "Please make sure there are more than 1 similar tickers selected. \n" )