def test_get_raises(self, exc_info): result = WaitResult() result.send(1, exc_info) with pytest.raises(self.CustomError): result.get()
def test_get_result_multiple_times(self): result = WaitResult() result.send(1, None) assert result.get() == 1 assert result.get() == 1