def test_subterms(): a = TableSymbol('a', '{x: int, y: int, z: int}') assert list(a._subterms()) == [a] assert set(a['x']._subterms()) == set([a, a['x']]) assert set(a['x'].map(inc, 'int')._subterms()) == \ set([a, a['x'], a['x'].map(inc, 'int')]) assert a in set((a['x'] + 1)._subterms())
def test_subterms(): a = TableSymbol("a", "{x: int, y: int, z: int}") assert list(a._subterms()) == [a] assert set(a["x"]._subterms()) == set([a, a["x"]]) assert set(a["x"].map(inc, "int")._subterms()) == set([a, a["x"], a["x"].map(inc, "int")]) assert a in set((a["x"] + 1)._subterms())