def METHOD_NAME():
    @task
    @annotate({"in_val": float, "return": (float, float)})
    def square(in_val):
        return in_val**2, in_val**3
    res = square(in_val=2.0)()
    assert res.output.out1 == 4.0
    assert res.output.out2 == 8.0