Remove helpers and demonstrate tempfile

Author: christopher-wild

tests/test_helpers.py

@@ -1,6 +1,8 @@
 import unittest
 import os
+import tempfile
 import numpy as np
+import shutil
 import pandas as pd
 from causal_testing.testing.causal_test_suite import CausalTestSuite
 from causal_testing.testing.causal_test_case import CausalTestCase
@@ -10,7 +12,6 @@
 from causal_testing.testing.estimators import LinearRegressionEstimator, LogisticRegressionEstimator
 from causal_testing.specification.causal_specification import CausalSpecification, Scenario
 from causal_testing.data_collection.data_collector import ObservationalDataCollector
-from tests.test_helpers import create_temp_dir_if_non_existent, remove_temp_dir_if_existent
 from causal_testing.specification.causal_dag import CausalDAG
 
 
@@ -29,8 +30,8 @@ def setUp(self) -> None:
         self.scenario = Scenario({A, C, D})
 
         # 2. Create DAG and dummy data and write to csvs
-        temp_dir_path = create_temp_dir_if_non_existent()
-        dag_dot_path = os.path.join(temp_dir_path, "dag.dot")
+        self.temp_dir_path = tempfile.mkdtemp()
+        dag_dot_path = os.path.join(self.temp_dir_path, "dag.dot")
         dag_dot = """digraph G { A -> C; D -> A; D -> C}"""
         with open(dag_dot_path, "w") as file:
             file.write(dag_dot)
@@ -64,6 +65,9 @@ def setUp(self) -> None:
 
         self.data_collector = ObservationalDataCollector(self.scenario, self.df)
 
+    def tearDown(self) -> None:
+        shutil.rmtree(self.temp_dir_path)
+
     def test_adding_test_object(self):
         "test an object can be added to the test_suite using the add_test_object function"
         test_suite = CausalTestSuite()