def test_total_duration_per_event_sorted():
# Assign
discovery = Discovery()
# Act
discovery.add_event(create_event(update_event={'duration': datetime.timedelta(seconds=10)}))
discovery.add_event(create_event(update_event={'duration': datetime.timedelta(seconds=50)},
update_data={'title': 'Fantastic choice!'}))
discovery.add_event(create_event(update_event={'duration': datetime.timedelta(seconds=50)}))
discovery.add_event(create_event(update_event={'duration': datetime.timedelta(seconds=20)},
update_data={'title': 'YES! choice!'}))
all_unique_events = discovery.get_all_unique_events()
total_duration_events = discovery.get_agg_duration_events()
sorted_total_duration_events = discovery.get_agg_duration_events_sorted()
# Assert
assert len(all_unique_events) == 3
assert len(total_duration_events) == 3
assert 'Amazing page' in str(sorted_total_duration_events[0][0])
assert sorted_total_duration_events[0][1] == 60
assert 'Fantastic' in str(sorted_total_duration_events[1][0])
assert sorted_total_duration_events[1][1] == 50
assert 'YES!' in str(sorted_total_duration_events[2][0])
assert sorted_total_duration_events[2][1] == 20
def test_total_duration_per_event():
# Assign
discovery = Discovery()
event1 = create_event(update_event={'duration': datetime.timedelta(seconds=10)})
event2 = create_event(update_event={'duration': datetime.timedelta(seconds=50)})
# Act
discovery.add_event(event1)
discovery.add_event(event2)
all_unique_events = discovery.get_all_unique_events()
total_duration_events = discovery.get_agg_duration_events()
# Assert
assert len(all_unique_events) == 1
assert len(total_duration_events) == 1
any_key = list(total_duration_events.keys())[0]
assert total_duration_events[any_key] == 60