cleanup testing detritus

rstudio · Aug 15, 2023 · eb5d21b · eb5d21b
1 parent 68d85fb
commit eb5d21b
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tests/testthat/setup.R b/tests/testthat/setup.R
@@ -0,0 +1,22 @@
+clean_python_tmp_dir <- function() {
+
+  if(!reticulate::py_available())
+    # python never initialized, nothing to do
+    return()
+
+  python_temp_dir <- dirname(reticulate::py_run_string(
+    "import tempfile; x=tempfile.NamedTemporaryFile().name",
+    local = TRUE
+  )$x)
+
+  detritus <- list.files(
+    path = python_temp_dir,
+    pattern = "__autograph_generated_file|__pycache__",
+    full.names = TRUE,
+    all.files = TRUE
+  )
+
+  unlink(detritus, recursive = TRUE)
+}
+
+withr::defer(clean_python_tmp_dir(), teardown_env())