Fix use of mock

sebastinas · sebastinas · commit 18fc6ff2082b · 2020-12-30T23:22:57.000+01:00
diff --git a/bpython/test/__init__.py b/bpython/test/__init__.py
@@ -13,7 +13,7 @@ def setUpClass(cls):
 
 class MagicIterMock(unittest.mock.MagicMock):
 
-    __next__ = mock.Mock(return_value=None)
+    __next__ = unittest.mock.Mock(return_value=None)
 
 
 TEST_CONFIG = os.path.join(os.path.dirname(__file__), "test.config")
