Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use parse_arguments method in tempest

Best Python code snippet using tempest_python

test_cli.py

Source:test_cli.py

...7class TestCliParseArguments(unittest.TestCase):8    """Test the command line interface and argument parsing."""9    def test_default_task_is_task_type(self):10        """Default task type should be 'task'."""11        parsed = cli.parse_arguments(['TestTask'])12        self.assertEqual(parsed.task_type, TwoDoTask.TASK_TYPE)13    def test_set_task_type_project(self):14        """A task can be a 'project' type: -t 1."""15        args = "TestTask -t 1".split()16        parsed = cli.parse_arguments(args)17        self.assertEqual(parsed.task_type, TwoDoTask.PROJECT_TYPE)18    def test_set_task_type_project_long_option(self):19        """A task can be a 'project' type: --type 1."""20        args = "TestTask --type 1".split()21        parsed = cli.parse_arguments(args)22        self.assertEqual(parsed.task_type, TwoDoTask.PROJECT_TYPE)23    def test_set_task_type_checklist(self):24        """A task can be a 'checklist' type: -t 2."""25        args = "TestTask -t 2".split()26        parsed = cli.parse_arguments(args)27        self.assertEqual(parsed.task_type, TwoDoTask.CHECKLIST_TYPE)28    def test_set_task_type_checklist_long_option(self):29        """A task can be a 'checklist' type: --type 2."""30        args = "TestTask --type 2".split()31        parsed = cli.parse_arguments(args)32        self.assertEqual(parsed.task_type, TwoDoTask.CHECKLIST_TYPE)33    def test_set_list_name(self):34        """A task can belong to a list: -l listname."""35        args = "TestTask -l business".split()36        parsed = cli.parse_arguments(args)37        self.assertEqual(parsed.for_list, 'business')38    def test_set_project_name(self):39        """A task can be a subtask in a project."""40        args = "TestTask -l business --project webpage".split()41        parsed = cli.parse_arguments(args)42        self.assertEqual(parsed.for_list, 'business')43        self.assertEqual(parsed.in_project, 'webpage')44    def test_set_list_name_long_option(self):45        """A task can belong to a list: --list listname."""46        args = "TestTask --list business".split()47        parsed = cli.parse_arguments(args)48        self.assertEqual(parsed.for_list, 'business')49    def test_set_note_of_task(self):50        """A task can have a note: -n "some notes to the task ..."."""51        args = [52            'TestTask',53            '-n',54            'This is a note for the task.'55        ]56        parsed = cli.parse_arguments(args)57        self.assertEqual(parsed.note, 'This is a note for the task.')58    def test_set_note_of_task_long_option(self):59        """A task can have a note: --note "some notes to the task ..."."""60        args = [61            'TestTask',62            '--note',63            'This is a note for the task.'64        ]65        parsed = cli.parse_arguments(args)66        self.assertEqual(parsed.note, 'This is a note for the task.')67    def test_set_priority(self):68        """Priority of a task can be set: -p 2."""69        args = "TestTask -p 2".split()70        parsed = cli.parse_arguments(args)71        self.assertEqual(parsed.priority, '2')72    def test_set_priority_long_option(self):73        """Priority of a task can be set: --priority 2."""74        args = "TestTask --priority 2".split()75        parsed = cli.parse_arguments(args)76        self.assertEqual(parsed.priority, '2')77    def test_not_starring_task_is_default(self):78        """The default task is not starred."""79        args = ["TestTask"]80        parsed = cli.parse_arguments(args)81        self.assertFalse(parsed.starred)82    def test_starr_task(self):83        """Starr a task with: -s."""84        args = "TestTask -s".split()85        parsed = cli.parse_arguments(args)86        self.assertTrue(parsed.starred)87    def test_starr_task_long_option(self):88        """Starr a task with: --starred."""89        args = "TestTask --starred".split()90        parsed = cli.parse_arguments(args)91        self.assertTrue(parsed.starred)92    def test_set_tags_in_task(self):93        """A Task can have tags."""94        args = "TestTask --tags business,customer,important".split()95        parsed = cli.parse_arguments(args)96        self.assertEqual(parsed.tags, "business,customer,important")97    def test_get_taskid(self):98        """Print the task id"""99        args = "TestTask --taskid -l mylist".split()100        parsed = cli.parse_arguments(args)101        self.assertEqual(parsed.taskid, True)102    def test_set_due_date_isoformat(self):103        """Set the tasks due date: -d YYYY-MM-DD."""104        args = "TestTask -d 2015-09-10".split()105        parsed = cli.parse_arguments(args)106        self.assertEqual(parsed.due, "2015-09-10")107    def test_use_parenttask(self):108        """Set the parent task of a subtask"""109        args = "SubTask -f Parenttask -l mylist".split()110        parsed = cli.parse_arguments(args)111        self.assertEqual(parsed.for_parent_name, "Parenttask")112    def test_set_due_date_isoformat_long_option(self):113        """Set the tasks due date: --due YYYY-MM-DD."""114        args = "TestTask --due 2015-09-10".split()115        parsed = cli.parse_arguments(args)116        self.assertEqual(parsed.due, "2015-09-10")117    def test_set_due_date_n_format(self):118        """Set the tasks due date: -d 2."""119        args = "TestTask -d 2".split()120        parsed = cli.parse_arguments(args)121        self.assertEqual(parsed.due, "2")122    def test_set_due_date_n_format_long_option(self):123        """Set the tasks due date: --due 14."""124        args = "TestTask --due 14".split()125        parsed = cli.parse_arguments(args)126        self.assertEqual(parsed.due, "14")127    def test_set_duetime(self):128        """Set the tasks due time: --dueTime '2015-09-10 12:00'."""129        args = "TestTask --dueTime".split()130        args.append('2015-09-10 12:00')131        parsed = cli.parse_arguments(args)132        self.assertEqual(parsed.dueTime, "2015-09-10 12:00")133    def test_set_start_isoformat(self):134        """Set the tasks start time: --start '2015-09-10 12:00'."""135        args = "TestTask --start".split()136        args.append('2015-09-10 12:00')137        parsed = cli.parse_arguments(args)138        self.assertEqual(parsed.start, "2015-09-10 12:00")139    def test_set_start_n_format(self):140        """Set the tasks start time: --start 7."""141        args = "TestTask --start 7".split()142        parsed = cli.parse_arguments(args)143        self.assertEqual(parsed.start, "7")144    def test_repeat_task(self):145        """Create a repeating task: --repeat 2."""146        args = "TestTask --repeat 2".split()147        parsed = cli.parse_arguments(args)148        self.assertEqual(parsed.repeat, "2")149    def test_ignoredefaults_default_to_false(self):150        """Do not ignore defaults."""151        parsed = cli.parse_arguments(['TestTask'])152        self.assertFalse(parsed.ignoreDefaults)153    def test_set_ignoredefaults(self):154        """Ignore defaults: -i."""155        args = "TestTask -i".split()156        parsed = cli.parse_arguments(args)157        self.assertTrue(parsed.ignoreDefaults)158    def test_set_ignoredefaults_long_option(self):159        """Ignore defaults: --ignoerDefaults."""160        args = "TestTask --ignoreDefaults".split()161        parsed = cli.parse_arguments(args)162        self.assertTrue(parsed.ignoreDefaults)163    def test_missing_args_raise_system_exit(self):164        """Raise SystemExit, if args are missing."""165        with self.assertRaises(SystemExit):166            cli.main([])167class TestCliGeneratesCorrectTwoDoTaskObject(unittest.TestCase):168    """Command line args should create correct task object."""169    def test_create_simple_task(self):170        """Create a task with a title."""171        parsed = cli.parse_arguments(['TestTask'])172        task = TwoDoTask(**vars(parsed))173        self.assertEqual(task.task, 'TestTask')174    def test_create_simple_task_has_correct_defaults(self):175        """Create a task with correct defaults."""176        parsed = cli.parse_arguments(['TestTask'])177        task = TwoDoTask(**vars(parsed))178        self.assertEqual(task.task, 'TestTask')179        self.assertEqual(task.type, TwoDoTask.TASK_TYPE)180        self.assertIsNone(task.for_list)181        self.assertIsNone(task.forParentName)182        self.assertIsNone(task.note)183        self.assertEqual(task.priority, '0')184        self.assertEqual(task.starred, '0')185        self.assertIsNone(task.tags)186        self.assertIsNone(task.due)187        self.assertIsNone(task.dueTime)188        self.assertIsNone(task.start)189        self.assertIsNone(task.repeat)190        self.assertIsNone(task.action)191        self.assertEqual(task.ignoreDefaults, '0')192    def test_task_has_repeat_and_priority(self):193        """Create a task with weekly repetition and high priority."""194        args = "TestTask --repeat 1 --priority 2".split()195        parsed = cli.parse_arguments(args)196        task = TwoDoTask(**vars(parsed))197        self.assertEqual(task.task, 'TestTask')198        self.assertEqual(task.type, TwoDoTask.TASK_TYPE)199        self.assertIsNone(task.for_list)200        self.assertIsNone(task.forParentName)201        self.assertIsNone(task.note)202        self.assertEqual(task.priority, '2')203        self.assertEqual(task.starred, '0')204        self.assertIsNone(task.tags)205        self.assertIsNone(task.due)206        self.assertIsNone(task.dueTime)207        self.assertIsNone(task.start)208        self.assertEqual(task.repeat, '1')209        self.assertIsNone(task.action)210        self.assertEqual(task.ignoreDefaults, '0')211    def test_task_is_starred(self):212        """Create a starred task."""213        args = "TestTask -s".split()214        parsed = cli.parse_arguments(args)215        task = TwoDoTask(**vars(parsed))216        self.assertEqual(task.task, 'TestTask')217        self.assertEqual(task.starred, '1')218    def test_set_task_to_ignore_defaults(self):219        """Create a task, which ignores date and time defaults."""220        args = "TestTask -i".split()221        parsed = cli.parse_arguments(args)222        task = TwoDoTask(**vars(parsed))223        self.assertEqual(task.task, 'TestTask')224        self.assertEqual(task.ignoreDefaults, '1')225    def test_call_of_cli_generates_output(self):226        """Call command generates no output."""227        msg = subprocess.check_output(['python', '-m', 'twodolib.cli', 'test'])228        self.assertGreater(len(msg), 0)229    def test_task_gets_action_long(self):230        """Create a task with an action."""231        args = "TestTask --action url:https://www.2doapp.com".split()232        parsed = cli.parse_arguments(args)233        task = TwoDoTask(**vars(parsed))234        self.assertEqual(task.task, 'TestTask')235        self.assertEqual(task.type, TwoDoTask.TASK_TYPE)236        self.assertIsNone(task.for_list)237        self.assertIsNone(task.forParentName)238        self.assertIsNone(task.note)239        self.assertEqual(task.priority, '0')240        self.assertEqual(task.starred, '0')241        self.assertIsNone(task.tags)242        self.assertIsNone(task.due)243        self.assertIsNone(task.dueTime)244        self.assertIsNone(task.start)245        self.assertIsNone(task.repeat)246        self.assertIsNotNone(task.action)247        self.assertEqual(task.ignoreDefaults, '0')248    def test_task_gets_action_short(self):249        """Create a task with an action."""250        args = "TestTask -a url:https://www.2doapp.com".split()251        parsed = cli.parse_arguments(args)252        task = TwoDoTask(**vars(parsed))253        self.assertEqual(task.task, 'TestTask')254        self.assertEqual(task.type, TwoDoTask.TASK_TYPE)255        self.assertIsNone(task.for_list)256        self.assertIsNone(task.forParentName)257        self.assertIsNone(task.note)258        self.assertEqual(task.priority, '0')259        self.assertEqual(task.starred, '0')260        self.assertIsNone(task.tags)261        self.assertIsNone(task.due)262        self.assertIsNone(task.dueTime)263        self.assertIsNone(task.start)264        self.assertIsNone(task.repeat)265        self.assertIsNotNone(task.action)...

