Function test_simple_method

python/pathway/tests/test_transformers.py:288–324

()

def test_simple_method():
    @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

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

    method_table = foo_transformer(table).table
    result = method_table.select(ret=method_table.c(10))

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


def test_select_method_column():