def test_optparse_options(self): parser = optparse.OptionParser() ArgumentParser.add_option_group(parser, 'foo', discovery=True, running=True, reporting=True, skip='[-d]') options, _ = parser.parse_args(['-j', '1']) self.assertEqual(options.jobs, 1)
def test_valid_shard_options(self): parser = ArgumentParser() parser.parse_args(['--total-shards', '1']) self.assertEqual(parser.exit_status, None) parser.parse_args(['--total-shards', '5', '--shard-index', '4']) self.assertEqual(parser.exit_status, None) parser.parse_args(['--total-shards', '5', '--shard-index', '0']) self.assertEqual(parser.exit_status, None)
def test_argv_from_args_foreign_argument(self): parser = ArgumentParser() parser.add_argument('--some-foreign-argument', default=False, action='store_true') args = parser.parse_args(['--some-foreign-argument', '--verbose']) self.assertEqual(['--verbose'], ArgumentParser().argv_from_args(args))
def check(argv, expected=None): parser = ArgumentParser() args = parser.parse_args(argv) actual_argv = parser.argv_from_args(args) expected = expected or argv self.assertEqual(expected, actual_argv)
def test_optparse_options(self): parser = optparse.OptionParser() ArgumentParser.add_option_group(parser, "foo", discovery=True, running=True, reporting=True, skip="[-d]") options, _ = parser.parse_args(["-j", "1"]) self.assertEqual(options.jobs, 1)