clean up mkdtemp usage

Committed via https://github.com/asottile/all-repos
This commit is contained in:
Anthony Sottile 2021-03-08 19:12:08 -08:00
parent 79b03ce8db
commit 7a6ec5616d
2 changed files with 7 additions and 11 deletions

View File

@ -2228,7 +2228,6 @@ file:
# content of conftest.py
import os
import shutil
import tempfile
import pytest
@ -2236,12 +2235,11 @@ file:
@pytest.fixture
def cleandir():
old_cwd = os.getcwd()
newpath = tempfile.mkdtemp()
os.chdir(newpath)
yield
os.chdir(old_cwd)
shutil.rmtree(newpath)
with tempfile.TemporaryDirectory() as newpath:
old_cwd = os.getcwd()
os.chdir(newpath)
yield
os.chdir(old_cwd)
and declare its use in a test module via a ``usefixtures`` marker:

View File

@ -1395,12 +1395,10 @@ class TestEarlyRewriteBailout:
**{
"test_setup_nonexisting_cwd.py": """\
import os
import shutil
import tempfile
d = tempfile.mkdtemp()
os.chdir(d)
shutil.rmtree(d)
with tempfile.TemporaryDirectory() as d:
os.chdir(d)
""",
"test_test.py": """\
def test():