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()