def test_one_arg(): stop = StopIteration(42) assert tasklets._get_return_value(stop) == 42
def test_two_args(): stop = StopIteration(42, 21) assert tasklets._get_return_value(stop) == (42, 21)
def test_no_args(): stop = StopIteration() assert tasklets._get_return_value(stop) is None