def test_arguments(self): parser = argparse.ArgumentParser() cache_updater.add_argparse_options(parser) args = cache_updater.parse_args( parser, ['--shard-total', '10', '--shard-index', '5']) self.assertEqual(args.shard_total, 10) self.assertEqual(args.shard_index, 5) self.assertRaises(SystemExit, cache_updater.parse_args, parser, ['--shard-total', '5', '--shard-index', '10']) self.assertRaises(SystemExit, cache_updater.parse_args, parser, ['--shard-total', '-5', '--shard-index', '10']) self.assertRaises(SystemExit, cache_updater.parse_args, parser, ['--shard-total', '5', '--shard-index', '-10'])
def main(argv): parser = argparse.ArgumentParser( prog="cache_updater", description=sys.modules['__main__'].__doc__, formatter_class=argparse.RawTextHelpFormatter) cache_updater.add_argparse_options(parser) infra_libs.logs.add_argparse_options(parser) args = cache_updater.parse_args(parser, argv) infra_libs.logs.process_argparse_options(args) # Do more processing here LOGGER.info('Cache_updater starting.') cache_updater.run(args.cache_dir, args.shard_index, args.shard_total)
def test_arguments(self): parser = argparse.ArgumentParser() cache_updater.add_argparse_options(parser) args = cache_updater.parse_args( parser, ['--shard-total', '10', '--shard-index', '5']) self.assertEqual(args.shard_total, 10) self.assertEqual(args.shard_index, 5) self.assertRaises( SystemExit, cache_updater.parse_args, parser, ['--shard-total', '5', '--shard-index', '10']) self.assertRaises( SystemExit, cache_updater.parse_args, parser, ['--shard-total', '-5', '--shard-index', '10']) self.assertRaises( SystemExit, cache_updater.parse_args, parser, ['--shard-total', '5', '--shard-index', '-10'])