def main(): _parser = ArgumentParser() _args = p.create_arguments_main(_parser).parse_args(argv[1:]) _opt = p.check_option(_args) OPTIONS[_opt] if _opt == 1 else None OPTIONS[_opt](_args.filename) if _opt > 1 and path_exists(_args.filename) else print_invalid_options()
def test_all_flags(self): _args = self.parser.parse_args(['-m', '-s', '-f', 'qr_data.txt', '-c']) self.assertEqual(p.check_option(_args), 1)
def test_classify_option(self): _args = self.parser.parse_args(['-f', 'qr_data.txt', '-c']) self.assertEqual(p.check_option(_args), 3)
def test_make_option(self): _args = self.parser.parse_args(['-f', 'qr_data.txt', '-m']) self.assertEqual(p.check_option(_args), 2)
def test_setup_option(self): _args = self.parser.parse_args(['-s']) self.assertEqual(p.check_option(_args), 1)
def test_with_empty_args(self): _args = self.parser.parse_args([]) self.assertEqual(p.check_option(_args), 0)