def test_update_disease(self):
repo = PatientRepository()
patient1 = Patients("A", "B", 101, "flu")
repo.add_patient(patient1)
repo.update_disease("Zika", patient1)
self.assertEqual(patient1.get_disease(), "Zika")
def test_add_new_disease(self):
repo = PatientRepository()
patient1 = Patients("K", "A", 2121, "flu")
repo.add_patient(patient1)
repo.add_new_disease("Zika", patient1)
self.assertEqual(patient1.get_disease(), "flu, Zika")