Merge pull request #168 from AI-SDC/deleting_test_folders

Delete folders after tests

test/test_stata_interface.py

@@ -1,6 +1,7 @@
 """This module contains unit tests for the stata interface."""
 
 import os
+import shutil
 
 import numpy as np
 import pandas as pd
@@ -33,6 +34,15 @@ def data() -> pd.DataFrame:
     return data
 
 
+def clean_up(name):
+    """Removes unwanted files or directory."""
+    if os.path.exists(name):
+        if os.path.isfile(name):
+            os.remove(name)
+        elif os.path.isdir(name):
+            shutil.rmtree(name)
+
+
 def dummy_acrohandler(
     data, command, varlist, exclusion, exp, weights, options
 ):  # pylint:disable=too-many-arguments
@@ -736,3 +746,10 @@ def test_stata_unknown(data):
     )
     correct = "acro command not recognised: foo"
     assert ret == correct, f"got:\n{ret}\nexpected:\n{correct}\n"
+
+
+def test_cleanup():
+    """Gets rid of files created during tests."""
+    names = ["test_outputs", "test_add_to_acro", "sdc_results", "RES_PYTEST"]
+    for name in names:
+        clean_up(name)