aboutsummaryrefslogtreecommitdiffziptar.gz
path: root/tests/__init__.py
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))