test_parse_start_runner_arguments.py

Source:test_parse_start_runner_arguments.py

1from mock import ANY, call2from hypernode_vagrant_runner.commands import parse_start_runner_arguments3from hypernode_vagrant_runner.settings import HYPERNODE_VAGRANT_PHP_VERSIONS, \4    HYPERNODE_VAGRANT_DEFAULT_USER, HYPERNODE_VAGRANT_DEFAULT_PHP_VERSION, \5    HYPERNODE_VAGRANT_USERS6from tests.testcase import TestCase7class TestParseStartRunnerArguments(TestCase):8    def setUp(self):9        self.argument_parser = self.set_up_patch('hypernode_vagrant_runner.commands.ArgumentParser')10        self.parse_arguments = self.set_up_patch('hypernode_vagrant_runner.commands.parse_arguments')11    def test_parse_start_runner_arguments_instantiates_argument_parser(self):12        parse_start_runner_arguments()13        self.argument_parser.assert_called_once_with(14            prog='hypernode-vagrant-runner',15            description=ANY16        )17    def test_parse_start_runner_arguments_parses_arguments(self):18        parse_start_runner_arguments()19        self.parse_arguments.assert_called_once_with(20            self.argument_parser.return_value21        )22    def test_parse_start_runner_adds_run_once_flag_with_default_false(self):23        parse_start_runner_arguments()24        expected_call = call(25            '--run-once', '-1',26            action='store_true',27            help=ANY28        )29        self.assertIn(30            expected_call,31            self.argument_parser.return_value.add_argument.mock_calls32        )33    def test_parse_start_runner_adds_project_path_flag(self):34        parse_start_runner_arguments()35        expected_call = call(36            '--project-path',37            help=ANY38        )39        self.assertIn(40            expected_call,41            self.argument_parser.return_value.add_argument.mock_calls42        )43    def test_parse_start_runner_adds_command_to_run_flag(self):44        parse_start_runner_arguments()45        expected_call = call(46            '--command-to-run', '-c',47            help=ANY48        )49        self.assertIn(50            expected_call,51            self.argument_parser.return_value.add_argument.mock_calls52        )53    def test_parse_start_runner_adds_path_flag_for_a_pre_existing_checkout_directory(self):54        parse_start_runner_arguments()55        expected_call = call(56            '--pre-existing-vagrant-path', '-p',57            help=ANY58        )59        self.assertIn(60            expected_call,61            self.argument_parser.return_value.add_argument.mock_calls62        )63    def test_parse_start_runner_adds_php_flag(self):64        parse_start_runner_arguments()65        expected_call = call(66            '--php',67            help=ANY,68            choices=HYPERNODE_VAGRANT_PHP_VERSIONS,69            default=HYPERNODE_VAGRANT_DEFAULT_PHP_VERSION70        )71        self.assertIn(72            expected_call,73            self.argument_parser.return_value.add_argument.mock_calls74        )75    def test_parse_start_runner_adds_enable_xdebug_flag(self):76        parse_start_runner_arguments()77        expected_call = call(78            '--enable-xdebug',79            action='store_true',80            help=ANY,81        )82        self.assertIn(83            expected_call,84            self.argument_parser.return_value.add_argument.mock_calls85        )86    def test_parse_start_runner_adds_skip_try_sudo_flag(self):87        parse_start_runner_arguments()88        expected_call = call(89            '--skip-try-sudo',90            action='store_true',91            help=ANY,92        )93        self.assertIn(94            expected_call,95            self.argument_parser.return_value.add_argument.mock_calls96        )97    def test_parse_start_runner_adds_user_flag(self):98        parse_start_runner_arguments()99        expected_call = call(100            '--user',101            help=ANY,102            choices=HYPERNODE_VAGRANT_USERS,103            default=HYPERNODE_VAGRANT_DEFAULT_USER104        )105        self.assertIn(106            expected_call,107            self.argument_parser.return_value.add_argument.mock_calls108        )109    def test_parse_start_runner_adds_xenial_flag_with_default_false(self):110        parse_start_runner_arguments()111        expected_call = call(112            '--xenial',113            action='store_true',114            help=ANY115        )116        self.assertIn(117            expected_call,118            self.argument_parser.return_value.add_argument.mock_calls119        )120    def test_parse_start_runner_arguments_returns_parsed_arguments(self):121        ret = parse_start_runner_arguments()122        self.assertEqual(ret, self.parse_arguments.return_value)123    def test_parse_start_runner_arguments_errors_when_php56_and_precise(self):124        self.parse_arguments.return_value.php = '5.6'125        self.parse_arguments.return_value.xenial = False126        parse_start_runner_arguments()127        self.argument_parser.return_value.error.assert_called_once_with(ANY)128    def test_parse_start_runner_arguments_errors_when_php71_and_precise(self):129        self.parse_arguments.return_value.php = '7.1'130        self.parse_arguments.return_value.xenial = False131        parse_start_runner_arguments()132        self.argument_parser.return_value.error.assert_called_once_with(ANY)133    def test_parse_start_runner_arguments_errors_when_php72_and_precise(self):134        self.parse_arguments.return_value.php = '7.2'135        self.parse_arguments.return_value.xenial = False136        parse_start_runner_arguments()137        self.argument_parser.return_value.error.assert_called_once_with(ANY)138    def test_parse_start_runner_arguments_does_not_error_when_php55_and_precise(self):139        self.parse_arguments.return_value.php = '5.5'140        self.parse_arguments.return_value.xenial = False141        parse_start_runner_arguments()142        self.assertFalse(self.argument_parser.return_value.error.called)143    def test_parse_start_runner_arguments_does_not_error_if_php56_and_xenial(self):144        self.parse_arguments.return_value.php = '5.6'145        parse_start_runner_arguments()146        self.assertFalse(self.argument_parser.return_value.error.called)147    def test_parse_start_runner_arguments_does_not_error_if_php71_and_xenial(self):148        self.parse_arguments.return_value.php = '7.1'149        parse_start_runner_arguments()150        self.assertFalse(self.argument_parser.return_value.error.called)151    def test_parse_start_runner_arguments_does_not_error_if_php72_and_xenial(self):152        self.parse_arguments.return_value.php = '7.2'153        parse_start_runner_arguments()...

