Merge pull request #273 from micahcochran/test-setup-class

unittest: call mkdir & rmtree once for the class
This commit is contained in:
Joris Van den Bossche committed 2016-02-27 00:24:41 +01:00
commit 77b9b3b7e8
1 file changed
+6 -4
+6 -4
View File
@@ -28,12 +28,14 @@ TRAVIS = bool(os.environ.get('TRAVIS', False))
class TestImageComparisons(unittest.TestCase):
def setUp(self):
self.tempdir = tempfile.mkdtemp()
@classmethod
def setUpClass(cls):
cls.tempdir = tempfile.mkdtemp()
return
def tearDown(self):
shutil.rmtree(self.tempdir)
@classmethod
def tearDownClass(cls):
shutil.rmtree(cls.tempdir)
return
def _compare_images(self, ax, filename, tol=10):