commit a74561a56d58187ebfdad25104486bbcd08e1de6
author Michael Foord <michael@voidspace.org.uk> Sun Mar 25 19:03:13 2012 +0100
committer Michael Foord <michael@voidspace.org.uk> Sun Mar 25 19:03:13 2012 +0100
tree e2b06fe1c13fc755316e7cebc152f64970bd9af7
parent 50a8c0ef5d6c7ae213d86940cc842c0f73fb273a

unittest.mock: set file_spec on first use

Lib/unittest/mock.py

diff --git a/Lib/unittest/mock.py b/Lib/unittest/mock.py
index 9a0bbf0..6ecc4c7 100644
--- a/Lib/unittest/mock.py
+++ b/Lib/unittest/mock.py
@@ -2138,11 +2138,15 @@
     'func_name',
 ])
 
-import _io
-file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
 
+file_spec = None
 
 def mock_open(mock=None, read_data=None):
+    global file_spec
+    if file_spec is None:
+        import _io
+        file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO))))
+
     if mock is None:
         mock = MagicMock(spec=file_spec)