Function test_method_in_groupby

python/pathway/tests/test_transformers.py:1072–1111

()

def test_method_in_groupby():
    @transformer
    class foo_transformer:
        class table(ClassArg):
            a: int = input_attribute()

            @method
            def method(self, arg) -> int:
                return self.a * arg

    input = T(
        """
        a | grouper
        1 | 1
        2 | 1
        3 | 2
        4 | 2
        5 | 3
        6 | 3
        """,
    )

    oracle = foo_transformer(input).table

    combined = input + oracle

    result = combined.groupby(combined.grouper).reduce(
        res=reducers.sum(combined.method(combined.a))
    )
    assert_table_equality_wo_index(
        result,
        T(
            """
            res
            5
            25
            61
            """,
        ),
    )