tests add cleanup

tnextday · Sep 24, 2022 · f381de8 · f381de8
1 parent 7f867a6
commit f381de8
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/tests.py b/tests.py
@@ -34,9 +34,6 @@ def check_recovery(self, filepath):
     def check_exists(self, filepath):
         self.assertTrue(Path(self.base_dir.join(filepath)).exists())
 
-    def tearDown(self):
-        shutil.rmtree(str(self.base_dir))
-
 
 class TestAnimeRenamer(unittest.TestCase):
 
@@ -109,13 +106,14 @@ def test_regexp_rename(self):
     def test_recovery(self):
         pass
 
-    def setUp(self):
-        pass
-
-    def teardown(self):
-        # shutil.rmtree(_test_dir)
-        pass
+    def setup(self):
+        print ("This is setUp")
 
+    def tearDown(self):
+        try:
+            shutil.rmtree(_test_dir)
+        except:
+            pass
 
 if __name__ == '__main__':
     unittest.main()