async def test_take_negative() -> None:
xs = AsyncObservable.from_iterable([1, 2, 3, 4, 5])
values = []
async def asend(value) -> None:
values.append(value)
with pytest.raises(ValueError):
take(-1, xs)
async def test_take_normal() -> None:
xs = AsyncObservable.from_iterable([1, 2, 3, 4, 5])
values = []
async def asend(value) -> None:
values.append(value)
ys = take(2, xs)
result = await run(ys, AnonymousAsyncObserver(asend))
assert result == 2
assert values == [1, 2]
async def test_take_empty() -> None:
xs = AsyncObservable.empty()
values = []
async def asend(value) -> None:
values.append(value)
ys = take(42, xs)
with pytest.raises(CancelledError):
await run(ys, AnonymousAsyncObserver(asend))
assert values == []
async def test_take_zero() -> None:
xs = AsyncObservable.from_iterable([1, 2, 3, 4, 5])
values = []
async def asend(value) -> None:
values.append(value)
ys = take(0, xs)
with pytest.raises(CancelledError):
await run(ys, AnonymousAsyncObserver(asend))
assert values == []