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)