Function execute_script

tests/test_interpreter.py:540–569

(
        max_shebang_length,  # type: int
        *extra_args  # type: str
    )

    python_exe = sys.executable

    def execute_script(
        max_shebang_length,  # type: int
        *extra_args  # type: str
    ):
        # type: (...) -> str
        with open(os.path.join(str(tmpdir), "script.py"), "w") as fp:
            fp.write(
                dedent(
                    """\
                    {shebang}

                    import json
                    import sys


                    with open(sys.argv[0]) as fp:
                        json.dump(dict(shebang=fp.readline(), argv=sys.argv), sys.stdout)
                    """
                ).format(
                    shebang=create_shebang(
                        python_exe=python_exe, max_shebang_length=max_shebang_length
                    )
                )
            )
        chmod_plus_x(fp.name)
        result = json.loads(
            subprocess.check_output(args=[fp.name] + list(extra_args)).decode("utf-8")
        )
        assert [fp.name] + list(extra_args) == result["argv"]
        return cast(str, result["shebang"])

    assert "#!{python_exe}\n".format(python_exe=python_exe) == execute_script(len(python_exe) * 2)
    assert "#!{python_exe}\n".format(python_exe=python_exe) == execute_script(