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 test_invalid_shard_options(self): parser = ArgumentParser() parser.parse_args(['--total-shards', '0']) self.assertEqual(parser.exit_status, 2) parser.parse_args(['--total-shards', '-1']) self.assertEqual(parser.exit_status, 2) parser.parse_args(['--total-shards', '5', '--shard-index', '-1']) self.assertEqual(parser.exit_status, 2) parser.parse_args(['--total-shards', '5', '--shard-index', '5']) self.assertEqual(parser.exit_status, 2) parser.parse_args(['--total-shards', '5', '--shard-index', '6']) self.assertEqual(parser.exit_status, 2)
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)