def add_test(self): er = ERModel() p1 = Patient("John", 2) self.assertTrue(er.isEmpty()) er.patients.add(p1) self.assertFalse(er.isEmpty())
def treat_test(self): er = ERModel() p1 = Patient("John", 2) p2 = Patient("Sally", 1) er.patients.add(p1) er.patients.add(p2) self.assertTrue(er.treatNext() == p2) self.assertTrue(er.treatNext() == p1) self.assertTrue(er.treatNext() == None)
def main(): model = ERModel() view = ERView(model) view.run()
def schedule_test(self): er = ERModel() p1 = Patient("John", 2) er.schedule(p1) self.assertFalse(er.patients.isEmpty())
def treatNext(self): """Treats the next patient and updates the display.""" patient = self.model.treatNext() self.outputArea.appendText(str(patient) + " is being treated\n") if self.model.isEmpty(): self.treatNextBtn["state"] = "disabled" self.treatAllBtn["state"] = "disabled" def treatAll(self): """Treats all the patients.""" while not self.model.isEmpty(): self.treatNext() def getCondition(self): """Returns a Condition object corresponding to the selected radio button in the view.""" text = self.conditionGrp.getSelectedButton()["text"].split()[0] if text == "Fair": number = 3 elif text == "Serious": number = 2 else: number = 1 return Condition(number) if __name__ == "__main__": model = ERModel() ERView(model).mainloop()