示例#1
0
def test_qit_declaration():
    p = Product(Range(1), Range(1))
    f = Function("f").takes(p.type, "p").returns(Int()).from_file("x.h")
    g = Function("g").takes(Int(), "x").returns(Int()).from_file("y.h")

    q = Qit()

    assert q.declarations(g) == ["qint g(qint x)"]
    assert len(q.declarations(p.iterate().map(f).map(g).map(g))) == 2