Initialize tmpdir.

2013-06-28 21:34:01 +02:00 · 2013-06-28 21:34:01 +02:00 · 75bbf10d78
commit 75bbf10d78
parent 27c90f07a5
1 changed files with 4 additions and 1 deletions
--- a/tests/test_comics.py
+++ b/tests/test_comics.py
@ -39,9 +39,12 @@ class _ComicTester(TestCase):
            self.url = self.scraperclass.starter()
            # create a temporary directory for images
            self.tmpdir = tempfile.mkdtemp()
+        else:
+            self.tmpdir = None

    def tearDown(self):
-        shutil.rmtree(self.tmpdir)
+        if self.tmpdir is not None:
+            shutil.rmtree(self.tmpdir)

    def get_saved_images(self):
        """Get saved images."""