def test_to_dict_two_match_events(self): event_a = PrimitiveEvent(timestamp=EpochNSClock.generate_timestamp()) event_b = PrimitiveEvent(timestamp=EpochNSClock.generate_timestamp()) match_a = MatchEvent(nfa_name=NFA_NAME_A, label=LABEL_LAYER_A, event=event_a) match_b = MatchEvent(nfa_name=NFA_NAME_A, label=LABEL_LAYER_B, event=event_b) version = RunVersion() version.add_level( BoboRun._generate_id(nfa_name=NFA_NAME_A, start_event_id=event_a.event_id)) version_str = version.get_version_as_str() # match a --next--> match b match_a.add_pointer_next(version=version_str, label=match_b.label, event_id=match_b.event.event_id) # match a dict self.assertDictEqual( match_a.to_dict(), { MatchEvent.NFA_NAME: NFA_NAME_A, MatchEvent.LABEL: LABEL_LAYER_A, MatchEvent.EVENT: event_a.to_dict(), MatchEvent.NEXT_IDS: { version_str: (match_b.label, match_b.event.event_id) }, MatchEvent.PREVIOUS_IDS: {} }) # match a <--previous-- match b match_b.add_pointer_previous(version=version_str, label=match_a.label, event_id=match_a.event.event_id) # match b dict self.assertDictEqual( match_b.to_dict(), { MatchEvent.NFA_NAME: NFA_NAME_A, MatchEvent.LABEL: LABEL_LAYER_B, MatchEvent.EVENT: event_b.to_dict(), MatchEvent.NEXT_IDS: {}, MatchEvent.PREVIOUS_IDS: { version_str: (match_a.label, match_a.event.event_id) } })
def test_to_dict(self): # Create primitive event p_timestamp = EpochNSClock.generate_timestamp() p_event = PrimitiveEvent(timestamp=p_timestamp, data=P_DATA) p_hist = "p_hist" # Create composite event c_timestamp = EpochNSClock.generate_timestamp() c_history = BoboHistory() c_event = CompositeEvent(timestamp=c_timestamp, name=C_NAME, history=c_history, data=C_DATA) c_hist = "c_hist" # Create list of events for history events = {p_hist: [p_event], c_hist: [c_event]} # Create history and add events history = BoboHistory(events=events) self.assertDictEqual(history.to_dict(), { p_hist: [p_event.to_dict()], c_hist: [c_event.to_dict()] })
def test_to_dict(self): p_timestamp = EpochNSClock.generate_timestamp() p_data = {"p_key": "p_value"} p_event = PrimitiveEvent(timestamp=p_timestamp, data=p_data) self.assertDictEqual( p_event.to_dict(), { PrimitiveEvent.TIMESTAMP: p_timestamp, PrimitiveEvent.DATA: p_data, PrimitiveEvent.EVENT_ID: p_event.event_id })