clear the directory for the function (#145)

* clear the directory for the function

* add the unit test

* unit test bug

* yapf

parl/utils/logger.py

@@ -143,7 +143,7 @@ def set_dir(dirname):
         _FILE_HANDLER.close()
         del _FILE_HANDLER
 
-    if not os.path.isdir(dirname):
+    shutil.rmtree(dirname, ignore_errors=True)
     _makedirs(dirname)
     LOG_DIR = dirname
     _set_file(os.path.join(dirname, 'log.log'))

parl/utils/tests/logger_test.py

@@ -15,6 +15,7 @@
 import unittest
 from parl.utils import logger
 import threading as th
+import os.path
 
 
 class TestLogger(unittest.TestCase):
@@ -45,6 +46,15 @@ class TestLogger(unittest.TestCase):
         logger.auto_set_dir(action='n')
         logger.auto_set_dir(action='k')
 
+    def test_set_dir(self):
+        logger.set_dir('./logger_dir')
+        temp_file = './logger_dir/tmp.file'
+        with open(temp_file, 'w') as t_file:
+            t_file.write("Are you OK? From Mr.Lei")
+        self.assertTrue(os.path.isfile(temp_file))
+        logger.set_dir('./logger_dir')
+        self.assertFalse(os.path.isfile(temp_file))
+
 
 if __name__ == '__main__':
     unittest.main()