Function test_lists_lexicographical

python/pathway/tests/test_operators.py:1331–1365

()

def test_lists_lexicographical():
    def make_list(n) -> list[int]:
        return list(range(n))

    table = T(
        """
    a | b
    5 | 5
    2 | 3
    4 | 3
    """
    ).with_columns(
        x=pw.apply(make_list, pw.this.a),
        y=pw.apply(make_list, pw.this.b),
    )

    result = table.select(
        pw.this.a,
        pw.this.b,
        eq=pw.this.x == pw.this.y,
        ne=pw.this.x != pw.this.y,
        lt=pw.this.x < pw.this.y,
        le=pw.this.x <= pw.this.y,
        gt=pw.this.x > pw.this.y,
        ge=pw.this.x >= pw.this.y,
    )
    expected = T(
        """
    a | b |  eq   |   ne  |   lt  |   le  |   gt  |   ge
    5 | 5 |  True | False | False |  True | False |  True
    2 | 3 | False |  True |  True |  True | False | False
    4 | 3 | False |  True | False | False |  True |  True
    """
    )
    assert_table_equality(result, expected)


@pytest.mark.parametrize("cast", ["a", "b"])