diff options
Diffstat (limited to 'tests/__init__.py')
| -rw-r--r-- | tests/__init__.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..ccdaff7 --- /dev/null +++ b/tests/__init__.py @@ -0,0 +1,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)) |