fix other python2 test issues

PythonCharmers · Aug 31, 2020 · c051026 · c051026
1 parent f006cad
commit c051026
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/test_past/test_misc.py b/tests/test_past/test_misc.py
@@ -11,7 +11,10 @@
 from contextlib import contextmanager
 
 from future.tests.base import unittest
-from past.builtins import cmp
+from future.utils import PY3
+
+if PY3:
+    from past.builtins import cmp
 
 _dir = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(_dir)
@@ -20,7 +23,7 @@
 
 @contextmanager
 def empty_context_manager(*args, **kwargs):
-    return dict(args=args, kwargs=kwargs)
+    yield dict(args=args, kwargs=kwargs)
 
 
 class TestCmp(unittest.TestCase):