def test_wf_output_mismatch():
    with pytest.raises(AssertionError):

        @workflow
        def my_wf(a: int, b: str) -> (int, str):
            return a
    with pytest.raises(AssertionError):

        @workflow
        def my_wf2(a: int, b: str) -> int:
            return (a, b)

    @workflow
    def my_wf3(a: int, b: str) -> int:
        return (a,)
    my_wf3(a=10, b='hello')
def test_wf_output_mismatch():
    with pytest.raises(AssertionError):

        @workflow
        def my_wf(a: int, b: str) -> (int, str):
            return a
    with pytest.raises(AssertionError):

        @workflow
        def my_wf2(a: int, b: str) -> int:
            return (a, b)

    @workflow
    def my_wf3(a: int, b: str) -> int:
        return (a,)
    my_wf3(a=10, b='hello')