def test_wf1_df():

    @task
    def t1(a: int) -> pandas.DataFrame:
        return pandas.DataFrame(data={'col1': [a, 2], 'col2': [a, 4]})

    @task
    def t2(df: pandas.DataFrame) -> pandas.DataFrame:
        return df.append(pandas.DataFrame(data={'col1': [5, 10], 'col2': [5, 10]}))

    @workflow
    def my_wf(a: int) -> pandas.DataFrame:
        df = t1(a=a)
        return t2(df=df)
    x = my_wf(a=20)
    assert isinstance(x, pandas.DataFrame)
    result_df = (x.reset_index(drop=True) == pandas.DataFrame(data={'col1': [20, 2, 5, 10], 'col2': [20, 4, 5, 10]}).reset_index(drop=True))
    assert result_df.all().all()
def test_wf1_df():

    @task
    def t1(a: int) -> pandas.DataFrame:
        return pandas.DataFrame(data={'col1': [a, 2], 'col2': [a, 4]})

    @task
    def t2(df: pandas.DataFrame) -> pandas.DataFrame:
        return df.append(pandas.DataFrame(data={'col1': [5, 10], 'col2': [5, 10]}))

    @workflow
    def my_wf(a: int) -> pandas.DataFrame:
        df = t1(a=a)
        return t2(df=df)
    x = my_wf(a=20)
    assert isinstance(x, pandas.DataFrame)
    result_df = (x.reset_index(drop=True) == pandas.DataFrame(data={'col1': [20, 2, 5, 10], 'col2': [20, 4, 5, 10]}).reset_index(drop=True))
    assert result_df.all().all()