Function test_table_add_method_table

python/pathway/tests/test_transformers.py:367–403

()

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

            @output_attribute
            def b(self) -> int:
                return self.a * 10

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

    input = T(
        """
            | a
        1   | 1
        2   | 2
        3   | 3
        """
    )

    method_table = foo_transformer(input).table + input.select(d=-input.a)
    result = method_table.select(ret=method_table.c(10))

    assert_table_equality(
        result,
        T(
            """
                | ret
            1   | 110
            2   | 220
            3   | 330
            """
        ),
    )


def test_method_call_in_transformer_with_input_attribute():