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",

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,12 @@
 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 +66,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()