Function test_method_in_ix

python/pathway/tests/test_transformers.py:1034–1069

()

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

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

    input = T(
        """
            | c | b
        7   | 1 | 10
        8   | 2 | 10
        9   | 3 | 20
        """
    )

    reduced = input.groupby(input.b).reduce(a=input.b)

    oracle = foo_transformer(reduced).table

    result = input.select(res=oracle.ix_ref(input.b).method(input.c))

    assert_table_equality(
        result,
        T(
            """
                | res
            7   | 10
            8   | 20
            9   | 60
            """
        ),
    )


def test_method_in_groupby():