def test_referrals_for_supervision_periods(self): supervision_period = \ StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, status=StateSupervisionPeriodStatus.TERMINATED, state_code='US_XX', start_date=date(2008, 3, 5), termination_date=date(2010, 5, 19), termination_reason=StateSupervisionPeriodTerminationReason.DISCHARGE, supervision_type=StateSupervisionType.PAROLE ) program_referrals = identifier.referrals_for_supervision_periods( state_code='US_XX', program_id='XXX', referral_date=date(2009, 3, 12), participation_status=StateProgramAssignmentParticipationStatus.DISCHARGED, assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_periods=[supervision_period], supervision_period_to_agent_associations=DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS ) self.assertListEqual([ ProgramReferralEvent( state_code='US_XX', program_id='XXX', event_date=date(2009, 3, 12), participation_status=StateProgramAssignmentParticipationStatus.DISCHARGED, assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_type=supervision_period.supervision_type ) ], program_referrals)
def test_referrals_for_supervision_periods_different_types(self): supervision_period_1 = \ StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, status=StateSupervisionPeriodStatus.TERMINATED, state_code='UT', start_date=date(2008, 3, 5), termination_date=date(2010, 5, 19), termination_reason= StateSupervisionPeriodTerminationReason.DISCHARGE, supervision_type=StateSupervisionType.PAROLE ) supervision_period_2 = \ StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, status=StateSupervisionPeriodStatus.TERMINATED, state_code='UT', start_date=date(2008, 3, 5), termination_date=date(2010, 5, 19), termination_reason= StateSupervisionPeriodTerminationReason.DISCHARGE, supervision_type=StateSupervisionType.PROBATION ) supervision_periods = [supervision_period_1, supervision_period_2] program_referrals = identifier.referrals_for_supervision_periods( state_code='UT', program_id='XXX', referral_date=date(2009, 3, 19), assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_periods=supervision_periods, supervision_period_to_agent_associations= DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS) self.assertEqual(2, len(program_referrals)) self.assertEqual([ ProgramReferralEvent( state_code='UT', program_id='XXX', event_date=date(2009, 3, 19), assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_type=supervision_period_1.supervision_type), ProgramReferralEvent( state_code='UT', program_id='XXX', event_date=date(2009, 3, 19), assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_type=supervision_period_2.supervision_type) ], program_referrals)
def test_referrals_for_supervision_periods_different_types(self): supervision_period_1 = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, status=StateSupervisionPeriodStatus.TERMINATED, state_code="US_XX", start_date=date(2008, 3, 5), termination_date=date(2010, 5, 19), termination_reason=StateSupervisionPeriodTerminationReason. DISCHARGE, supervision_type=StateSupervisionType.PAROLE, ) supervision_period_2 = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, status=StateSupervisionPeriodStatus.TERMINATED, state_code="US_XX", start_date=date(2008, 3, 5), termination_date=date(2010, 5, 19), termination_reason=StateSupervisionPeriodTerminationReason. DISCHARGE, supervision_type=StateSupervisionType.PROBATION, ) supervision_periods = [supervision_period_1, supervision_period_2] program_referrals = identifier.referrals_for_supervision_periods( state_code="US_XX", program_id="XXX", referral_date=date(2009, 3, 19), participation_status=StateProgramAssignmentParticipationStatus. DENIED, assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_periods=supervision_periods, supervision_period_to_agent_associations= DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS, ) self.assertListEqual( [ ProgramReferralEvent( state_code="US_XX", program_id="XXX", event_date=date(2009, 3, 19), participation_status= StateProgramAssignmentParticipationStatus.DENIED, assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_type=supervision_period_1.supervision_type, ), ProgramReferralEvent( state_code="US_XX", program_id="XXX", event_date=date(2009, 3, 19), participation_status= StateProgramAssignmentParticipationStatus.DENIED, assessment_score=39, assessment_type=StateAssessmentType.LSIR, supervision_type=supervision_period_2.supervision_type, ), ], program_referrals, )