Function write_pth

tests/integration/test_issue_1025.py:26–33

(
        pth_path,  # type: str
        sitedir,  # type: str
    )

    cleanups = []

    def write_pth(
        pth_path,  # type: str
        sitedir,  # type: str
    ):
        # type: (...) -> None
        cleanups.append(lambda: safe_delete(pth_path))
        with open(pth_path, "w") as fp:
            fp.write("import site; site.addsitedir({!r})\n".format(sitedir))

    try:
        yield write_pth