mirror of
https://github.com/panda3d/panda3d.git
synced 2025-09-25 05:57:15 -04:00
tests: Add unit tests for send() and throw() on generator wrapper
This commit is contained in:
parent
2625c19ccf
commit
469df73ce7
@ -46,6 +46,31 @@ def check_result(fut, expected):
|
||||
return True
|
||||
|
||||
|
||||
def test_future_await_send():
|
||||
fut = core.AsyncFuture()
|
||||
|
||||
i = fut.__await__()
|
||||
assert i.send(None) == fut
|
||||
assert i.send(None) == fut
|
||||
assert i.send(None) == fut
|
||||
assert i.send(None) == fut
|
||||
|
||||
fut.set_result(123)
|
||||
|
||||
with pytest.raises(StopIteration) as e:
|
||||
i.send(None)
|
||||
|
||||
assert e.value.value == 123
|
||||
|
||||
|
||||
def test_future_await_throw():
|
||||
fut = core.AsyncFuture()
|
||||
|
||||
i = fut.__await__()
|
||||
with pytest.raises(RuntimeError):
|
||||
i.throw(RuntimeError)
|
||||
|
||||
|
||||
def test_future_cancelled():
|
||||
fut = core.AsyncFuture()
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user