Index: tools/testrunner/local/pool_unittest.py |
diff --git a/tools/testrunner/local/pool_unittest.py b/tools/testrunner/local/pool_unittest.py |
index bf2b3f856245677ef749b2332467a31dd4a87e05..335d20a6bf9a71a8efd476cc4b67aa83e812b801 100644 |
--- a/tools/testrunner/local/pool_unittest.py |
+++ b/tools/testrunner/local/pool_unittest.py |
@@ -17,7 +17,7 @@ class PoolTest(unittest.TestCase): |
results = set() |
pool = Pool(3) |
for result in pool.imap_unordered(Run, [[x] for x in range(0, 10)]): |
- results.add(result) |
+ results.add(result.value) |
self.assertEquals(set(range(0, 10)), results) |
def testException(self): |
@@ -25,7 +25,7 @@ class PoolTest(unittest.TestCase): |
pool = Pool(3) |
for result in pool.imap_unordered(Run, [[x] for x in range(0, 12)]): |
# Item 10 will not appear in results due to an internal exception. |
- results.add(result) |
+ results.add(result.value) |
expect = set(range(0, 12)) |
expect.remove(10) |
self.assertEquals(expect, results) |
@@ -34,8 +34,8 @@ class PoolTest(unittest.TestCase): |
results = set() |
pool = Pool(3) |
for result in pool.imap_unordered(Run, [[x] for x in range(0, 10)]): |
- results.add(result) |
- if result < 30: |
- pool.add([result + 20]) |
+ results.add(result.value) |
+ if result.value < 30: |
+ pool.add([result.value + 20]) |
self.assertEquals(set(range(0, 10) + range(20, 30) + range(40, 50)), |
results) |