def test_insert_after_2(self):
results = []
cs = CallSequence(
('a', lambda : results.append('a ran')),
('b', lambda : results.append('b ran')),
)
assert cs.count == 2
cs.insert_after('a', ('c', lambda : results.append('c ran')))
assert cs.count == 3
cs()
assert results == ['a ran', 'c ran', 'b ran']
def test_insert_after_4(self):
cs = CallSequence()
with pytest.raises(KeyError):
cs.insert_after('z', ('c', lambda : results.append('c ran')))