Function test_restrict

python/pathway/tests/test_persistence.py:359–376

(tmp_path, mode)

)
@only_with_license_key("mode", [api.PersistenceMode.OPERATOR_PERSISTING])
def test_restrict(tmp_path, mode):
    class InputSchema(pw.Schema):
        a: int = pw.column_definition(primary_key=True)

    def logic(t_1: pw.Table, t_2: pw.Table) -> pw.Table:
        t_2.promise_universe_is_subset_of(t_1)
        return t_1.restrict(t_2)

    run, _, input_path_2 = get_two_tables_runner(
        tmp_path, mode, logic, InputSchema, terminate_on_error=False
    )

    run(["a", "1", "2", "3"], ["a", "1"], {"1,1"})
    run(["a"], ["a", "3"], {"3,1"})
    run(["a", "4", "5"], ["a", "5"], {"5,1"})
    run(["a", "6"], ["a", "4", "6"], {"4,1", "6,1"})
    os.remove(input_path_2 / "3")
    run(["a"], ["a"], {"5,-1"})


@pytest.mark.parametrize(