1 files changed, 15 insertions, 0 deletions
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..c17e6de
--- /dev/null
+++ b/tests/__init__.py
@@ -0,0 +1,15 @@
+import atexit
+import os
+import shutil
+import sys
+import tempfile
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+import worldgen_path  # noqa: E402,F401  (mapgen on sys.path)
+
+if "WORLDMAP_TEST_TMP" not in os.environ:
+    _root = tempfile.mkdtemp(prefix="worldmap-viewer-tests-")
+    os.environ["WORLDMAP_TEST_TMP"] = os.environ["TMPDIR"] = tempfile.tempdir = _root
+    _pid = os.getpid()
+    atexit.register(lambda: os.getpid() == _pid and shutil.rmtree(_root, True))