def test_remove_empty_folder_from_base_folder_1(self): try: base_folder = tempfile.mkdtemp() removed = DataCleaner.remove_empty_folder_from_base_folder( base_folder) self.assertEqual([], removed) finally: shutil.rmtree(base_folder)
def test_remove_empty_folder_from_base_folder_2(self): try: base_folder = tempfile.mkdtemp() tmp1 = tempfile.mkdtemp(dir=base_folder) tmp2 = tempfile.mkdtemp(dir=base_folder) tmp3 = tempfile.mkdtemp(dir=tmp1) tempfile.mkstemp(dir=tmp2) removed = DataCleaner.remove_empty_folder_from_base_folder( base_folder) self.assertEqual([tmp3, tmp1], removed) finally: shutil.rmtree(base_folder)