def _load_employee(self, employee_flds, cp_nbr):
try:
employee = Employee.objects.get(uid=employee_flds[0])
except Employee.DoesNotExist:
from decimal import Decimal
employee = Employee()
employee.uid = employee_flds[0]
employee.name = employee_flds[1]
employee.fed_id = employee_flds[2]
employee.grade = employee_flds[3]
employee.step = employee_flds[4]
employee.fte = employee_flds[5]
employee.salary = employee_flds[6]
employee.gross_pay = Decimal(employee_flds[7]) * 26
employee.cp_nbr = cp_nbr
employee.save()
self.employee_id = employee.id
def test_next_under(self):
employee = Employee()
employee.id = 1
employee.salary = 89752
employee.fte = Decimal(.83)
assignment = Assignment()
assignment.start_date = date(2013, 11, 1)
assignment.end_date = date(2014, 4, 30)
assignment.effort = 70
assignment.project_id = 1
assignment.role_id = 1
employee.assignments.add(assignment)
assignment = Assignment()
assignment.start_date = date(2013, 11, 1)
assignment.end_date = date(2014, 4, 30)
assignment.effort = 13
assignment.project_id = 2
assignment.role_id = 1
employee.assignments.add(assignment)
actual = employee.next_under(start_date=date(2014, 1, 1))
self.assertEqual((2014,5), actual)
employee.assignments.all().delete()
assignment = Assignment()
assignment.start_date = date(2013, 11, 1)
assignment.end_date = date(2014, 4, 30)
assignment.effort = 70
assignment.project_id = 1
assignment.role_id = 1
employee.assignments.add(assignment)
assignment = Assignment()
assignment.start_date = date(2013, 11, 1)
assignment.end_date = date(2014, 3, 31)
assignment.effort = 13
assignment.project_id = 2
assignment.role_id = 1
employee.assignments.add(assignment)
actual = employee.next_under(start_date=date(2014, 1, 1))
self.assertEqual((2014,4), actual)
