blob: ccdaff7736202da96449eba6883ca1b93d817e8c (
plain)
1
2
3
4
5
6
7
8
9
10
|
import atexit
import os
import shutil
import tempfile
if "WORLDGEN_TEST_TMP" not in os.environ:
_root = tempfile.mkdtemp(prefix="worldgen-tests-")
os.environ["WORLDGEN_TEST_TMP"] = os.environ["TMPDIR"] = tempfile.tempdir = _root
_pid = os.getpid()
atexit.register(lambda: os.getpid() == _pid and shutil.rmtree(_root, True))
|