def test_train_r3(tmpdir): save_path = tmpdir.mkdir('save_dir') args = ['--n-folds', '3', '--n-epochs', '2', '--save-path', str(save_path), os.path.join(data_dir, 'n1000_r3/')] args = Train_CMD.parse_args(args) Train_CMD.main(args)
def test_train(tmpdir, caplog): caplog.set_level(logging.INFO) save_path = tmpdir.mkdir('save_dir') args = [ '--n-folds', '2', '--n-epochs', '2', '--force-overwrite', '--save-path', str(save_path), os.path.join(data_dir, 'n10_r2/feature_files.tsv') ] args = Train_CMD.parse_args(args) Train_CMD.main(args)
def test_train_r3_pkl_only(tmpdir): save_path = tmpdir.mkdir('save_dir') args = ['--n-folds', '3', '--n-epochs', '2', '--pickle-only', '--save-path', str(save_path), os.path.join(data_dir, 'n1000_r3/')] args = Train_CMD.parse_args(args) with pytest.raises(SystemExit) as pytest_wrapped_e: Train_CMD.main(args) assert pytest_wrapped_e.type == SystemExit assert pytest_wrapped_e.value.code == 0
def test_train_pkl_only(tmpdir, caplog): caplog.set_level(logging.INFO) save_path = tmpdir.mkdir('save_dir') args = [ '--n-folds', '2', '--n-epochs', '2', '--pickle-only', '--force-overwrite', '--save-path', str(save_path), os.path.join(data_dir, 'n10_r2/feature_files.tsv') ] args = Train_CMD.parse_args(args) with pytest.raises(SystemExit) as pytest_wrapped_e: Train_CMD.main(args) assert pytest_wrapped_e.type == SystemExit assert pytest_wrapped_e.value.code == 0
def main(args=None): """Main entry point for application """ if args is None: args = sys.argv[1:] desc = 'DeepMAsED: Deep learning for Metagenome Assembly Error Detection' epi = """DESCRIPTION: Usage: DeepMAsED <subcommand> <subcommand_params> Example: DeepMAsED train -h For general info, see https://github.com/leylabmpi/DeepMAsED/ """ parser = argparse.ArgumentParser( description=desc, epilog=epi, formatter_class=argparse.RawTextHelpFormatter) parser.add_argument('--version', action='version', version=__version__) # subparsers subparsers = parser.add_subparsers() ## train train = Train.parse_args(subparsers=subparsers) train.set_defaults(func=Train.main) ## predict predict = Predict.parse_args(subparsers=subparsers) predict.set_defaults(func=Predict.main) ## evaluate evaluate = Evaluate.parse_args(subparsers=subparsers) evaluate.set_defaults(func=Evaluate.main) ## evaluate features = Features.parse_args(subparsers=subparsers) features.set_defaults(func=Features.main) # parsing args if args: args = parser.parse_args(args) else: args = parser.parse_args() # running subcommands if len(vars(args)) > 0: args.func(args) else: parser.parse_args(['--help'])
def main(args=None): """Main entry point for application """ if args is None: args = sys.argv[1:] desc = 'Template python script' epi = """DESCRIPTION: Test """ parser = argparse.ArgumentParser(description=desc, epilog=epi, formatter_class=argparse.RawTextHelpFormatter) parser.add_argument('--version', action='version', version='0.0.1') # subparsers subparsers = parser.add_subparsers() ## train train = Train.parse_args(subparsers=subparsers) train.set_defaults(func=Train.main) ## predict predict = Predict.parse_args(subparsers=subparsers) predict.set_defaults(func=Predict.main) ## evaluate evaluate = Evaluate.parse_args(subparsers=subparsers) evaluate.set_defaults(func=Evaluate.main) # parsing args if args: args = parser.parse_args(args) else: args = parser.parse_args() # running subcommands if len(vars(args)) > 0: args.func(args) else: parser.parse_args(['--help'])
def test_help(): args = ['-h'] with pytest.raises(SystemExit) as pytest_wrapped_e: Train_CMD.parse_args(args) assert pytest_wrapped_e.type == SystemExit assert pytest_wrapped_e.value.code == 0