Utilize temp_dir from os_helper.

jaraco · May 16, 2024 · b754ab7 · b754ab7
1 parent cdbd58b
commit b754ab7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 12 deletions.
diff --git a/tests/compat/py39.py b/tests/compat/py39.py
@@ -5,6 +5,7 @@
 
 os_helper = try_import('os_helper') or from_test_support(
     'FakePath',
+    'temp_dir',
 )
 
 sys.modules[__name__ + '.os_helper'] = os_helper
diff --git a/tests/test_path.py b/tests/test_path.py
@@ -2,14 +2,12 @@
 import itertools
 import contextlib
 import pathlib
-import tempfile
-import shutil
 import pickle
 import sys
 import unittest
 import zipfile
 
-from .compat.py39.os_helper import FakePath
+from .compat.py39.os_helper import temp_dir, FakePath
 
 import jaraco.itertools
 from jaraco.functools import compose
@@ -63,15 +61,6 @@ def build_alpharep_fixture():
     return zf
 
 
-@contextlib.contextmanager
-def temp_dir():
-    tmpdir = tempfile.mkdtemp()
-    try:
-        yield pathlib.Path(tmpdir)
-    finally:
-        shutil.rmtree(tmpdir)
-
-
 alpharep_generators = [
     Invoked.wrap(build_alpharep_fixture),
     Invoked.wrap(compose(zipp.CompleteDirs.inject, build_alpharep_fixture)),