def test_events_add(self): rec = er.EventRecord() e1 = e.Event(1) rec.add_event(e1) assert rec.get_event_list() == [e1]
def test_constructor_with_event_list(self): e1 = e.Event(1) e2 = e.Event(1) e_list = [e1, e2] rec = er.EventRecord(e_list) assert rec.get_event_list() == [e1, e2]
def __init__(self, service_time=1): self._id = self.__class__._current_id self.__class__._current_id += 1 self._time = 1 self._customer_queue = [] self._event_record = er.EventRecord() self._tellers = []
def test_filter_by_type_returns_none(self): bank = b.Bank() cust = c.Customer(bank) e1 = e.CustomerArrivalEvent(1, cust) e2 = e.CustomerArrivalEvent(1, cust) e_list = [e1, e2] rec = er.EventRecord(e_list) filtered_rec = rec.filter_by_type(e.CustomerServiceEvent) assert filtered_rec.get_event_list() == []
def test_filter_by_type_with_mixed_events(self): bank = b.Bank() cust = c.Customer(bank) e1 = e.CustomerArrivalEvent(1, cust) e2 = e.CustomerArrivalEvent(1, cust) e3 = e.CustomerDepartureEvent(3, cust) e_list = [e1, e2, e3] rec = er.EventRecord(e_list) filtered_rec = rec.filter_by_type(e.CustomerArrivalEvent) assert filtered_rec.get_event_list() == [e1, e2]
def test_empty_constructor(self): rec = er.EventRecord() assert rec.get_event_list() == []