cleaning up tests

Author: aysim319

google_symptoms/tests/conftest.py

@@ -72,7 +72,7 @@ def logger():
 def params():
     params = {
         "common": {
-            "export_dir": "./receiving"
+            "export_dir": f"{TEST_DIR}/receiving"
         },
         "indicator": {
             "export_start_date": "2020-02-20",
@@ -105,12 +105,12 @@ def params_w_patch(params):
 
 @pytest.fixture(scope="session")
 def run_as_module(params):
-    if exists("receiving"):
+    if exists(f"{TEST_DIR}/receiving"):
         # Clean receiving directory
-        for fname in listdir("receiving"):
-            remove(join("receiving", fname))
+        for fname in listdir(f"{TEST_DIR}/receiving"):
+            remove(join(f"{TEST_DIR}/receiving", fname))
     else:
-        makedirs("receiving")
+        makedirs(f"{TEST_DIR}/receiving")
 
     with mock.patch("delphi_google_symptoms.pull.initialize_credentials",
                     return_value=None) as mock_credentials:

google_symptoms/tests/test_patch.py

@@ -54,7 +54,7 @@ def side_effect(*args, **kwargs):
             mock_read_gbq.side_effect = side_effect
             start_date = datetime.strptime(params_w_patch["patch"]["start_issue"], "%Y-%m-%d")
 
-            # patch()
+            patch()
 
             patch_path = Path(f"{TEST_DIR}/{params_w_patch['patch']['patch_dir']}")
 

google_symptoms/tests/test_run.py

@@ -9,8 +9,13 @@
 from conftest import TEST_DIR
 
 class TestRun:
-    def test_output_files_exist(self):
-        output_files = listdir("receiving")
+    @classmethod
+    def teardown_class(cls):
+        print('cleaning up tests...')
+        shutil.rmtree(f"{TEST_DIR}/receiving/")
+
+    def test_output_files_exist(self, run_as_module):
+        output_files = listdir(f"{TEST_DIR}/receiving")
         smoothed_files = sorted(list(set([file for file in output_files if "smoothed" in file])))
         raw_files = sorted(list(set([file for file in output_files if "raw" in file])))
         csv_files = {"raw": raw_files, "smoothed": smoothed_files}
@@ -62,11 +67,10 @@ def test_output_files_exist(self):
             csv_dates = list(set([datetime.strptime(f.split('_')[0], "%Y%m%d") for f in csv_files[smther] if smther in f]))
             assert set(csv_files[smther]).issuperset(set(expected_files))
 
-        shutil.rmtree(f"{TEST_DIR}/receiving/")
 
-    def test_output_file_format(self, run_as_module):
+    def test_output_file_format(self):
         df = pd.read_csv(
-            join("receiving", "20200810_state_s03_smoothed_search.csv")
+            join(f"{TEST_DIR}/receiving", "20200810_state_s03_smoothed_search.csv")
         )
         assert (df.columns.values == [
                 "geo_id", "val", "se", "sample_size"]).all()