def test_get_all_subsequent_events__simple(self):
"""Verify behavior of _get_all_subsequent_events.
"""
timeline = Timeline()
timeline._subsequent_events = {'one': set(['two']),
'two': set(['three']),
'three': set(['four']),
'four': set()}
subsequent_events = timeline._get_all_subsequent_events('one')
self.assertEqual(set(['two', 'three', 'four']), subsequent_events)
def test_get_all_subsequent_events__branching(self):
"""Verify behavior of _get_all_subsequent_events when events overlap.
"""
timeline = Timeline()
timeline._subsequent_events = {'one': set(['two']),
'two': set(['three']),
'three': set(['six', 'four']),
'four': set(['five']),
'five': set(['six']),
'six': set(['seven']),
'seven': set()}
subsequent_events = timeline._get_all_subsequent_events('one')
self.assertEqual(set(['two', 'three', 'four', 'five', 'six', 'seven']), subsequent_events)
