def test_get_average_completed(self): testAssign, testPatient, testProcedure = self.test_create_assignedProcedure( ) # test 1 procedure 0 time tProcedure = Procedure.objects.create(procedure_name="bloodwork") assigned = AssignedProcedures.assign_procedure_to_patient( 2, testPatient, tProcedure) spoof_time = assigned.created_at assigned.completed = True assigned.date_completed = spoof_time assigned.save() time = AssignedProcedures.average_completion_time(tProcedure.id) self.assertEqual(time, "0") # test 2 procedures 1 time assigned = AssignedProcedures.assign_procedure_to_patient( 3, testPatient, tProcedure) spoof_time = assigned.created_at assigned.completed = True assigned.date_completed = spoof_time + timedelta(days=2) assigned.save() time = AssignedProcedures.average_completion_time(tProcedure.id) self.assertEqual(time, "1")
def average_completion_time(procedure_id): return AssignedProcedures.average_completion_time(procedure_id) + " days"