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