Test compatibility with python < 3.8

1 files changed, 18 insertions, 0 deletions

diff --git a/tests/utils.py b/tests/utils.py
new file mode 100644
index 0000000..a56a120
--- /dev/null
+++ b/tests/utils.py
@@ -0,0 +1,18 @@
+import shutil
+import sys
+
+
+def copytree(src, dest, dirs_exist_ok=False):
+    """
+    Helper function to remove existing directories
+
+    Used in the tests for compatibility with python < 3.8
+    """
+    if sys.version_info >= (3, 8):
+        shutil.copytree(src, dest, dirs_exist_ok=dirs_exist_ok)
+    else:
+        if dirs_exist_ok:
+            if not dest.startswith('/tmp'):
+                raise ValueError("Refusing to delete a directory outside /tmp")
+            shutil.rmtree(dest)
+        shutil.copytree(src, dest)