Function test_twoclass_method2

python/pathway/tests/test_transformers.py:942–1031

()

def test_twoclass_method2():
    # Same as test_twoclass_method but now the common table is shared
    (
        foo_transformer,
        transformer2,
        transformer_common,
    ) = _fixture_twoclass_transformers()

    hypers = T(
        """
            | a
        11 | 1
        12 | 2
        """
    )
    table = T(
        """
            | a1 | h1 | a2 | h2 | ac | hc
        6   |  1 | 11 |  1 | 13 |  1 | 14
        7   |  2 | 11 |  2 | 13 |  2 | 14
        8   |  3 | 12 |  3 | 13 |  3 | 14
        """
    ).with_columns(
        h1=hypers.pointer_from(this.h1),
        h2=hypers.pointer_from(this.h2),
        hc=hypers.pointer_from(this.hc),
    )

    method_table: Table = foo_transformer(
        hypers=hypers, table=table.select(a=table.a1, h=table.h1)
    ).table
    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
        """
    )

    method_table2: Table = transformer2(
        hypers=hypers2, table=table.select(a=table.a2, h=table.h2)
    ).table
    assert_table_equality(
        method_table2.select(b=method_table2.b, c=method_table2.c(1000000)),
        T(
            """
                |   b |         c