-

cool-RR · cool-RR · commit b9d57b3dba7b · 2014-01-11T16:48:45.000+02:00
diff --git a/.gitignore b/.gitignore
@@ -17,4 +17,6 @@ coverage.xml
 *.coverage
 .coverage_html_report
 
-_build
+_build
+
+__pycache__
diff --git a/source_py3/python_toolbox/temp_file_tools.py b/source_py3/python_toolbox/temp_file_tools.py
@@ -34,5 +34,6 @@ def create_temp_folder(suffix='', prefix=tempfile.template):
     Use the `suffix` and `prefix` string arguments to dictate a suffix and/or a
     prefix to the temporary folder's name in the filesystem.        
     '''
-    yield pathlib.Path(tempfile.mkdtemp(suffix=suffix, prefix=prefix))
-    shutil.rmtree(self.path)
+    temp_folder = pathlib.Path(tempfile.mkdtemp(suffix=suffix, prefix=prefix))
+    yield temp_folder
+    shutil.rmtree(str(temp_folder))
diff --git a/source_py3/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py b/source_py3/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py
@@ -15,7 +15,7 @@
 
 
 def test_basic():
-    with () as tf1:
+    with create_temp_folder() as tf1:
         assert isinstance(tf1, pathlib.Path)
         assert tf1.exists()
         assert tf1.is_dir()
