Function test_twoclass_method

python/pathway/tests/test_transformers.py:829–939

()

def test_twoclass_method():
    (
        foo_transformer,
        transformer2,
        transformer_common,
    ) = _fixture_twoclass_transformers()

    hypers = T(
        """
            | a
        11 | 1
        12 | 2
        """
    )
    table = T(
        """
            | a |  h
        6   | 1 | 11
        7   | 2 | 11
        8   | 3 | 12
        """
    ).with_columns(h=hypers.pointer_from(this.h))

    method_table: Table = foo_transformer(hypers, table).table

    assert_table_equality(
        method_table.select(method_table.b),
        T(
            """
                | b
            6   | 11
            7   | 21
            8   | 32
            """
        ),
    )

    assert_table_equality(
        method_table.select(b=method_table.b, c=method_table.c(1000)),
        T(
            """
                |  b |     c
            6   | 11 | 12000
            7   | 21 | 23000
            8   | 32 | 35000
            """
        ),
    )

    hypers2 = T(
        """
            | a
        13 | 3
        """
    )
    table2 = T(
        """
            | a |  h