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,
)
