Function test_argmax_rows_02

python/pathway/tests/test_utils.py:552–579

()

def test_argmax_rows_02():
    input = T(
        """
        | foo   | bar
    0   |  1    | 5
    1   |  2    | 6
    2   |  3    | 8
    3   |  2    | 8
    4   |  3    | 8
    5   |  3    | 5
    6   |  5    | 2
    7   |  6    | 1
    """
    )
    expected = T(
        """
        | foo   | bar
    0   |  1    | 5
    2   |  3    | 8
    3   |  2    | 8
    6   |  5    | 2
    7   |  6    | 1
    """
    )

    assert_table_equality_wo_index(
        argmax_rows(input, *[input.foo], what=input.bar), expected
    )


def test_table_from_rows_stream():