cubicweb/devtools/testlib.py
changeset 11744 a6dc650bc230
parent 11725 904ee9cd0cf9
child 11748 c6d37774f4d0
--- a/cubicweb/devtools/testlib.py	Mon Oct 24 09:29:08 2016 +0200
+++ b/cubicweb/devtools/testlib.py	Mon Oct 24 09:31:55 2016 +0200
@@ -26,6 +26,7 @@
 from contextlib import contextmanager
 from inspect import isgeneratorfunction
 from itertools import chain
+import tempfile
 
 from six import text_type, string_types
 from six.moves import range
@@ -56,8 +57,10 @@
     from unittest2 import TestCase
     if not hasattr(TestCase, 'subTest'):
         raise ImportError('no subTest support in available unittest2')
+    from backports.tempfile import TemporaryDirectory  # noqa
 else:
     from unittest import TestCase
+    from tempfile import TemporaryDirectory  # noqa
 
 # in python 2.7, DeprecationWarning are not shown anymore by default
 warnings.filterwarnings('default', category=DeprecationWarning)