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')))