def test_find_program_referrals_assessment_after_referral(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code='US_CA', program_id='PG3', referral_date=date(2009, 10, 3)) assessment_1 = StateAssessment.new_with_defaults( state_code='US_CA', assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 3, 10)) assessment_2 = StateAssessment.new_with_defaults( state_code='US_CA', assessment_type=StateAssessmentType.ORAS, assessment_score=29, assessment_date=date(2009, 10, 4)) program_assignments = [program_assignment] assessments = [assessment_1, assessment_2] supervision_periods = [] program_referrals = identifier.find_program_referrals( program_assignments, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS) self.assertEqual(1, len(program_referrals)) self.assertEqual([ ProgramReferralEvent(state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, assessment_score=33, assessment_type=StateAssessmentType.ORAS) ], program_referrals)
def test_find_assessment_score_change_insufficient_assessments(self): assessment_1 = StateAssessment.new_with_defaults( state_code='CA', assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2015, 3, 23) ) assessment_2 = StateAssessment.new_with_defaults( state_code='CA', assessment_type=StateAssessmentType.ORAS, assessment_score=29, assessment_date=date(2015, 11, 2) ) assessments = [assessment_1, assessment_2] start_date = date(2015, 2, 22) termination_date = date(2016, 3, 12) assessment_score_change, \ end_assessment_score, \ end_assessment_level, \ end_assessment_type = \ assessment_utils.find_assessment_score_change( start_date, termination_date, assessments ) self.assertIsNone(assessment_score_change) self.assertIsNone(end_assessment_score) self.assertIsNone(end_assessment_level) self.assertIsNone(end_assessment_type)
def test_find_most_recent_applicable_assessment_US_MO(self): state_code = 'US_MO' lsir_assessment = StateAssessment.new_with_defaults( state_code=state_code, assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 28), assessment_score=17 ) oras_assessment = StateAssessment.new_with_defaults( state_code=state_code, assessment_date=date(2018, 4, 29), assessment_score=17 ) assessments = [lsir_assessment, oras_assessment] applicable_assessment_types = [StateAssessmentType.ORAS_COMMUNITY_SUPERVISION, StateAssessmentType.ORAS_COMMUNITY_SUPERVISION_SCREENING] for assessment_type in applicable_assessment_types: oras_assessment.assessment_type = assessment_type for pipeline in assessment_utils._ASSESSMENT_TYPES_TO_INCLUDE_FOR_CLASS: most_recent_assessment = assessment_utils.find_most_recent_applicable_assessment_of_class_for_state( date(2018, 4, 30), assessments, pipeline, state_code ) self.assertEqual(most_recent_assessment, oras_assessment)
def test_assessments_in_compliance_month(self): evaluation_date = date(2018, 4, 30) assessment_out_of_range = StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 3, 10)) assessment_out_of_range_2 = StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 5, 10)) assessment_1 = StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 2)) assessment_2 = StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 10)) assessment_3 = StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 28)) assessments = [ assessment_out_of_range, assessment_out_of_range_2, assessment_1, assessment_2, assessment_3 ] expected_assessments = [assessment_1, assessment_2, assessment_3] self.assertEqual( len(expected_assessments), _assessments_in_compliance_month(evaluation_date, assessments))
def test_find_most_recent_applicable_assessment_LSIR(self, mock_assessment_types): mock_assessment_types.return_value = [StateAssessmentType.LSIR] assessment_1 = StateAssessment.new_with_defaults( state_code='US_XX', assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 28), assessment_score=17 ) assessment_2 = StateAssessment.new_with_defaults( state_code='US_XX', assessment_type=StateAssessmentType.ORAS_COMMUNITY_SUPERVISION_SCREENING, assessment_date=date(2018, 4, 29), assessment_score=17 ) assessments = [assessment_1, assessment_2] most_recent_assessment = assessment_utils.find_most_recent_applicable_assessment_of_class_for_state( date(2018, 4, 30), assessments, StateAssessmentClass.RISK, 'US_XX' ) self.assertEqual(most_recent_assessment, assessment_1)
def test_find_program_referrals_multiple_assessments(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code="US_XX", program_id="PG3", referral_date=date(2009, 10, 3), participation_status=StateProgramAssignmentParticipationStatus. IN_PROGRESS, ) assessment_1 = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 3, 10), ) assessment_2 = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.ORAS, assessment_score=29, assessment_date=date(2009, 9, 14), ) 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, ) assessments = [assessment_1, assessment_2] supervision_periods = [supervision_period] program_referrals = identifier.find_program_referrals( program_assignment, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS, ) self.assertListEqual( [ ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, participation_status=program_assignment. participation_status, assessment_score=29, assessment_type=StateAssessmentType.ORAS, supervision_type=supervision_period.supervision_type, ) ], program_referrals, )
def test_find_program_referrals_multiple_assessments(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code='US_CA', program_id='PG3', referral_date=date(2009, 10, 3)) assessment_1 = StateAssessment.new_with_defaults( state_code='US_CA', assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 3, 10)) assessment_2 = StateAssessment.new_with_defaults( state_code='US_CA', assessment_type=StateAssessmentType.ORAS, assessment_score=29, assessment_date=date(2009, 9, 14)) supervision_period = \ 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 ) program_assignments = [program_assignment] assessments = [assessment_1, assessment_2] supervision_periods = [supervision_period] program_referrals = identifier.find_program_referrals( program_assignments, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS) self.assertEqual(1, len(program_referrals)) self.assertEqual([ ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, assessment_score=29, assessment_type=StateAssessmentType.ORAS, supervision_type=supervision_period.supervision_type) ], program_referrals)
def test_assessment_is_up_to_date_assessment_before_starting_probation(self): supervision_period = \ StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, external_id='sp1', state_code='US_ID', custodial_authority='US_ID_DOC', start_date=date(2018, 3, 5), admission_reason=StateSupervisionPeriodAdmissionReason.COURT_SENTENCE, supervision_period_supervision_type=StateSupervisionPeriodSupervisionType.PROBATION ) assessment = StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_score=33, assessment_level=StateAssessmentLevel.HIGH, assessment_date=date(2018, 1, 3) ) # This person started on probation more than NEW_SUPERVISION_ASSESSMENT_DEADLINE_DAYS ago, and they do not have # a recent assessment, so their assessment is not in compliance start_of_supervision = supervision_period.start_date - \ relativedelta(days=NEW_SUPERVISION_ASSESSMENT_DEADLINE_DAYS) compliance_evaluation_date = supervision_period.start_date + \ relativedelta(days=1) assessment_up_to_date = _assessment_is_up_to_date( supervision_period, start_of_supervision, compliance_evaluation_date, most_recent_assessment=assessment ) self.assertTrue(assessment_up_to_date)
def test_assessment_is_up_to_date(self): supervision_period = \ StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, external_id='sp1', state_code='US_ID', custodial_authority='US_ID_DOC', start_date=date(2018, 3, 5), termination_date=date(2018, 5, 19), admission_reason=StateSupervisionPeriodAdmissionReason.COURT_SENTENCE, termination_reason=StateSupervisionPeriodTerminationReason.DISCHARGE, supervision_period_supervision_type=StateSupervisionPeriodSupervisionType.PROBATION ) assessment = StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_score=33, assessment_level=StateAssessmentLevel.HIGH, assessment_date=date(2018, 3, 10) ) start_of_supervision = supervision_period.start_date compliance_evaluation_date = date(2018, 4, 30) assessment_up_to_date = _assessment_is_up_to_date( supervision_period, start_of_supervision, compliance_evaluation_date, assessment ) self.assertTrue(assessment_up_to_date)
def test_find_program_referrals_assessment_after_referral(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code="US_XX", program_id="PG3", referral_date=date(2009, 10, 3), participation_status=StateProgramAssignmentParticipationStatus. DISCHARGED, ) assessment_1 = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 3, 10), ) assessment_2 = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.ORAS, assessment_score=29, assessment_date=date(2009, 10, 4), ) assessments = [assessment_1, assessment_2] supervision_periods = [] program_referrals = identifier.find_program_referrals( program_assignment, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS, ) self.assertListEqual( [ ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, participation_status=program_assignment. participation_status, assessment_score=33, assessment_type=StateAssessmentType.ORAS, ) ], program_referrals, )
def test_find_program_referrals_officer_info(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code='US_CA', program_id='PG3', referral_date=date(2009, 10, 3), participation_status=StateProgramAssignmentParticipationStatus. DISCHARGED, ) assessment = StateAssessment.new_with_defaults( state_code='US_CA', assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 7, 10)) supervision_period = \ 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 ) program_assignments = [program_assignment] assessments = [assessment] supervision_periods = [supervision_period] supervision_period_agent_associations = { supervision_period.supervision_period_id: { 'agent_id': 000, 'agent_external_id': 'OFFICER10', 'district_external_id': 'DISTRICT8', 'supervision_period_id': supervision_period.supervision_period_id } } program_referrals = identifier.find_program_referrals( program_assignments, assessments, supervision_periods, supervision_period_agent_associations) self.assertEqual(1, len(program_referrals)) self.assertEqual([ ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, participation_status=program_assignment.participation_status, assessment_score=33, assessment_type=StateAssessmentType.ORAS, supervision_type=supervision_period.supervision_type, supervising_officer_external_id='OFFICER10', supervising_district_external_id='DISTRICT8') ], program_referrals)
def test_find_assessment_score_change(self): assessment_1 = StateAssessment.new_with_defaults( state_code='CA', assessment_type=StateAssessmentType.ORAS, assessment_level=StateAssessmentLevel.HIGH, assessment_score=33, assessment_date=date(2015, 3, 23) ) assessment_2 = StateAssessment.new_with_defaults( state_code='CA', assessment_type=StateAssessmentType.ORAS, assessment_score=29, assessment_level=StateAssessmentLevel.HIGH, assessment_date=date(2015, 11, 2) ) assessment_3 = StateAssessment.new_with_defaults( state_code='CA', assessment_type=StateAssessmentType.ORAS, assessment_score=23, assessment_level=StateAssessmentLevel.MEDIUM, assessment_date=date(2016, 1, 13) ) assessments = [assessment_1, assessment_2, assessment_3] start_date = date(2015, 2, 22) termination_date = date(2016, 3, 12) assessment_score_change, \ end_assessment_score, \ end_assessment_level, \ end_assessment_type = \ assessment_utils.find_assessment_score_change( start_date, termination_date, assessments ) self.assertEqual(-6, assessment_score_change) self.assertEqual(assessment_3.assessment_score, end_assessment_score) self.assertEqual(assessment_3.assessment_level, end_assessment_level) self.assertEqual(assessment_3.assessment_type, end_assessment_type)
def test_find_program_referrals(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code="US_XX", program_id="PG3", referral_date=date(2009, 10, 3), participation_status=StateProgramAssignmentParticipationStatus. IN_PROGRESS, ) assessment = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 7, 10), ) supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=999, 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_site="OFFICE_1", ) assessments = [assessment] supervision_periods = [supervision_period] program_referrals = self.identifier._find_program_referrals( program_assignment, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS, ) self.assertListEqual( [ ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, participation_status=program_assignment. participation_status, assessment_score=33, assessment_type=StateAssessmentType.ORAS, supervision_type=supervision_period.supervision_type, supervising_officer_external_id="XXX", supervising_district_external_id="OFFICE_1", level_1_supervision_location_external_id="OFFICE_1", ) ], program_referrals, )
def test_find_program_events(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code='US_CA', program_id='PG3', referral_date=date(2020, 1, 3), participation_status=StateProgramAssignmentParticipationStatus. IN_PROGRESS, program_location_id='LOCATION X', start_date=date(2020, 1, 1)) assessment = StateAssessment.new_with_defaults( state_code='US_CA', assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2019, 7, 10)) supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, status=StateSupervisionPeriodStatus.TERMINATED, state_code='UT', start_date=date(2019, 3, 5), supervision_type=StateSupervisionType.PAROLE) program_assignments = [program_assignment] assessments = [assessment] supervision_periods = [supervision_period] program_events = identifier.find_program_events( program_assignments, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS) expected_events = [ ProgramReferralEvent( state_code=program_assignment.state_code, event_date=program_assignment.referral_date, program_id=program_assignment.program_id, supervision_type=supervision_period.supervision_type, participation_status=program_assignment.participation_status, assessment_score=assessment.assessment_score, assessment_type=assessment.assessment_type), ProgramParticipationEvent( state_code=program_assignment.state_code, event_date=program_assignment.start_date, program_id=program_assignment.program_id, program_location_id=program_assignment.program_location_id, supervision_type=supervision_period.supervision_type), ProgramParticipationEvent( state_code=program_assignment.state_code, event_date=program_assignment.start_date + relativedelta(days=1), program_id=program_assignment.program_id, program_location_id=program_assignment.program_location_id, supervision_type=supervision_period.supervision_type) ] self.assertListEqual(program_events, expected_events)
def test_person_and_kwargs_for_identifier_no_person(self): assessment = StateAssessment.new_with_defaults(state_code='US_XX') arg_to_entities_map = { # There should never be two StatePerson entities with the same person_id. This should fail loudly. 'assessments': iter([assessment]) } with pytest.raises(ValueError): _ = person_and_kwargs_for_identifier(arg_to_entities_map)
def test_us_id_case_compliance_on_date(self): supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, external_id='sp1', state_code='US_ID', custodial_authority='US_ID_DOC', start_date=date(2018, 3, 5), termination_date=date(2018, 5, 19), admission_reason=StateSupervisionPeriodAdmissionReason. COURT_SENTENCE, termination_reason=StateSupervisionPeriodTerminationReason. DISCHARGE, supervision_period_supervision_type= StateSupervisionPeriodSupervisionType.PROBATION, supervision_level=StateSupervisionLevel.MEDIUM) case_type = StateSupervisionCaseType.GENERAL assessments = [ StateAssessment.new_with_defaults( state_code='US_ID', assessment_type=StateAssessmentType.LSIR, assessment_score=33, assessment_level=StateAssessmentLevel.HIGH, assessment_date=date(2018, 3, 10)) ] supervision_contacts = [ StateSupervisionContact.new_with_defaults( state_code='US_ID', contact_date=date(2018, 3, 6), contact_type=StateSupervisionContactType.FACE_TO_FACE, status=StateSupervisionContactStatus.COMPLETED), StateSupervisionContact.new_with_defaults( state_code='US_ID', contact_date=date(2018, 4, 6), contact_type=StateSupervisionContactType.FACE_TO_FACE, status=StateSupervisionContactStatus.COMPLETED), ] start_of_supervision = supervision_period.start_date compliance_evaluation_date = date(2018, 4, 30) compliance = us_id_case_compliance_on_date( supervision_period, case_type, start_of_supervision, compliance_evaluation_date, assessments, supervision_contacts) self.assertEqual( SupervisionCaseCompliance( date_of_evaluation=compliance_evaluation_date, assessment_count=0, assessment_up_to_date=True, face_to_face_count=1, face_to_face_frequency_sufficient=True), compliance)
def test_same_dates(self) -> None: assessment_1 = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 28), assessment_score=17, external_id="2", ) assessment_2 = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 28), assessment_score=21, external_id="10", ) for pipeline in assessment_utils._ASSESSMENT_TYPES_TO_INCLUDE_FOR_CLASS: most_recent_assessment = assessment_utils.find_most_recent_applicable_assessment_of_class_for_state( date(2018, 4, 30), [assessment_1, assessment_2], pipeline, "US_XX") self.assertEqual(most_recent_assessment, assessment_2)
def test_person_and_kwargs_for_identifier_two_people_same_id(self): person_input_1 = StatePerson.new_with_defaults(person_id=123) person_input_2 = StatePerson.new_with_defaults(person_id=123) assessment = StateAssessment.new_with_defaults(state_code='US_XX') arg_to_entities_map = { # There should never be two StatePerson entities with the same person_id. This should fail loudly. 'person': iter([person_input_1, person_input_2]), 'assessments': iter([assessment]) } with pytest.raises(ValueError): _ = person_and_kwargs_for_identifier(arg_to_entities_map)
def test_find_most_recent_applicable_assessment_LSIR_no_matches(self, mock_assessment_types): mock_assessment_types.return_value = [StateAssessmentType.LSIR] assessment = StateAssessment.new_with_defaults( state_code='US_XX', assessment_type=StateAssessmentType.ORAS_COMMUNITY_SUPERVISION_SCREENING, assessment_date=date(2018, 4, 29), assessment_score=17 ) most_recent_assessment = assessment_utils.find_most_recent_applicable_assessment_of_class_for_state( date(2018, 4, 30), [assessment], StateAssessmentClass.RISK, 'US_XX') self.assertIsNone(most_recent_assessment)
def test_person_and_kwargs_for_identifier(self): person_input = StatePerson.new_with_defaults(person_id=123) assessment = StateAssessment.new_with_defaults(state_code='US_XX') arg_to_entities_map = { 'person': iter([person_input]), 'assessments': iter([assessment]) } person, kwargs = person_and_kwargs_for_identifier(arg_to_entities_map) expected_kwargs = {'assessments': [assessment]} self.assertEqual(person, person_input) self.assertEqual(expected_kwargs, kwargs)
def test_find_most_recent_applicable_assessment_no_assessment_types_for_pipeline( self, mock_assessment_types): mock_assessment_types.return_value = None assessment = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType. ORAS_COMMUNITY_SUPERVISION_SCREENING, assessment_date=date(2018, 4, 29), assessment_score=17, ) most_recent_assessment = ( assessment_utils. find_most_recent_applicable_assessment_of_class_for_state( date(2018, 4, 30), [assessment], StateAssessmentClass.RISK, "US_XX")) self.assertIsNone(most_recent_assessment, assessment)
def test_reassessment_requirements_are_not_met(self) -> None: start_of_supervision = date(2018, 3, 5) # This was a Monday supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, external_id="sp1", state_code=StateCode.US_ND.value, start_date=start_of_supervision, termination_date=date(2018, 5, 19), admission_reason=StateSupervisionPeriodAdmissionReason. COURT_SENTENCE, termination_reason=StateSupervisionPeriodTerminationReason. DISCHARGE, supervision_period_supervision_type= StateSupervisionPeriodSupervisionType.PROBATION, status=StateSupervisionPeriodStatus.PRESENT_WITHOUT_INFO, ) assessment_date = date(2010, 4, 2) assessment_score = 25 assessment = StateAssessment.new_with_defaults( state_code=StateCode.US_ND.value, assessment_type=StateAssessmentType.LSIR, assessment_date=assessment_date, assessment_score=assessment_score, ) us_nd_supervision_compliance = UsNdSupervisionCaseCompliance( self.person, supervision_period=supervision_period, case_type=StateSupervisionCaseType.GENERAL, start_of_supervision=start_of_supervision, assessments=[assessment], supervision_contacts=[], ) reassessment_deadline = ( us_nd_supervision_compliance._next_recommended_reassessment( assessment_date, assessment_score, )) self.assertEqual(reassessment_deadline, date(2010, 10, 31))
def test_reassessment_requirements_are_not_met(self): supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, external_id="sp1", state_code=StateCode.US_ND.value, start_date=date(2018, 3, 5), # This was a Monday termination_date=date(2018, 5, 19), admission_reason=StateSupervisionPeriodAdmissionReason. COURT_SENTENCE, termination_reason=StateSupervisionPeriodTerminationReason. DISCHARGE, supervision_period_supervision_type= StateSupervisionPeriodSupervisionType.PROBATION, status=StateSupervisionPeriodStatus.PRESENT_WITHOUT_INFO, ) assessment = StateAssessment.new_with_defaults( state_code=StateCode.US_ND.value, assessment_type=StateAssessmentType.LSIR, assessment_date=date(2010, 4, 2), ) start_of_supervision = supervision_period.start_date evaluation_date = start_of_supervision us_nd_supervision_compliance = UsNdSupervisionCaseCompliance( supervision_period=supervision_period, case_type=StateSupervisionCaseType.GENERAL, start_of_supervision=evaluation_date, assessments=[assessment], supervision_contacts=[], ) days_past_reassessment = ( us_nd_supervision_compliance._num_days_past_required_reassessment( evaluation_date, assessment.assessment_date, assessment.assessment_score, )) self.assertEqual(days_past_reassessment, 2682)
def test_find_program_referrals_officer_info_us_nd(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code="US_ND", program_id="PG3", referral_date=date(2009, 10, 3), participation_status=StateProgramAssignmentParticipationStatus. DISCHARGED, ) assessment = StateAssessment.new_with_defaults( state_code="US_ND", assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 7, 10), ) supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, status=StateSupervisionPeriodStatus.TERMINATED, state_code="US_ND", start_date=date(2008, 3, 5), termination_date=date(2010, 5, 19), termination_reason=StateSupervisionPeriodTerminationReason. DISCHARGE, supervision_type=StateSupervisionType.PAROLE, supervision_site="DISTRICT8", ) assessments = [assessment] supervision_periods = [supervision_period] supervision_period_agent_associations = { supervision_period.supervision_period_id: { "agent_id": 000, "agent_external_id": "OFFICER10", "supervision_period_id": supervision_period.supervision_period_id, } } program_referrals = identifier.find_program_referrals( program_assignment, assessments, supervision_periods, supervision_period_agent_associations, ) self.maxDiff = None self.assertListEqual( [ ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, participation_status=program_assignment. participation_status, assessment_score=33, assessment_type=StateAssessmentType.ORAS, supervision_type=supervision_period.supervision_type, supervising_officer_external_id="OFFICER10", supervising_district_external_id="DISTRICT8", level_1_supervision_location_external_id="DISTRICT8", level_2_supervision_location_external_id=None, ) ], program_referrals, )
def test_find_program_referrals_multiple_supervisions(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code='US_XX', program_id='PG3', referral_date=date(2009, 10, 3), participation_status=StateProgramAssignmentParticipationStatus.PENDING, ) assessment = StateAssessment.new_with_defaults( state_code='US_XX', assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2009, 7, 10) ) 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(2006, 12, 1), termination_date=date(2013, 1, 4), termination_reason=StateSupervisionPeriodTerminationReason.DISCHARGE, supervision_type=StateSupervisionType.PROBATION ) assessments = [assessment] supervision_periods = [supervision_period_1, supervision_period_2] program_referrals = identifier.find_program_referrals( program_assignment, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATIONS ) self.assertListEqual( [ ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, participation_status=program_assignment.participation_status, assessment_score=33, assessment_type=StateAssessmentType.ORAS, supervision_type=supervision_period_1.supervision_type ), ProgramReferralEvent( state_code=program_assignment.state_code, program_id=program_assignment.program_id, event_date=program_assignment.referral_date, participation_status=program_assignment.participation_status, assessment_score=33, assessment_type=StateAssessmentType.ORAS, supervision_type=supervision_period_2.supervision_type ) ], program_referrals)
def test_find_program_events(self): program_assignment = StateProgramAssignment.new_with_defaults( state_code="US_XX", program_id="PG3", referral_date=date(2020, 1, 3), participation_status=StateProgramAssignmentParticipationStatus. IN_PROGRESS, program_location_id="LOCATION X", start_date=date(2020, 1, 1), ) assessment = StateAssessment.new_with_defaults( state_code="US_XX", assessment_type=StateAssessmentType.ORAS, assessment_score=33, assessment_date=date(2019, 7, 10), ) supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=999, status=StateSupervisionPeriodStatus.TERMINATED, state_code="US_XX", start_date=date(2019, 3, 5), termination_date=date(2020, 10, 1), supervision_type=StateSupervisionType.PAROLE, supervision_site="OFFICE_1", ) program_assignments = [program_assignment] assessments = [assessment] supervision_periods = [supervision_period] program_events = identifier.find_program_events( program_assignments, assessments, supervision_periods, DEFAULT_SUPERVISION_PERIOD_AGENT_ASSOCIATION_LIST, ) expected_events = [ ProgramReferralEvent( state_code=program_assignment.state_code, event_date=program_assignment.referral_date, program_id=program_assignment.program_id, supervision_type=supervision_period.supervision_type, participation_status=program_assignment.participation_status, assessment_score=assessment.assessment_score, assessment_type=assessment.assessment_type, supervising_officer_external_id="XXX", supervising_district_external_id="OFFICE_1", level_1_supervision_location_external_id="OFFICE_1", level_2_supervision_location_external_id=None, ), ProgramParticipationEvent( state_code=program_assignment.state_code, event_date=program_assignment.start_date, is_first_day_in_program=True, program_id=program_assignment.program_id, program_location_id=program_assignment.program_location_id, supervision_type=supervision_period.supervision_type, ), ProgramParticipationEvent( state_code=program_assignment.state_code, event_date=program_assignment.start_date + relativedelta(days=1), is_first_day_in_program=False, program_id=program_assignment.program_id, program_location_id=program_assignment.program_location_id, supervision_type=supervision_period.supervision_type, ), ] self.assertListEqual(program_events, expected_events)
def test_completed_assessments_in_compliance_month(self) -> None: evaluation_date = date(2018, 4, 30) assessment_out_of_range = StateAssessment.new_with_defaults( state_code=StateCode.US_ND.value, assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 3, 10), ) assessment_out_of_range_2 = StateAssessment.new_with_defaults( state_code=StateCode.US_ND.value, assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 5, 10), ) assessment_1 = StateAssessment.new_with_defaults( state_code=StateCode.US_ND.value, assessment_type=StateAssessmentType.LSIR, assessment_score=1, assessment_date=date(2018, 4, 30), ) assessment_2 = StateAssessment.new_with_defaults( state_code=StateCode.US_ND.value, assessment_type=StateAssessmentType.LSIR, assessment_score=100, assessment_date=date(2018, 4, 30), ) assessment_no_score = StateAssessment.new_with_defaults( state_code=StateCode.US_ND.value, assessment_type=StateAssessmentType.LSIR, assessment_date=date(2018, 4, 28), ) supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=111, external_id="sp1", state_code=StateCode.US_ND.value, start_date=date(2018, 3, 5), # This was a Monday termination_date=date(2018, 5, 19), admission_reason=StateSupervisionPeriodAdmissionReason. COURT_SENTENCE, termination_reason=StateSupervisionPeriodTerminationReason. DISCHARGE, supervision_period_supervision_type= StateSupervisionPeriodSupervisionType.PROBATION, status=StateSupervisionPeriodStatus.PRESENT_WITHOUT_INFO, ) assessments = [ assessment_out_of_range, assessment_out_of_range_2, assessment_1, assessment_2, assessment_no_score, ] expected_assessments = [assessment_1, assessment_2] us_nd_supervision_compliance = UsNdSupervisionCaseCompliance( self.person, supervision_period=supervision_period, case_type=StateSupervisionCaseType.GENERAL, start_of_supervision=evaluation_date, assessments=assessments, supervision_contacts=[], ) self.assertEqual( len(expected_assessments), us_nd_supervision_compliance._completed_assessments_on_date( evaluation_date), )
def testConvert_FullIngestInfo(self): # Arrange metadata = IngestMetadata('us_nd', _JURISDICTION_ID, _INGEST_TIME, system_level=SystemLevel.STATE) ingest_info = IngestInfo() ingest_info.state_agents.add(state_agent_id='AGENT_ID1', full_name='AGENT WILLIAMS') ingest_info.state_agents.add(state_agent_id='AGENT_ID2', full_name='AGENT HERNANDEZ') ingest_info.state_agents.add(state_agent_id='AGENT_ID3', full_name='AGENT SMITH') ingest_info.state_agents.add(state_agent_id='JUDGE_AGENT_ID_1', full_name='JUDGE JUDY') # We expect the external_ids coming in to have the format # [type]:[external_id] ii_person_external_id_1 = US_ND_ELITE + ':' + 'EXTERNAL_ID1' ii_person_external_id_2 = US_ND_SID + ':' + 'EXTERNAL_ID2' ingest_info.state_people.add( state_person_id='PERSON_ID', state_person_race_ids=['RACE_ID1', 'RACE_ID2'], state_person_ethnicity_ids=['ETHNICITY_ID'], state_alias_ids=['ALIAS_ID1', 'ALIAS_ID2'], state_person_external_ids_ids=[ ii_person_external_id_1, ii_person_external_id_2 ], state_assessment_ids=['ASSESSMENT_ID'], state_sentence_group_ids=['GROUP_ID1', 'GROUP_ID2']) ingest_info.state_person_races.add( state_person_race_id='RACE_ID1', race='WHITE', ) ingest_info.state_person_races.add(state_person_race_id='RACE_ID2', race='OTHER') ingest_info.state_person_ethnicities.add( state_person_ethnicity_id='ETHNICITY_ID', ethnicity='HISPANIC') ingest_info.state_aliases.add(state_alias_id='ALIAS_ID1', full_name='LONNY BREAUX') ingest_info.state_aliases.add(state_alias_id='ALIAS_ID2', full_name='FRANK OCEAN') ingest_info.state_person_external_ids.add( state_person_external_id_id=ii_person_external_id_1, id_type=US_ND_ELITE) ingest_info.state_person_external_ids.add( state_person_external_id_id=ii_person_external_id_2, id_type=US_ND_SID) ingest_info.state_assessments.add(state_assessment_id='ASSESSMENT_ID', assessment_class='MENTAL_HEALTH', conducting_agent_id='AGENT_ID1') ingest_info.state_sentence_groups.add( state_sentence_group_id='GROUP_ID1', state_supervision_sentence_ids=['SUPERVISION_SENTENCE_ID1'], state_incarceration_sentence_ids=[ 'INCARCERATION_SENTENCE_ID1', 'INCARCERATION_SENTENCE_ID2' ]) ingest_info.state_sentence_groups.add( state_sentence_group_id='GROUP_ID2', state_supervision_sentence_ids=['SUPERVISION_SENTENCE_ID2'], state_fine_ids=['FINE_ID']) ingest_info.state_fines.add(state_fine_id='FINE_ID', status='PAID') ingest_info.state_supervision_sentences.add( state_supervision_sentence_id='SUPERVISION_SENTENCE_ID1', state_charge_ids=['CHARGE_ID1', 'CHARGE_ID2'], state_supervision_period_ids=['S_PERIOD_ID1']) ingest_info.state_supervision_sentences.add( state_supervision_sentence_id='SUPERVISION_SENTENCE_ID2', state_charge_ids=['CHARGE_ID2'], state_supervision_period_ids=['S_PERIOD_ID2']) ingest_info.state_incarceration_sentences.add( state_incarceration_sentence_id='INCARCERATION_SENTENCE_ID1', state_charge_ids=['CHARGE_ID1'], state_incarceration_period_ids=['I_PERIOD_ID']) ingest_info.state_incarceration_sentences.add( state_incarceration_sentence_id='INCARCERATION_SENTENCE_ID2', state_charge_ids=['CHARGE_ID2', 'CHARGE_ID3'], state_supervision_period_ids=['S_PERIOD_ID3']) ingest_info.state_charges.add( state_charge_id='CHARGE_ID1', state_court_case_id='CASE_ID', state_bond_id='BOND_ID', classification_type='M', classification_subtype='1', ) ingest_info.state_charges.add( state_charge_id='CHARGE_ID2', state_court_case_id='CASE_ID', classification_type='M', classification_subtype='2', ) ingest_info.state_charges.add(state_charge_id='CHARGE_ID3', state_court_case_id='CASE_ID', classification_type='F', classification_subtype='3') ingest_info.state_court_cases.add( state_court_case_id='CASE_ID', judge_id='JUDGE_AGENT_ID_1', ) ingest_info.state_bonds.add(state_bond_id='BOND_ID', status='POSTED') ingest_info.state_supervision_periods.add( state_supervision_period_id='S_PERIOD_ID1', state_supervision_violation_ids=['VIOLATION_ID'], supervision_type='PAROLE', supervision_level='MED') ingest_info.state_supervision_periods.add( state_supervision_period_id='S_PERIOD_ID2', supervision_type='PAROLE') ingest_info.state_supervision_periods.add( state_supervision_period_id='S_PERIOD_ID3', state_assessment_ids=['ASSESSMENT_ID'], supervision_type='PROBATION') ingest_info.state_incarceration_periods.add( state_incarceration_period_id='I_PERIOD_ID', state_incarceration_incident_ids=['INCIDENT_ID'], state_parole_decision_ids=['DECISION_ID'], state_assessment_ids=['ASSESSMENT_ID'], ) ingest_info.state_supervision_violations.add( state_supervision_violation_id='VIOLATION_ID', state_supervision_violation_response_ids=['RESPONSE_ID']) ingest_info.state_supervision_violation_responses.add( state_supervision_violation_response_id='RESPONSE_ID', response_type='CITATION') ingest_info.state_incarceration_incidents.add( state_incarceration_incident_id='INCIDENT_ID', incident_type='CONTRABAND', responding_officer_id='AGENT_ID2', state_incarceration_incident_outcome_ids=['INCIDENT_OUTCOME_ID'], ) ingest_info.state_incarceration_incident_outcomes.add( state_incarceration_incident_outcome_id='INCIDENT_OUTCOME_ID', outcome_type='GOOD_TIME_LOSS', date_effective='2/10/2018', state_code='US_ND', outcome_description='Good time', punishment_length_days='7', ) ingest_info.state_parole_decisions.add( state_parole_decision_id='DECISION_ID', decision_agent_ids=['AGENT_ID2', 'AGENT_ID3']) # Act result = self._convert_and_throw_on_errors(ingest_info, metadata) # Assert incident_outcome = StateIncarcerationIncidentOutcome.new_with_defaults( external_id='INCIDENT_OUTCOME_ID', outcome_type=StateIncarcerationIncidentOutcomeType.GOOD_TIME_LOSS, outcome_type_raw_text='GOOD_TIME_LOSS', date_effective=datetime.date(year=2018, month=2, day=10), state_code='US_ND', outcome_description='GOOD TIME', punishment_length_days=7, ) incident = StateIncarcerationIncident.new_with_defaults( external_id='INCIDENT_ID', state_code='US_ND', incident_type=StateIncarcerationIncidentType.CONTRABAND, incident_type_raw_text='CONTRABAND', responding_officer=StateAgent.new_with_defaults( external_id='AGENT_ID2', state_code='US_ND', full_name='{"full_name": "AGENT HERNANDEZ"}', ), incarceration_incident_outcomes=[incident_outcome]) assessment = StateAssessment.new_with_defaults( external_id='ASSESSMENT_ID', state_code='US_ND', assessment_class=StateAssessmentClass.MENTAL_HEALTH, assessment_class_raw_text='MENTAL_HEALTH', conducting_agent=StateAgent.new_with_defaults( external_id='AGENT_ID1', state_code='US_ND', full_name='{"full_name": "AGENT WILLIAMS"}', )) violation = StateSupervisionViolation.new_with_defaults( external_id='VIOLATION_ID', state_code='US_ND', supervision_violation_responses=[ StateSupervisionViolationResponse.new_with_defaults( external_id='RESPONSE_ID', state_code='US_ND', response_type=StateSupervisionViolationResponseType. CITATION, response_type_raw_text='CITATION') ]) court_case = StateCourtCase.new_with_defaults( external_id='CASE_ID', state_code='US_ND', status=StateCourtCaseStatus.PRESENT_WITHOUT_INFO, court_type=StateCourtType.PRESENT_WITHOUT_INFO, judge=StateAgent.new_with_defaults( external_id='JUDGE_AGENT_ID_1', state_code='US_ND', full_name='{"full_name": "JUDGE JUDY"}', )) charge_1 = StateCharge.new_with_defaults( external_id='CHARGE_ID1', classification_type=StateChargeClassificationType.MISDEMEANOR, classification_type_raw_text='M', classification_subtype='1', state_code='US_ND', status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case, bond=StateBond.new_with_defaults(external_id='BOND_ID', state_code='US_ND', status=BondStatus.POSTED, status_raw_text='POSTED')) charge_2 = StateCharge.new_with_defaults( external_id='CHARGE_ID2', classification_type=StateChargeClassificationType.MISDEMEANOR, classification_type_raw_text='M', classification_subtype='2', state_code='US_ND', status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case) charge_3 = StateCharge.new_with_defaults( external_id='CHARGE_ID3', state_code='US_ND', classification_type=StateChargeClassificationType.FELONY, classification_type_raw_text='F', classification_subtype='3', status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case) expected_result = [ StatePerson.new_with_defaults( external_ids=[ StatePersonExternalId.new_with_defaults( external_id='EXTERNAL_ID1', state_code='US_ND', id_type=US_ND_ELITE), StatePersonExternalId.new_with_defaults( external_id='EXTERNAL_ID2', state_code='US_ND', id_type=US_ND_SID) ], races=[ StatePersonRace(race=Race.WHITE, race_raw_text='WHITE', state_code='US_ND'), StatePersonRace(race=Race.OTHER, race_raw_text='OTHER', state_code='US_ND'), ], ethnicities=[ StatePersonEthnicity(ethnicity=Ethnicity.HISPANIC, ethnicity_raw_text='HISPANIC', state_code='US_ND') ], aliases=[ StatePersonAlias.new_with_defaults( full_name='{"full_name": "LONNY BREAUX"}', state_code='US_ND'), StatePersonAlias.new_with_defaults( full_name='{"full_name": "FRANK OCEAN"}', state_code='US_ND'), ], assessments=[assessment], sentence_groups=[ StateSentenceGroup.new_with_defaults( external_id='GROUP_ID1', status=StateSentenceStatus.PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_sentences=[ StateSupervisionSentence.new_with_defaults( external_id='SUPERVISION_SENTENCE_ID1', state_code='US_ND', status=StateSentenceStatus. PRESENT_WITHOUT_INFO, charges=[charge_1, charge_2], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id='S_PERIOD_ID1', status=StateSupervisionPeriodStatus. PRESENT_WITHOUT_INFO, supervision_level=StateSupervisionLevel .MEDIUM, supervision_level_raw_text='MED', state_code='US_ND', supervision_type=StateSupervisionType. PAROLE, supervision_type_raw_text='PAROLE', supervision_violations=[violation]) ]) ], incarceration_sentences=[ StateIncarcerationSentence.new_with_defaults( external_id='INCARCERATION_SENTENCE_ID1', state_code='US_ND', status=StateSentenceStatus. PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType. STATE_PRISON, charges=[charge_1], incarceration_periods=[ StateIncarcerationPeriod.new_with_defaults( external_id='I_PERIOD_ID', status=StateIncarcerationPeriodStatus. PRESENT_WITHOUT_INFO, incarceration_type= StateIncarcerationType.STATE_PRISON, state_code='US_ND', incarceration_incidents=[incident], parole_decisions=[ StateParoleDecision. new_with_defaults( external_id='DECISION_ID', state_code='US_ND', decision_agents=[ StateAgent. new_with_defaults( external_id='AGENT_ID2', state_code='US_ND', full_name= '{"full_name": ' '"AGENT HERNANDEZ"}'), StateAgent. new_with_defaults( external_id='AGENT_ID3', state_code='US_ND', full_name= '{"full_name": ' '"AGENT SMITH"}') ]) ], assessments=[assessment], ) ]), StateIncarcerationSentence.new_with_defaults( external_id='INCARCERATION_SENTENCE_ID2', state_code='US_ND', status=StateSentenceStatus. PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType. STATE_PRISON, charges=[charge_2, charge_3], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id='S_PERIOD_ID3', status=StateSupervisionPeriodStatus. PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_type=StateSupervisionType. PROBATION, supervision_type_raw_text='PROBATION', assessments=[assessment]) ]) ]), StateSentenceGroup.new_with_defaults( external_id='GROUP_ID2', status=StateSentenceStatus.PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_sentences=[ StateSupervisionSentence.new_with_defaults( external_id='SUPERVISION_SENTENCE_ID2', state_code='US_ND', status=StateSentenceStatus. PRESENT_WITHOUT_INFO, charges=[charge_2], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id='S_PERIOD_ID2', status=StateSupervisionPeriodStatus. PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_type=StateSupervisionType. PAROLE, supervision_type_raw_text='PAROLE', ) ]) ], fines=[ StateFine.new_with_defaults( external_id='FINE_ID', state_code='US_ND', status=StateFineStatus.PAID, status_raw_text='PAID') ]) ]) ] print(expected_result, "\n\n\n", result) self.assertEqual(result, expected_result)
def testConvert_FullIngestInfo(self): # Arrange metadata = IngestMetadata('us_nd', _JURISDICTION_ID, _INGEST_TIME, system_level=SystemLevel.STATE) ingest_info = IngestInfo() ingest_info.state_agents.add(state_agent_id='AGENT_ID1', full_name='AGENT WILLIAMS') ingest_info.state_agents.add(state_agent_id='AGENT_ID2', full_name='AGENT HERNANDEZ') ingest_info.state_agents.add(state_agent_id='AGENT_ID3', full_name='AGENT SMITH') ingest_info.state_agents.add(state_agent_id='AGENT_ID4', full_name='AGENT PO') ingest_info.state_agents.add(state_agent_id='JUDGE_AGENT_ID_1', full_name='JUDGE JUDY') ingest_info.state_agents.add(state_agent_id='AGENT_ID_PO', full_name='AGENT PAROLEY') ingest_info.state_agents.add(state_agent_id='AGENT_ID_TERM', full_name='AGENT TERMY', agent_type='SUPERVISION_OFFICER') ingest_info.state_agents.add( state_agent_id='AGENT_ID_SUPERVISING', full_name='SUPERVISING AGENT', ) # We expect the external_ids coming in to have the format # [type]:[external_id] ii_person_external_id_1 = US_ND_ELITE + ':' + 'EXTERNAL_ID1' ii_person_external_id_2 = US_ND_SID + ':' + 'EXTERNAL_ID2' ingest_info.state_people.add( state_person_id='PERSON_ID', state_person_race_ids=['RACE_ID1', 'RACE_ID2'], state_person_ethnicity_ids=['ETHNICITY_ID'], state_alias_ids=['ALIAS_ID1', 'ALIAS_ID2'], state_person_external_ids_ids=[ ii_person_external_id_1, ii_person_external_id_2 ], state_assessment_ids=['ASSESSMENT_ID'], state_program_assignment_ids=['PROGRAM_ASSIGNMENT_ID'], state_sentence_group_ids=['GROUP_ID1', 'GROUP_ID2'], supervising_officer_id='AGENT_ID_SUPERVISING', ) ingest_info.state_person_races.add( state_person_race_id='RACE_ID1', race='WHITE', ) ingest_info.state_person_races.add(state_person_race_id='RACE_ID2', race='OTHER') ingest_info.state_person_ethnicities.add( state_person_ethnicity_id='ETHNICITY_ID', ethnicity='HISPANIC') ingest_info.state_aliases.add(state_alias_id='ALIAS_ID1', full_name='LONNY BREAUX') ingest_info.state_aliases.add(state_alias_id='ALIAS_ID2', full_name='FRANK OCEAN') ingest_info.state_person_external_ids.add( state_person_external_id_id=ii_person_external_id_1, id_type=US_ND_ELITE) ingest_info.state_person_external_ids.add( state_person_external_id_id=ii_person_external_id_2, id_type=US_ND_SID) ingest_info.state_assessments.add(state_assessment_id='ASSESSMENT_ID', assessment_class='MENTAL_HEALTH', conducting_agent_id='AGENT_ID1') ingest_info.state_program_assignments.add( state_program_assignment_id='PROGRAM_ASSIGNMENT_ID', participation_status='DISCHARGED', referral_date='2019/02/10', start_date='2019/02/11', discharge_date='2019/02/12', program_id='PROGRAM_ID', program_location_id='PROGRAM_LOCATION_ID', discharge_reason='COMPLETED', referring_agent_id='AGENT_ID4') ingest_info.state_sentence_groups.add( state_sentence_group_id='GROUP_ID1', state_supervision_sentence_ids=['SUPERVISION_SENTENCE_ID1'], state_incarceration_sentence_ids=[ 'INCARCERATION_SENTENCE_ID1', 'INCARCERATION_SENTENCE_ID2' ]) ingest_info.state_sentence_groups.add( state_sentence_group_id='GROUP_ID2', state_supervision_sentence_ids=['SUPERVISION_SENTENCE_ID2'], state_fine_ids=['FINE_ID']) ingest_info.state_fines.add(state_fine_id='FINE_ID', status='PAID') ingest_info.state_supervision_sentences.add( state_supervision_sentence_id='SUPERVISION_SENTENCE_ID1', state_charge_ids=['CHARGE_ID1', 'CHARGE_ID2'], state_supervision_period_ids=['S_PERIOD_ID1']) ingest_info.state_supervision_sentences.add( state_supervision_sentence_id='SUPERVISION_SENTENCE_ID2', state_charge_ids=['CHARGE_ID2'], state_supervision_period_ids=['S_PERIOD_ID2']) ingest_info.state_incarceration_sentences.add( state_incarceration_sentence_id='INCARCERATION_SENTENCE_ID1', state_charge_ids=['CHARGE_ID1'], state_incarceration_period_ids=['I_PERIOD_ID']) ingest_info.state_incarceration_sentences.add( state_incarceration_sentence_id='INCARCERATION_SENTENCE_ID2', state_charge_ids=['CHARGE_ID2', 'CHARGE_ID3'], state_supervision_period_ids=['S_PERIOD_ID3']) ingest_info.state_charges.add( state_charge_id='CHARGE_ID1', state_court_case_id='CASE_ID', state_bond_id='BOND_ID', classification_type='M', classification_subtype='1', ncic_code='5006', ) ingest_info.state_charges.add( state_charge_id='CHARGE_ID2', state_court_case_id='CASE_ID', classification_type='M', classification_subtype='2', ) ingest_info.state_charges.add( state_charge_id='CHARGE_ID3', state_court_case_id='CASE_ID', classification_type='F', classification_subtype='3', ncic_code='5006', description='Obstruction of investigation', ) ingest_info.state_court_cases.add( state_court_case_id='CASE_ID', judge_id='JUDGE_AGENT_ID_1', ) ingest_info.state_bonds.add(state_bond_id='BOND_ID', status='POSTED') ingest_info.state_supervision_periods.add( state_supervision_period_id='S_PERIOD_ID1', state_supervision_violation_entry_ids=['VIOLATION_ID'], supervision_type='PAROLE', supervision_level='MED', state_supervision_contact_ids=['SUPERVISION_CONTACT_ID'], state_program_assignment_ids=['PROGRAM_ASSIGNMENT_ID']) ingest_info.state_supervision_periods.add( state_supervision_period_id='S_PERIOD_ID2', supervision_type='PAROLE') ingest_info.state_supervision_periods.add( state_supervision_period_id='S_PERIOD_ID3', state_assessment_ids=['ASSESSMENT_ID'], supervising_officer_id='AGENT_ID_PO', supervision_type='PROBATION', state_supervision_case_type_entry_ids=['CASE_TYPE_ID'], ) ingest_info.state_supervision_case_type_entries.add( state_supervision_case_type_entry_id='CASE_TYPE_ID', case_type='DOMESTIC_VIOLENCE') ingest_info.state_incarceration_periods.add( state_incarceration_period_id='I_PERIOD_ID', state_incarceration_incident_ids=['INCIDENT_ID'], state_parole_decision_ids=['DECISION_ID'], state_assessment_ids=['ASSESSMENT_ID'], state_program_assignment_ids=['PROGRAM_ASSIGNMENT_ID'], source_supervision_violation_response_id='RESPONSE_ID') ingest_info.state_supervision_violation_type_entries.add( state_supervision_violation_type_entry_id='VIOLATION_TYPE_ENTRY_ID', violation_type='FELONY', state_code='US_ND') ingest_info.state_supervision_violated_condition_entries.add( state_supervision_violated_condition_entry_id= 'VIOLATED_CONDITION_ENTRY_ID', condition='CURFEW', state_code='US_ND') ingest_info.state_supervision_violations.add( state_supervision_violation_id='VIOLATION_ID', state_supervision_violation_response_ids=['RESPONSE_ID'], state_supervision_violated_condition_entry_ids=[ 'VIOLATED_CONDITION_ENTRY_ID' ], state_supervision_violation_type_entry_ids=[ 'VIOLATION_TYPE_ENTRY_ID' ], ) ingest_info.state_supervision_violated_condition_entries.add( state_supervision_violated_condition_entry_id= 'VIOLATED_CONDITION_ENTRY_ID', condition='CURFEW', state_code='US_ND') ingest_info.state_supervision_violation_response_decision_entries.add( state_supervision_violation_response_decision_entry_id= 'VIOLATION_RESPONSE_DECISION_ENTRY_ID', decision='REVOCATION', revocation_type='REINCARCERATION', state_code='US_ND') ingest_info.state_supervision_violation_responses.add( state_supervision_violation_response_id='RESPONSE_ID', decision_agent_ids=['AGENT_ID_TERM'], state_supervision_violation_response_decision_entry_ids=[ 'VIOLATION_RESPONSE_DECISION_ENTRY_ID' ], response_type='CITATION') ingest_info.state_incarceration_incidents.add( state_incarceration_incident_id='INCIDENT_ID', incident_type='CONTRABAND', responding_officer_id='AGENT_ID2', state_incarceration_incident_outcome_ids=['INCIDENT_OUTCOME_ID'], ) ingest_info.state_incarceration_incident_outcomes.add( state_incarceration_incident_outcome_id='INCIDENT_OUTCOME_ID', outcome_type='GOOD_TIME_LOSS', date_effective='2/10/2018', hearing_date='2/6/2018', report_date='2/8/2018', state_code='US_ND', outcome_description='Good time', punishment_length_days='7', ) ingest_info.state_parole_decisions.add( state_parole_decision_id='DECISION_ID', decision_agent_ids=['AGENT_ID2', 'AGENT_ID3']) ingest_info.state_supervision_contacts.add( state_supervision_contact_id='SUPERVISION_CONTACT_ID', contacted_agent_id='AGENT_ID_PO') # Act result = self._convert_and_throw_on_errors(ingest_info, metadata) # Assert supervision_contact = StateSupervisionContact.new_with_defaults( external_id='SUPERVISION_CONTACT_ID', state_code='US_ND', contacted_agent=StateAgent.new_with_defaults( external_id='AGENT_ID_PO', state_code='US_ND', agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT PAROLEY"}'), ) incident_outcome = StateIncarcerationIncidentOutcome.new_with_defaults( external_id='INCIDENT_OUTCOME_ID', outcome_type=StateIncarcerationIncidentOutcomeType.GOOD_TIME_LOSS, outcome_type_raw_text='GOOD_TIME_LOSS', date_effective=datetime.date(year=2018, month=2, day=10), hearing_date=datetime.date(year=2018, month=2, day=6), report_date=datetime.date(year=2018, month=2, day=8), state_code='US_ND', outcome_description='GOOD TIME', punishment_length_days=7, ) incident = StateIncarcerationIncident.new_with_defaults( external_id='INCIDENT_ID', state_code='US_ND', incident_type=StateIncarcerationIncidentType.CONTRABAND, incident_type_raw_text='CONTRABAND', responding_officer=StateAgent.new_with_defaults( external_id='AGENT_ID2', state_code='US_ND', agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT HERNANDEZ"}', ), incarceration_incident_outcomes=[incident_outcome]) assessment = StateAssessment.new_with_defaults( external_id='ASSESSMENT_ID', state_code='US_ND', assessment_class=StateAssessmentClass.MENTAL_HEALTH, assessment_class_raw_text='MENTAL_HEALTH', conducting_agent=StateAgent.new_with_defaults( external_id='AGENT_ID1', agent_type=StateAgentType.PRESENT_WITHOUT_INFO, state_code='US_ND', full_name='{"full_name": "AGENT WILLIAMS"}', )) program_assignment = StateProgramAssignment.new_with_defaults( external_id='PROGRAM_ASSIGNMENT_ID', state_code='US_ND', participation_status=StateProgramAssignmentParticipationStatus. DISCHARGED, participation_status_raw_text='DISCHARGED', referral_date=datetime.date(year=2019, month=2, day=10), start_date=datetime.date(year=2019, month=2, day=11), discharge_date=datetime.date(year=2019, month=2, day=12), program_id='PROGRAM_ID', program_location_id='PROGRAM_LOCATION_ID', discharge_reason=StateProgramAssignmentDischargeReason.COMPLETED, discharge_reason_raw_text='COMPLETED', referring_agent=StateAgent.new_with_defaults( external_id='AGENT_ID4', agent_type=StateAgentType.PRESENT_WITHOUT_INFO, state_code='US_ND', full_name='{"full_name": "AGENT PO"}')) response = StateSupervisionViolationResponse.new_with_defaults( external_id='RESPONSE_ID', state_code='US_ND', response_type=StateSupervisionViolationResponseType.CITATION, response_type_raw_text='CITATION', decision_agents=[ StateAgent.new_with_defaults( external_id='AGENT_ID_TERM', state_code='US_ND', full_name='{"full_name": "AGENT TERMY"}', agent_type=StateAgentType.SUPERVISION_OFFICER, agent_type_raw_text='SUPERVISION_OFFICER', ) ], supervision_violation_response_decisions=[ StateSupervisionViolationResponseDecisionEntry. new_with_defaults( state_code='US_ND', decision=StateSupervisionViolationResponseDecision. REVOCATION, decision_raw_text='REVOCATION', revocation_type= StateSupervisionViolationResponseRevocationType. REINCARCERATION, revocation_type_raw_text='REINCARCERATION') ]) violation = StateSupervisionViolation.new_with_defaults( external_id='VIOLATION_ID', state_code='US_ND', supervision_violation_responses=[response], supervision_violation_types=[ StateSupervisionViolationTypeEntry.new_with_defaults( state_code='US_ND', violation_type=StateSupervisionViolationType.FELONY, violation_type_raw_text='FELONY', ) ], supervision_violated_conditions=[ StateSupervisionViolatedConditionEntry.new_with_defaults( state_code='US_ND', condition='CURFEW', ) ]) court_case = StateCourtCase.new_with_defaults( external_id='CASE_ID', state_code='US_ND', status=StateCourtCaseStatus.PRESENT_WITHOUT_INFO, court_type=StateCourtType.PRESENT_WITHOUT_INFO, judge=StateAgent.new_with_defaults( external_id='JUDGE_AGENT_ID_1', state_code='US_ND', agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "JUDGE JUDY"}', )) charge_1 = StateCharge.new_with_defaults( external_id='CHARGE_ID1', classification_type=StateChargeClassificationType.MISDEMEANOR, classification_type_raw_text='M', classification_subtype='1', ncic_code='5006', description='FALSE STATEMENT', state_code='US_ND', status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case, bond=StateBond.new_with_defaults(external_id='BOND_ID', state_code='US_ND', status=BondStatus.POSTED, status_raw_text='POSTED')) charge_2 = StateCharge.new_with_defaults( external_id='CHARGE_ID2', classification_type=StateChargeClassificationType.MISDEMEANOR, classification_type_raw_text='M', classification_subtype='2', state_code='US_ND', status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case) charge_3 = StateCharge.new_with_defaults( external_id='CHARGE_ID3', state_code='US_ND', classification_type=StateChargeClassificationType.FELONY, classification_type_raw_text='F', classification_subtype='3', ncic_code='5006', description='OBSTRUCTION OF INVESTIGATION', status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case) incarceration_sentence_1 = StateIncarcerationSentence.new_with_defaults( external_id='INCARCERATION_SENTENCE_ID1', state_code='US_ND', status=StateSentenceStatus.PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType.STATE_PRISON, charges=[charge_1], incarceration_periods=[ StateIncarcerationPeriod.new_with_defaults( external_id='I_PERIOD_ID', status=StateIncarcerationPeriodStatus.PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType.STATE_PRISON, state_code='US_ND', incarceration_incidents=[incident], program_assignments=[program_assignment], parole_decisions=[ StateParoleDecision.new_with_defaults( external_id='DECISION_ID', state_code='US_ND', decision_agents=[ StateAgent.new_with_defaults( external_id='AGENT_ID2', agent_type=StateAgentType. PRESENT_WITHOUT_INFO, state_code='US_ND', full_name='{"full_name": "AGENT HERNANDEZ"}' ), StateAgent.new_with_defaults( external_id='AGENT_ID3', state_code='US_ND', agent_type=StateAgentType. PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT SMITH"}') ]) ], assessments=[assessment], source_supervision_violation_response=response, ) ]) incarceration_sentence_2 = StateIncarcerationSentence.new_with_defaults( external_id='INCARCERATION_SENTENCE_ID2', state_code='US_ND', status=StateSentenceStatus.PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType.STATE_PRISON, charges=[charge_2, charge_3], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id='S_PERIOD_ID3', status=StateSupervisionPeriodStatus.PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_type=StateSupervisionType.PROBATION, supervision_type_raw_text='PROBATION', assessments=[assessment], supervising_officer=StateAgent.new_with_defaults( external_id='AGENT_ID_PO', state_code='US_ND', agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT PAROLEY"}', ), case_type_entries=[ StateSupervisionCaseTypeEntry.new_with_defaults( case_type=StateSupervisionCaseType. DOMESTIC_VIOLENCE, case_type_raw_text='DOMESTIC_VIOLENCE', state_code='US_ND', external_id='CASE_TYPE_ID') ]) ]) expected_result = [ StatePerson.new_with_defaults( state_code='US_ND', external_ids=[ StatePersonExternalId.new_with_defaults( external_id='EXTERNAL_ID1', state_code='US_ND', id_type=US_ND_ELITE), StatePersonExternalId.new_with_defaults( external_id='EXTERNAL_ID2', state_code='US_ND', id_type=US_ND_SID) ], races=[ StatePersonRace(race=Race.WHITE, race_raw_text='WHITE', state_code='US_ND'), StatePersonRace(race=Race.OTHER, race_raw_text='OTHER', state_code='US_ND'), ], ethnicities=[ StatePersonEthnicity(ethnicity=Ethnicity.HISPANIC, ethnicity_raw_text='HISPANIC', state_code='US_ND') ], aliases=[ StatePersonAlias.new_with_defaults( full_name='{"full_name": "LONNY BREAUX"}', state_code='US_ND'), StatePersonAlias.new_with_defaults( full_name='{"full_name": "FRANK OCEAN"}', state_code='US_ND'), ], supervising_officer=StateAgent.new_with_defaults( external_id='AGENT_ID_SUPERVISING', state_code='US_ND', agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "SUPERVISING AGENT"}'), assessments=[assessment], program_assignments=[program_assignment], sentence_groups=[ StateSentenceGroup.new_with_defaults( external_id='GROUP_ID1', status=StateSentenceStatus.PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_sentences=[ StateSupervisionSentence.new_with_defaults( external_id='SUPERVISION_SENTENCE_ID1', state_code='US_ND', status=StateSentenceStatus. PRESENT_WITHOUT_INFO, charges=[charge_1, charge_2], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id='S_PERIOD_ID1', status=StateSupervisionPeriodStatus. PRESENT_WITHOUT_INFO, supervision_level=StateSupervisionLevel .MEDIUM, supervision_level_raw_text='MED', state_code='US_ND', supervision_type=StateSupervisionType. PAROLE, supervision_type_raw_text='PAROLE', supervision_violation_entries=[ violation ], program_assignments=[ program_assignment ], supervision_contacts=[ supervision_contact ], ) ]) ], incarceration_sentences=[ incarceration_sentence_1, incarceration_sentence_2 ]), StateSentenceGroup.new_with_defaults( external_id='GROUP_ID2', status=StateSentenceStatus.PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_sentences=[ StateSupervisionSentence.new_with_defaults( external_id='SUPERVISION_SENTENCE_ID2', state_code='US_ND', status=StateSentenceStatus. PRESENT_WITHOUT_INFO, charges=[charge_2], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id='S_PERIOD_ID2', status=StateSupervisionPeriodStatus. PRESENT_WITHOUT_INFO, state_code='US_ND', supervision_type=StateSupervisionType. PAROLE, supervision_type_raw_text='PAROLE', ) ]) ], fines=[ StateFine.new_with_defaults( external_id='FINE_ID', state_code='US_ND', status=StateFineStatus.PAID, status_raw_text='PAID') ]) ]) ] print("\n", expected_result, "\n\n\n", result) self.assertCountEqual(expected_result, result)
def testConvert_FullIngestInfo(self): # Arrange metadata = FakeIngestMetadata.for_state(region="us_nd") ingest_info = IngestInfo() ingest_info.state_agents.add(state_agent_id="AGENT_ID1", full_name="AGENT WILLIAMS") ingest_info.state_agents.add(state_agent_id="AGENT_ID2", full_name="AGENT HERNANDEZ") ingest_info.state_agents.add(state_agent_id="AGENT_ID3", full_name="AGENT SMITH") ingest_info.state_agents.add(state_agent_id="AGENT_ID4", full_name="AGENT PO") ingest_info.state_agents.add(state_agent_id="JUDGE_AGENT_ID_1", full_name="JUDGE JUDY") ingest_info.state_agents.add(state_agent_id="AGENT_ID_PO", full_name="AGENT PAROLEY") ingest_info.state_agents.add( state_agent_id="AGENT_ID_TERM", full_name="AGENT TERMY", agent_type="SUPERVISION_OFFICER", ) ingest_info.state_agents.add( state_agent_id="AGENT_ID_SUPERVISING", full_name="SUPERVISING AGENT", ) # We expect the external_ids coming in to have the format # [type]:[external_id] ii_person_external_id_1 = US_ND_ELITE + ":" + "EXTERNAL_ID1" ii_person_external_id_2 = US_ND_SID + ":" + "EXTERNAL_ID2" ingest_info.state_people.add( state_person_id="PERSON_ID", state_person_race_ids=["RACE_ID1", "RACE_ID2"], state_person_ethnicity_ids=["ETHNICITY_ID"], state_alias_ids=["ALIAS_ID1", "ALIAS_ID2"], state_person_external_ids_ids=[ ii_person_external_id_1, ii_person_external_id_2, ], state_assessment_ids=["ASSESSMENT_ID"], state_program_assignment_ids=["PROGRAM_ASSIGNMENT_ID"], state_sentence_group_ids=["GROUP_ID1", "GROUP_ID2"], supervising_officer_id="AGENT_ID_SUPERVISING", ) ingest_info.state_person_races.add( state_person_race_id="RACE_ID1", race="WHITE", ) ingest_info.state_person_races.add(state_person_race_id="RACE_ID2", race="OTHER") ingest_info.state_person_ethnicities.add( state_person_ethnicity_id="ETHNICITY_ID", ethnicity="HISPANIC") ingest_info.state_aliases.add(state_alias_id="ALIAS_ID1", full_name="LONNY BREAUX") ingest_info.state_aliases.add(state_alias_id="ALIAS_ID2", full_name="FRANK OCEAN") ingest_info.state_person_external_ids.add( state_person_external_id_id=ii_person_external_id_1, id_type=US_ND_ELITE) ingest_info.state_person_external_ids.add( state_person_external_id_id=ii_person_external_id_2, id_type=US_ND_SID) ingest_info.state_assessments.add( state_assessment_id="ASSESSMENT_ID", assessment_class="MENTAL_HEALTH", conducting_agent_id="AGENT_ID1", ) ingest_info.state_program_assignments.add( state_program_assignment_id="PROGRAM_ASSIGNMENT_ID", participation_status="DISCHARGED", referral_date="2019/02/10", start_date="2019/02/11", discharge_date="2019/02/12", program_id="PROGRAM_ID", program_location_id="PROGRAM_LOCATION_ID", discharge_reason="COMPLETED", referring_agent_id="AGENT_ID4", ) ingest_info.state_sentence_groups.add( state_sentence_group_id="GROUP_ID1", state_supervision_sentence_ids=["SUPERVISION_SENTENCE_ID1"], state_incarceration_sentence_ids=[ "INCARCERATION_SENTENCE_ID1", "INCARCERATION_SENTENCE_ID2", ], ) ingest_info.state_sentence_groups.add( state_sentence_group_id="GROUP_ID2", state_supervision_sentence_ids=["SUPERVISION_SENTENCE_ID2"], state_fine_ids=["FINE_ID"], ) ingest_info.state_fines.add(state_fine_id="FINE_ID", status="PAID") ingest_info.state_supervision_sentences.add( state_supervision_sentence_id="SUPERVISION_SENTENCE_ID1", state_charge_ids=["CHARGE_ID1", "CHARGE_ID2"], state_supervision_period_ids=["S_PERIOD_ID1"], ) ingest_info.state_supervision_sentences.add( state_supervision_sentence_id="SUPERVISION_SENTENCE_ID2", state_charge_ids=["CHARGE_ID2"], state_supervision_period_ids=["S_PERIOD_ID2"], ) ingest_info.state_incarceration_sentences.add( state_incarceration_sentence_id="INCARCERATION_SENTENCE_ID1", state_charge_ids=["CHARGE_ID1"], state_incarceration_period_ids=["I_PERIOD_ID"], ) ingest_info.state_incarceration_sentences.add( state_incarceration_sentence_id="INCARCERATION_SENTENCE_ID2", state_charge_ids=["CHARGE_ID2", "CHARGE_ID3"], state_supervision_period_ids=["S_PERIOD_ID3"], ) ingest_info.state_charges.add( state_charge_id="CHARGE_ID1", state_court_case_id="CASE_ID", state_bond_id="BOND_ID", classification_type="M", classification_subtype="1", ncic_code="5006", ) ingest_info.state_charges.add( state_charge_id="CHARGE_ID2", state_court_case_id="CASE_ID", classification_type="M", classification_subtype="2", ) ingest_info.state_charges.add( state_charge_id="CHARGE_ID3", state_court_case_id="CASE_ID", classification_type="F", classification_subtype="3", ncic_code="5006", description="Obstruction of investigation", ) ingest_info.state_court_cases.add( state_court_case_id="CASE_ID", judge_id="JUDGE_AGENT_ID_1", ) ingest_info.state_bonds.add(state_bond_id="BOND_ID", status="POSTED") ingest_info.state_supervision_periods.add( state_supervision_period_id="S_PERIOD_ID1", state_supervision_violation_entry_ids=["VIOLATION_ID"], supervision_type="PAROLE", supervision_level="MED", state_supervision_contact_ids=["SUPERVISION_CONTACT_ID"], state_program_assignment_ids=["PROGRAM_ASSIGNMENT_ID"], ) ingest_info.state_supervision_periods.add( state_supervision_period_id="S_PERIOD_ID2", supervision_type="PAROLE") ingest_info.state_supervision_periods.add( state_supervision_period_id="S_PERIOD_ID3", state_assessment_ids=["ASSESSMENT_ID"], supervising_officer_id="AGENT_ID_PO", supervision_type="PROBATION", state_supervision_case_type_entry_ids=["CASE_TYPE_ID"], ) ingest_info.state_supervision_case_type_entries.add( state_supervision_case_type_entry_id="CASE_TYPE_ID", case_type="DOMESTIC_VIOLENCE", ) ingest_info.state_incarceration_periods.add( state_incarceration_period_id="I_PERIOD_ID", state_incarceration_incident_ids=["INCIDENT_ID"], state_parole_decision_ids=["DECISION_ID"], state_assessment_ids=["ASSESSMENT_ID"], state_program_assignment_ids=["PROGRAM_ASSIGNMENT_ID"], source_supervision_violation_response_id="RESPONSE_ID", ) ingest_info.state_supervision_violation_type_entries.add( state_supervision_violation_type_entry_id="VIOLATION_TYPE_ENTRY_ID", violation_type="FELONY", state_code="US_ND", ) ingest_info.state_supervision_violated_condition_entries.add( state_supervision_violated_condition_entry_id= "VIOLATED_CONDITION_ENTRY_ID", condition="CURFEW", state_code="US_ND", ) ingest_info.state_supervision_violations.add( state_supervision_violation_id="VIOLATION_ID", state_supervision_violation_response_ids=["RESPONSE_ID"], state_supervision_violated_condition_entry_ids=[ "VIOLATED_CONDITION_ENTRY_ID" ], state_supervision_violation_type_entry_ids=[ "VIOLATION_TYPE_ENTRY_ID" ], ) ingest_info.state_supervision_violated_condition_entries.add( state_supervision_violated_condition_entry_id= "VIOLATED_CONDITION_ENTRY_ID", condition="CURFEW", state_code="US_ND", ) ingest_info.state_supervision_violation_response_decision_entries.add( state_supervision_violation_response_decision_entry_id= "VIOLATION_RESPONSE_DECISION_ENTRY_ID", decision="REVOCATION", revocation_type="REINCARCERATION", state_code="US_ND", ) ingest_info.state_supervision_violation_responses.add( state_supervision_violation_response_id="RESPONSE_ID", decision_agent_ids=["AGENT_ID_TERM"], state_supervision_violation_response_decision_entry_ids=[ "VIOLATION_RESPONSE_DECISION_ENTRY_ID" ], response_type="CITATION", ) ingest_info.state_incarceration_incidents.add( state_incarceration_incident_id="INCIDENT_ID", incident_type="CONTRABAND", responding_officer_id="AGENT_ID2", state_incarceration_incident_outcome_ids=["INCIDENT_OUTCOME_ID"], ) ingest_info.state_incarceration_incident_outcomes.add( state_incarceration_incident_outcome_id="INCIDENT_OUTCOME_ID", outcome_type="GOOD_TIME_LOSS", date_effective="2/10/2018", hearing_date="2/6/2018", report_date="2/8/2018", state_code="US_ND", outcome_description="Good time", punishment_length_days="7", ) ingest_info.state_parole_decisions.add( state_parole_decision_id="DECISION_ID", decision_agent_ids=["AGENT_ID2", "AGENT_ID3"], ) ingest_info.state_supervision_contacts.add( state_supervision_contact_id="SUPERVISION_CONTACT_ID", contacted_agent_id="AGENT_ID_PO", ) # Act result = self._convert_and_throw_on_errors(ingest_info, metadata) # Assert supervision_contact = StateSupervisionContact.new_with_defaults( external_id="SUPERVISION_CONTACT_ID", state_code="US_ND", contacted_agent=StateAgent.new_with_defaults( external_id="AGENT_ID_PO", state_code="US_ND", agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT PAROLEY"}', ), ) incident_outcome = StateIncarcerationIncidentOutcome.new_with_defaults( external_id="INCIDENT_OUTCOME_ID", outcome_type=StateIncarcerationIncidentOutcomeType.GOOD_TIME_LOSS, outcome_type_raw_text="GOOD_TIME_LOSS", date_effective=datetime.date(year=2018, month=2, day=10), hearing_date=datetime.date(year=2018, month=2, day=6), report_date=datetime.date(year=2018, month=2, day=8), state_code="US_ND", outcome_description="GOOD TIME", punishment_length_days=7, ) incident = StateIncarcerationIncident.new_with_defaults( external_id="INCIDENT_ID", state_code="US_ND", incident_type=StateIncarcerationIncidentType.CONTRABAND, incident_type_raw_text="CONTRABAND", responding_officer=StateAgent.new_with_defaults( external_id="AGENT_ID2", state_code="US_ND", agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT HERNANDEZ"}', ), incarceration_incident_outcomes=[incident_outcome], ) assessment = StateAssessment.new_with_defaults( external_id="ASSESSMENT_ID", state_code="US_ND", assessment_class=StateAssessmentClass.MENTAL_HEALTH, assessment_class_raw_text="MENTAL_HEALTH", conducting_agent=StateAgent.new_with_defaults( external_id="AGENT_ID1", agent_type=StateAgentType.PRESENT_WITHOUT_INFO, state_code="US_ND", full_name='{"full_name": "AGENT WILLIAMS"}', ), ) program_assignment = StateProgramAssignment.new_with_defaults( external_id="PROGRAM_ASSIGNMENT_ID", state_code="US_ND", participation_status=StateProgramAssignmentParticipationStatus. DISCHARGED, participation_status_raw_text="DISCHARGED", referral_date=datetime.date(year=2019, month=2, day=10), start_date=datetime.date(year=2019, month=2, day=11), discharge_date=datetime.date(year=2019, month=2, day=12), program_id="PROGRAM_ID", program_location_id="PROGRAM_LOCATION_ID", discharge_reason=StateProgramAssignmentDischargeReason.COMPLETED, discharge_reason_raw_text="COMPLETED", referring_agent=StateAgent.new_with_defaults( external_id="AGENT_ID4", agent_type=StateAgentType.PRESENT_WITHOUT_INFO, state_code="US_ND", full_name='{"full_name": "AGENT PO"}', ), ) response = StateSupervisionViolationResponse.new_with_defaults( external_id="RESPONSE_ID", state_code="US_ND", response_type=StateSupervisionViolationResponseType.CITATION, response_type_raw_text="CITATION", decision_agents=[ StateAgent.new_with_defaults( external_id="AGENT_ID_TERM", state_code="US_ND", full_name='{"full_name": "AGENT TERMY"}', agent_type=StateAgentType.SUPERVISION_OFFICER, agent_type_raw_text="SUPERVISION_OFFICER", ) ], supervision_violation_response_decisions=[ StateSupervisionViolationResponseDecisionEntry. new_with_defaults( state_code="US_ND", decision=StateSupervisionViolationResponseDecision. REVOCATION, decision_raw_text="REVOCATION", revocation_type= StateSupervisionViolationResponseRevocationType. REINCARCERATION, revocation_type_raw_text="REINCARCERATION", ) ], ) violation = StateSupervisionViolation.new_with_defaults( external_id="VIOLATION_ID", state_code="US_ND", supervision_violation_responses=[response], supervision_violation_types=[ StateSupervisionViolationTypeEntry.new_with_defaults( state_code="US_ND", violation_type=StateSupervisionViolationType.FELONY, violation_type_raw_text="FELONY", ) ], supervision_violated_conditions=[ StateSupervisionViolatedConditionEntry.new_with_defaults( state_code="US_ND", condition="CURFEW", ) ], ) court_case = StateCourtCase.new_with_defaults( external_id="CASE_ID", state_code="US_ND", status=StateCourtCaseStatus.PRESENT_WITHOUT_INFO, court_type=StateCourtType.PRESENT_WITHOUT_INFO, judge=StateAgent.new_with_defaults( external_id="JUDGE_AGENT_ID_1", state_code="US_ND", agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "JUDGE JUDY"}', ), ) charge_1 = StateCharge.new_with_defaults( external_id="CHARGE_ID1", classification_type=StateChargeClassificationType.MISDEMEANOR, classification_type_raw_text="M", classification_subtype="1", ncic_code="5006", description="FALSE STATEMENT", state_code="US_ND", status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case, bond=StateBond.new_with_defaults( external_id="BOND_ID", state_code="US_ND", status=BondStatus.POSTED, status_raw_text="POSTED", ), ) charge_2 = StateCharge.new_with_defaults( external_id="CHARGE_ID2", classification_type=StateChargeClassificationType.MISDEMEANOR, classification_type_raw_text="M", classification_subtype="2", state_code="US_ND", status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case, ) charge_3 = StateCharge.new_with_defaults( external_id="CHARGE_ID3", state_code="US_ND", classification_type=StateChargeClassificationType.FELONY, classification_type_raw_text="F", classification_subtype="3", ncic_code="5006", description="OBSTRUCTION OF INVESTIGATION", status=ChargeStatus.PRESENT_WITHOUT_INFO, court_case=court_case, ) incarceration_sentence_1 = StateIncarcerationSentence.new_with_defaults( external_id="INCARCERATION_SENTENCE_ID1", state_code="US_ND", status=StateSentenceStatus.PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType.STATE_PRISON, charges=[charge_1], incarceration_periods=[ StateIncarcerationPeriod.new_with_defaults( external_id="I_PERIOD_ID", status=StateIncarcerationPeriodStatus.PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType.STATE_PRISON, state_code="US_ND", incarceration_incidents=[incident], program_assignments=[program_assignment], parole_decisions=[ StateParoleDecision.new_with_defaults( external_id="DECISION_ID", state_code="US_ND", decision_agents=[ StateAgent.new_with_defaults( external_id="AGENT_ID2", agent_type=StateAgentType. PRESENT_WITHOUT_INFO, state_code="US_ND", full_name= '{"full_name": "AGENT HERNANDEZ"}', ), StateAgent.new_with_defaults( external_id="AGENT_ID3", state_code="US_ND", agent_type=StateAgentType. PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT SMITH"}', ), ], ) ], assessments=[assessment], source_supervision_violation_response=response, ) ], ) incarceration_sentence_2 = StateIncarcerationSentence.new_with_defaults( external_id="INCARCERATION_SENTENCE_ID2", state_code="US_ND", status=StateSentenceStatus.PRESENT_WITHOUT_INFO, incarceration_type=StateIncarcerationType.STATE_PRISON, charges=[charge_2, charge_3], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id="S_PERIOD_ID3", status=StateSupervisionPeriodStatus.PRESENT_WITHOUT_INFO, state_code="US_ND", supervision_type=StateSupervisionType.PROBATION, supervision_type_raw_text="PROBATION", assessments=[assessment], supervising_officer=StateAgent.new_with_defaults( external_id="AGENT_ID_PO", state_code="US_ND", agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "AGENT PAROLEY"}', ), case_type_entries=[ StateSupervisionCaseTypeEntry.new_with_defaults( case_type=StateSupervisionCaseType. DOMESTIC_VIOLENCE, case_type_raw_text="DOMESTIC_VIOLENCE", state_code="US_ND", external_id="CASE_TYPE_ID", ) ], ) ], ) expected_result = [ StatePerson.new_with_defaults( state_code="US_ND", external_ids=[ StatePersonExternalId.new_with_defaults( external_id="EXTERNAL_ID1", state_code="US_ND", id_type=US_ND_ELITE, ), StatePersonExternalId.new_with_defaults( external_id="EXTERNAL_ID2", state_code="US_ND", id_type=US_ND_SID, ), ], races=[ StatePersonRace(race=Race.WHITE, race_raw_text="WHITE", state_code="US_ND"), StatePersonRace(race=Race.OTHER, race_raw_text="OTHER", state_code="US_ND"), ], ethnicities=[ StatePersonEthnicity( ethnicity=Ethnicity.HISPANIC, ethnicity_raw_text="HISPANIC", state_code="US_ND", ) ], aliases=[ StatePersonAlias.new_with_defaults( full_name='{"full_name": "LONNY BREAUX"}', state_code="US_ND"), StatePersonAlias.new_with_defaults( full_name='{"full_name": "FRANK OCEAN"}', state_code="US_ND"), ], supervising_officer=StateAgent.new_with_defaults( external_id="AGENT_ID_SUPERVISING", state_code="US_ND", agent_type=StateAgentType.PRESENT_WITHOUT_INFO, full_name='{"full_name": "SUPERVISING AGENT"}', ), assessments=[assessment], program_assignments=[program_assignment], sentence_groups=[ StateSentenceGroup.new_with_defaults( external_id="GROUP_ID1", status=StateSentenceStatus.PRESENT_WITHOUT_INFO, state_code="US_ND", supervision_sentences=[ StateSupervisionSentence.new_with_defaults( external_id="SUPERVISION_SENTENCE_ID1", state_code="US_ND", status=StateSentenceStatus. PRESENT_WITHOUT_INFO, charges=[charge_1, charge_2], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id="S_PERIOD_ID1", status=StateSupervisionPeriodStatus. PRESENT_WITHOUT_INFO, supervision_level=StateSupervisionLevel .MEDIUM, supervision_level_raw_text="MED", state_code="US_ND", supervision_type=StateSupervisionType. PAROLE, supervision_type_raw_text="PAROLE", supervision_violation_entries=[ violation ], program_assignments=[ program_assignment ], supervision_contacts=[ supervision_contact ], ) ], ) ], incarceration_sentences=[ incarceration_sentence_1, incarceration_sentence_2, ], ), StateSentenceGroup.new_with_defaults( external_id="GROUP_ID2", status=StateSentenceStatus.PRESENT_WITHOUT_INFO, state_code="US_ND", supervision_sentences=[ StateSupervisionSentence.new_with_defaults( external_id="SUPERVISION_SENTENCE_ID2", state_code="US_ND", status=StateSentenceStatus. PRESENT_WITHOUT_INFO, charges=[charge_2], supervision_periods=[ StateSupervisionPeriod.new_with_defaults( external_id="S_PERIOD_ID2", status=StateSupervisionPeriodStatus. PRESENT_WITHOUT_INFO, state_code="US_ND", supervision_type=StateSupervisionType. PAROLE, supervision_type_raw_text="PAROLE", ) ], ) ], fines=[ StateFine.new_with_defaults( external_id="FINE_ID", state_code="US_ND", status=StateFineStatus.PAID, status_raw_text="PAID", ) ], ), ], ) ] print("\n", expected_result, "\n\n\n", result) self.assertCountEqual(expected_result, result)