Method test_invoke_default

axlearn/cli/utils_test.py:246–288

(self, flags: list[str])

        (["--root_default", "override"],),
    )
    def test_invoke_default(self, flags: list[str]):
        # Test invoking with a flag for which a default already exists.
        args = _parse(
            [
                self.root_module,
                "child1",
                "grandchild1",
                "command1",
                "--required=test1",
            ]
            + flags
        )
        self.assertEqual(
            args.argv,
            [self.root_module, "--required=test1"] + flags + ["--undefok=root,root_default,child1"],
        )
        returncode, stdout, _ = _run(args)
        self.assertEqual(returncode, 0)
        self.assertEqual(stdout, "required: test1, optional: None, root_default: override")

        # If the flags appear after "--", they are ignored, so the default flags should be kept.
        args = _parse(
            [
                self.root_module,
                "child1",
                "grandchild1",
                "command1",
                "--required=test1",
                "--",
            ]
            + flags
        )
        self.assertEqual(
            args.argv,
            [
                self.root_module,
                "--required=test1",
                "--undefok=root,root_default,child1",
                "--root_default=some_value",  # Keep the default value.
                "--",
            ]
            + flags,  # Overrides come after "--".
        )

    def test_filter_argv(self):
        # Test invoking a command which specifies a flag filter.