def test_constructor_and_get(self):
p = Patient(pd.TEST_PATIENT)
self.assertEqual(p.diseases, pd.TEST_PATIENT['diseases'])
self.assertEqual(p.currentTreatmentArm,
pd.TEST_PATIENT['currentTreatmentArm'])
self.assertEqual(p.currentPatientStatus,
pd.TEST_PATIENT['currentPatientStatus'])
self.assertEqual(p.currentStepNumber,
pd.TEST_PATIENT['currentStepNumber'])
self.assertEqual(p.patientTriggers, pd.TEST_PATIENT['patientTriggers'])
self.assertEqual(p.patientAssignments,
pd.TEST_PATIENT['patientAssignments'])
self.assertEqual(p.patientAssignmentIdx,
pd.TEST_PATIENT['patientAssignmentIdx'])
self.assertEqual(p.get_patient_assignment_step_number(),
pd.DEFAULT_PAT_ASSNMNT_STEP_NUM)
def test_get_patient_assignment_step_number(self, patient_json,
exp_step_number):
p = Patient(patient_json)
step_number = p.get_patient_assignment_step_number()
self.assertEqual(step_number, exp_step_number)