Minor import cleanup in tests

wbolster · Oct 18, 2013 · a0ff80a · a0ff80a
1 parent b2e75b4
commit a0ff80a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_plyvel.py b/test/test_plyvel.py
@@ -2,7 +2,7 @@
 
 from __future__ import unicode_literals
 
-from contextlib import contextmanager
+import contextlib
 import os
 import shutil
 import stat
@@ -43,7 +43,7 @@
 # Utilities
 #
 
-@contextmanager
+@contextlib.contextmanager
 def tmp_db(name_prefix, create=True, delete=True, **kwargs):
     name = tempfile.mkdtemp(prefix=name_prefix + '-', dir=TEST_DBS_DIR)
     if create: