test_prot_query.py 文件源码

def test_main(self):

        out_dir = "clue/functional_tests/527ef1c3"

        # Delete out_dir if it exists already
        if os.path.exists(out_dir):
            shutil.rmtree(out_dir)

        config_path = "https://s3.amazonaws.com/data.clue.io/psp/examples/example_user_input.yml"
        second_config_path = "clue/functional_tests/test_prot_query_alt_psp_on_clue.yml"
        args_string = "-u {} -o {} -p {}".format(
            config_path, out_dir, second_config_path).split()
        args = prot_query.build_parser().parse_args(args_string)
        prot_query.main(args)

        self.assertTrue(os.path.exists(os.path.join(out_dir, "INTROSPECT_CONN.gct")))
        self.assertTrue(os.path.exists(os.path.join(out_dir, "CONCATED_CONN.gct")))
        self.assertTrue(os.path.exists(os.path.join(out_dir, "success.txt")))

        # Alt config file only specified 2 cell lines (excludes YAPC)
        self.assertFalse(os.path.exists(os.path.join(out_dir, "YAPC_CONN.gct")))

        # Clean up
        shutil.rmtree(out_dir)