test_link_checker.py

Source:test_link_checker.py

...4def test_parser_shared():5    subcmds = ["deeds", "legalcode", "rdf", "index", "combined", "canonical"]6    # Test defaults7    for subcmd in subcmds:8        args = link_checker.parse_arguments([subcmd])9        assert args.limit == 010        assert args.log_level == 3011        assert args.root_url == "https://creativecommons.org"12    # Test arguments13    for subcmd in subcmds:14        # Test --limit15        args = link_checker.parse_arguments([subcmd, "--limit", "10"])16        assert args.limit == 1017        args = link_checker.parse_arguments([subcmd, "--limit=100"])18        assert args.limit == 10019        # Test Logging Levels -q/--quiet20        args = link_checker.parse_arguments([subcmd, "-q"])21        assert args.log_level == 4022        args = link_checker.parse_arguments([subcmd, "-qq"])23        assert args.log_level == 5024        args = link_checker.parse_arguments([subcmd, "-qqq"])25        assert args.log_level == 5026        args = link_checker.parse_arguments([subcmd, "-q", "--quiet"])27        assert args.log_level == 5028        # Test Logging Levels -v/--verbose29        args = link_checker.parse_arguments([subcmd, "-v"])30        assert args.log_level == 2031        args = link_checker.parse_arguments([subcmd, "-vv"])32        assert args.log_level == 1033        args = link_checker.parse_arguments([subcmd, "-vvv"])34        assert args.log_level == 1035        args = link_checker.parse_arguments([subcmd, "-v", "--verbose"])36        assert args.log_level == 1037        # Test Logging Levels with both -v and -q38        args = link_checker.parse_arguments([subcmd, "-vq"])39        assert args.log_level == 3040        args = link_checker.parse_arguments([subcmd, "-vvq"])41        assert args.log_level == 2042        args = link_checker.parse_arguments([subcmd, "-vqq"])43        assert args.log_level == 4044        # Test --root-url45        args = link_checker.parse_arguments(46            [subcmd, "--root-url", "https://pytest.creativecommons.org"]47        )48        assert args.root_url == "https://pytest.creativecommons.org"49def test_parser_shared_licenses():50    subcmds = ["deeds", "legalcode", "rdf", "combined", "canonical"]51    # Test defaults52    for subcmd in subcmds:53        args = link_checker.parse_arguments([subcmd])54        assert args.local is False55    # Test argumetns56    for subcmd in subcmds:57        # Test --local58        args = link_checker.parse_arguments([subcmd, "--local"])59        assert args.local is True60def test_parser_shared_rdf():61    subcmds = ["rdf", "index"]62    # Test defaults63    for subcmd in subcmds:64        args = link_checker.parse_arguments([subcmd])65        assert args.local_index is False66    # Test argumetns67    for subcmd in subcmds:68        # Test --local69        args = link_checker.parse_arguments([subcmd, "--local-index"])70        assert args.local_index is True71def test_parser_shared_reporting(tmpdir):72    subcmds = ["deeds", "legalcode", "rdf", "index", "combined"]73    # Test defaults74    for subcmd in subcmds:75        args = link_checker.parse_arguments([subcmd])76        assert bool(args.output_errors) is False77    # Test arguments78    for subcmd in subcmds:79        # Test --output-errors with default value80        args = link_checker.parse_arguments([subcmd, "--output-errors"])81        assert bool(args.output_errors) is True82        assert args.output_errors.name == "errorlog.txt"83        # Test --output-errors with custom value84        output_file = tmpdir.join("errorlog.txt")85        args = link_checker.parse_arguments(86            [subcmd, "--output-errors", output_file.strpath]87        )88        assert bool(args.output_errors) is True...

