def test_wait_for_one_with_error_and_completed(): def raise_error(): raise DummyTestError() task1 = Task(raise_error).start() time.sleep(0.1) with pytest.raises(DummyTestError): Task.wait_for_one([task1], timeout=1)
def test_wait_for_one(): def coro(t): time.sleep(t) return "test" task1 = Task(coro).start(0.1) task2 = Task(coro).start(0.2) Task.wait_for_one([task1, task2], timeout=1) assert not task1.is_running and task2.is_running, "Failed to wait for task"