mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-24 01:50:06 +00:00
36 lines
699 B
Python
36 lines
699 B
Python
from datahub.utilities.delayed_iter import delayed_iter
|
|
|
|
|
|
def test_delayed_iter():
|
|
events = []
|
|
|
|
def maker(n):
|
|
for i in range(n):
|
|
events.append(("add", i))
|
|
yield i
|
|
|
|
for i in delayed_iter(maker(4), 2):
|
|
events.append(("remove", i))
|
|
|
|
assert events == [
|
|
("add", 0),
|
|
("add", 1),
|
|
("add", 2),
|
|
("remove", 0),
|
|
("add", 3),
|
|
("remove", 1),
|
|
("remove", 2),
|
|
("remove", 3),
|
|
]
|
|
|
|
events.clear()
|
|
for i in delayed_iter(maker(2), None):
|
|
events.append(("remove", i))
|
|
|
|
assert events == [
|
|
("add", 0),
|
|
("add", 1),
|
|
("remove", 0),
|
|
("remove", 1),
|
|
]
|