Skip to content

Commit b1306c4

Browse files
authored
Update test_batched.py
1 parent 210b084 commit b1306c4

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/unit/standalone/test_batched.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,9 +11,9 @@ def test_batched_futures(self):
1111
f.set_result(i)
1212
lst.append(f)
1313
self.assertEqual(batched_futures(lst=lst, n=3, skip_set=set()), [0, 1, 2])
14-
self.assertEqual(batched_futures(lst=lst, skip_set={0, 1, 2}, n=3), [3, 4, 5])
15-
self.assertEqual(batched_futures(lst=lst, skip_set={0, 1, 2, 3, 4, 5}, n=3), [6, 7, 8])
16-
self.assertEqual(batched_futures(lst=lst, skip_set={0, 1, 2, 3, 4, 5, 6, 7, 8}, n=3), [9])
14+
self.assertEqual(batched_futures(lst=lst, skip_set={id(0), id(1), id(2)}, n=3), [3, 4, 5])
15+
self.assertEqual(batched_futures(lst=lst, skip_set={id(0), id(1), id(2), id(3), id(4), id(5)}, n=3), [6, 7, 8])
16+
self.assertEqual(batched_futures(lst=lst, skip_set={id(0), id(1), id(2), id(3), id(4), id(5), id(6), id(7), id(8)}, n=3), [9])
1717

1818
def test_batched_futures_not_finished(self):
1919
lst = []

0 commit comments

Comments
 (0)