@@ -74,40 +74,41 @@ def test_map_timeout(self):
74
74
self .assertEqual ([None , None ], results )
75
75
76
76
def test_map_with_buffersize (self ):
77
+ iterable = range (4 )
77
78
with self .assertRaisesRegex (ValueError , "buffersize must be None or >= 1." ):
78
- self .executor .map (bool , [], buffersize = 0 )
79
-
80
- it = range (4 )
79
+ self .executor .map (bool , iterable , buffersize = 0 )
80
+ self .assertEqual (
81
+ list (self .executor .map (str , iterable , buffersize = 1 )),
82
+ list (map (str , iterable )),
83
+ )
81
84
self .assertEqual (
82
- list (self .executor .map (str , it , buffersize = 1 )),
83
- list (map (str , it )),
85
+ list (self .executor .map (str , iterable , buffersize = 2 )),
86
+ list (map (str , iterable )),
84
87
)
85
88
86
89
def test_map_with_buffersize_and_timeout (self ):
87
- it = self .executor .map (time .sleep , (0 , 1 ), timeout = 0.5 )
88
- next (it )
90
+ results = self .executor .map (time .sleep , (0 , 1 ), timeout = 0.5 )
91
+ next (results )
89
92
with self .assertRaises (TimeoutError ):
90
- next (it )
93
+ next (results )
91
94
92
95
def test_map_with_buffersize_on_infinite_iterable (self ):
93
96
results = self .executor .map (str , itertools .count (1 ), buffersize = 1 )
94
97
self .assertEqual (next (iter (results )), "1" )
95
98
96
99
def test_map_with_buffersize_on_iterable_smaller_than_buffer (self ):
97
- it = range (2 )
98
- results = self .executor .map (str , it , buffersize = 10 )
99
- self .assertListEqual (list (results ), list (map (str , it )))
100
+ iterable = range (2 )
101
+ results = self .executor .map (str , iterable , buffersize = 8 )
102
+ self .assertListEqual (list (results ), list (map (str , iterable )))
100
103
101
104
def test_map_with_buffersize_on_empty_iterable (self ):
102
- it = iter ([])
103
- results = self .executor .map (str , it , buffersize = 10 )
105
+ results = self .executor .map (str , [], buffersize = 8 )
104
106
self .assertListEqual (list (results ), [])
105
107
106
108
def test_map_with_buffersize_when_buffer_becomes_full (self ):
107
- manager = multiprocessing .Manager ()
108
109
iterable = range (8 )
109
110
buffersize = 4
110
- buffered_results = manager .list ()
111
+ buffered_results = multiprocessing . Manager () .list ()
111
112
self .executor .map (buffered_results .append , iterable , buffersize = buffersize )
112
113
self .executor .shutdown (wait = True )
113
114
self .assertSetEqual (
0 commit comments