test_parse_arguments.py

Source:test_parse_arguments.py

1from cgitb import reset2from time import time3import unittest4from api.parse_arguments import Parse_arguments5import pandas as pd6from datetime import datetime, timedelta7import calendar8class TestQuery(unittest.TestCase):9    def test_parse_date(self):10        today = datetime.now()11        lst = [str(today), '395', '4662']12        parse_arguments = Parse_arguments(lst)13        result = parse_arguments.parse_date()14        self.assertTrue(type(result) == pd._libs.tslibs.timestamps.Timestamp)15        self.assertEquals(result, pd.to_datetime(today))16    def test_extract_day_of_week(self):17        18        today = datetime.now()19        lst = [str(today), '395', '4662']20        parse_arguments = Parse_arguments(lst)21        result = parse_arguments.extract_day_of_week()22        self.assertTrue(type(result) == str)23        self.assertEquals(result, calendar.day_name[today.weekday()])24        25    def test_transform_to_forecast_date(self):26        tomorrow = datetime.now() + timedelta(days=1)27        hour = tomorrow.hour28        # find out which forecast time it is closest to29        forecast_hours = [0, 6, 12, 18]30        diff = 631        set_hour = 032        for fh in forecast_hours:33            if abs(hour - fh) < diff:34                diff = abs(hour - fh)35                set_hour = fh36        tomorrow = tomorrow.replace(hour=set_hour, minute=00, second=00, microsecond=00)37        38        lst = [str(tomorrow), '395', '4662']39        parse_arguments = Parse_arguments(lst)40        result = parse_arguments.transform_to_forecast_date()41        print("result: ", result)42        print("tomorrow: ", tomorrow)43        self.assertTrue(type(result) == pd._libs.tslibs.timestamps.Timestamp)44        self.assertEquals(result, tomorrow)45        46    def test_get_beginningstop(self):47        today = datetime.now()48        lst = [str(today), '395', '4662']49        parse_arguments = Parse_arguments(lst)50        result = parse_arguments.get_beginningstop()51        52        self.assertTrue(type(result) == str)53        self.assertEquals(result, '395')54    def test_get_endingstop(self):55        56        today = datetime.now()57        lst = [str(today), '395', '4662']58        parse_arguments = Parse_arguments(lst)59        result = parse_arguments.get_endingstop()60        61        self.assertTrue(type(result) == str)62        self.assertEquals(result, '4662')63    def test_get_lineid(self):64        today = datetime.now()65        lst = [str(today), '395', '4662']66        parse_arguments = Parse_arguments(lst)67        result = parse_arguments.get_lineid('77A')68        self.assertTrue(type(result) == str)69        self.assertEquals(result, '77A')70if __name__== '__main__':...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.