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="",
    )
예제 #2
0
    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="",
    )
예제 #4
0
 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"
             )