def test_insert_many_nonexistent_causes_exception():
comp = Computation()
comp.add_node('a')
comp.insert('a', 0)
with assert_raises(NonExistentNodeException):
comp.insert_many([('a', 1), ('b', 2)])
assert comp.v.a == 0
def test_insert_many():
comp = Computation()
l = list(range(100))
random.shuffle(l)
prev = None
for x in l:
if prev is None:
comp.add_node(x)
else:
comp.add_node(x, lambda n: n + 1, kwds={'n': prev})
prev = x
comp.insert_many([(x, x) for x in range(100)])
for x in range(100):
assert comp[x] == (States.UPTODATE, x)