How to use the pandera.check_output function in pandera

    @check_output(out_schema, "out_dataframe")
    def test_func3(x, in_dataframe=None):
        return {
            "x": x,
            "out_dataframe": in_dataframe.assign(f=["a", "b", "a"]),
        }

    @check_output(out_schema)
    def test_func1(dataframe, x):
        return dataframe.assign(f=["a", "b", "a"])

        @check_output(out_schema)
        def transform_secord_arg_with_dict_getter(self, x, df):
            return _transform_helper(df)

    @check_output(out_schema, 1)
    def func_input_transform2(_, df):
        return _, df

        @check_output(out_schema)
        def transform_first_arg(self, df):
            return _transform_helper(df)

    @check_output(out_schema)
    def func_input_transform1(df):
        return df

    @check_output(out_schema, 0)
    def test_func2(x, dataframe):
        return dataframe.assign(f=["a", "b", "a"]), x

        @check_output(out_schema)
        def transform_secord_arg_with_dict_getter(self, x, df):
            return _transform_helper(df)

        @check_output(out_schema)
        def transform_secord_arg_with_list_getter(self, x, df):
            return _transform_helper(df)

    @check_output(out_schema, 1)
    def func_input_transform2(_, df):
        return _, df