Function test_multiple_ix

python/pathway/tests/test_common.py:4447–4499

()

def test_multiple_ix():
    indexed_table = T(
        """
           | col
        2  | a
        3  | b
        4  | c
        5  | d
        """
    )

    indexer1 = T(
        """
          | key
        1 | 4
        2 | 3
        3 | 2
        4 | 1
    """
    ).with_columns(key=indexed_table.pointer_from(pw.this.key))

    indexer2 = T(
        """
          | key
        1 | 6
        2 | 5
        3 | 4
        4 | 3
    """
    ).with_columns(key=indexed_table.pointer_from(pw.this.key))

    a = (
        indexed_table.ix(indexer1.key, allow_misses=True)
        .filter(pw.this.col.is_not_none())
        .select(col1=pw.this.col)
    )
    b = (
        indexed_table.ix(indexer2.key, allow_misses=True)
        .filter(pw.this.col.is_not_none())
        .select(col2=pw.this.col)
    )
    result = a.intersect(b)
    result = a.restrict(result) + b.restrict(result)
    assert_table_equality_wo_index(
        result,
        T(
            """
        col1 | col2
           a |    c
           b |    d
        """
        ),
    )


def test_join_desugaring_assign_id():