Ensure that concurrent.future.Future has the right module name (e.g. in exceptions)

Author: pelson

Lib/concurrent/futures/__init__.py

@@ -33,6 +33,12 @@
 )
 
 
+# Remove the _base name from the exposed module.
+Future.__module__ = CancelledError.__module__ = Executor.__module__ = (
+    'concurrent.futures'
+)
+
+
 def __dir__():
     return __all__ + ('__author__', '__doc__')
 

Lib/test/test_concurrent_futures/test_future.py

@@ -134,6 +134,16 @@ def test_repr(self):
                 repr(SUCCESSFUL_FUTURE),
                 '<Future at 0x[0-9a-f]+ state=finished returned int>')
 
+    def test_class_str(self):
+        self.assertEqual(str(futures.Future),
+                         "<class 'concurrent.futures.Future'>")
+        self.assertEqual(str(futures.CancelledError),
+                         "<class 'concurrent.futures.CancelledError'>")
+        self.assertEqual(str(futures.TimeoutError),
+                         "<class 'TimeoutError'>")
+        self.assertEqual(str(futures.Executor),
+                         "<class 'concurrent.futures.Executor'>")
+
     def test_cancel(self):
         f1 = create_future(state=PENDING)
         f2 = create_future(state=RUNNING)