var a:[int]int;

procedure Allocate() returns ({:linear "tid"} xls: int);

function {:builtin "MapConst"} MapConstBool(bool) : [int]bool;
function {:inline} {:linear "tid"} TidCollector(x: int) : [int]bool
{
  MapConstBool(false)[x := true]
}

procedure {:entrypoint} {:yields} main() 
{
    var {:linear "tid"} i: int;
    var {:linear "tid"} j: int;
    call i := Allocate();
    call j := Allocate();
    par i := t(i) | j := Yield(j);
    par i := u(i) | j := u(j);
}

procedure {:yields} {:stable} t({:linear "tid"} i': int) returns ({:linear "tid"} i: int)
{
    i := i';

    a[i] := 42;
    call i := Yield(i);
    assert a[i] == 42;
}

procedure {:yields} {:stable} u({:linear "tid"} i': int) returns ({:linear "tid"} i: int) 
{
    i := i';

    a[i] := 42;
    yield;
    assert a[i] == 42;
}

procedure {:yields} {:stable} Yield({:linear "tid"} i': int) returns ({:linear "tid"} i: int)
ensures i == i';
ensures old(a)[i] == a[i];
{
    i := i';
    yield;
    assert old(a)[i] == a[i];
}