def test_getMostRelevantSupervisionType_chooseDualIfExists(self) -> None: for supervision_type in StateSupervisionPeriodSupervisionType: types = { StateSupervisionPeriodSupervisionType.DUAL, supervision_type } self.assertEqual( StateSupervisionPeriodSupervisionType.DUAL, get_most_relevant_supervision_type(types), )
def test_getMostRelevantSupervisionType_dualIfProbationAndParole( self) -> None: types = { StateSupervisionPeriodSupervisionType.PROBATION, StateSupervisionPeriodSupervisionType.PAROLE, } self.assertEqual( StateSupervisionPeriodSupervisionType.DUAL, get_most_relevant_supervision_type(types), )
def us_id_get_most_recent_supervision_period_supervision_type_before_upper_bound_day( upper_bound_exclusive_date: date, lower_bound_inclusive_date: Optional[date], supervision_periods: List[StateSupervisionPeriod], ) -> Optional[StateSupervisionPeriodSupervisionType]: """Finds the most recent nonnull supervision period supervision type on the supervision periods, preceding or overlapping the provided date. An optional lower bound may be provided to limit the lookback window. Returns the most recent StateSupervisionPeriodSupervisionType. If there is no valid supervision type found (e.g. the person has only been incarcerated for the time window), returns None. In the case where multiple SupervisionPeriodSupervisionTypes end on the same day, this returns only the most relevant SupervisionPeriodSupervisionType based on our own ranking. """ supervision_types_by_end_date: Dict[ date, Set[StateSupervisionPeriodSupervisionType]] = defaultdict(set) lower_bound_exclusive_date = (lower_bound_inclusive_date - relativedelta(days=1) if lower_bound_inclusive_date else date.min) for supervision_period in supervision_periods: start_date = supervision_period.start_date if not start_date: continue termination_date = (supervision_period.termination_date if supervision_period.termination_date else date.today()) supervision_period_supervision_type = ( supervision_period.supervision_period_supervision_type) if not supervision_period_supervision_type: continue if not date_spans_overlap_exclusive( start_1=lower_bound_exclusive_date, end_1=upper_bound_exclusive_date, start_2=start_date, end_2=termination_date, ): continue supervision_types_by_end_date[termination_date].add( supervision_period_supervision_type) if not supervision_types_by_end_date: return None max_end_date = max(supervision_types_by_end_date.keys()) return get_most_relevant_supervision_type( supervision_types_by_end_date[max_end_date])
def supervision_period_supervision_type_mapper( supervision_types_str: Optional[str], ) -> Optional[StateSupervisionPeriodSupervisionType]: """Maps a list of supervision type codes to a supervision type. If both probation and parole types are present, this will return StateSupervisionPeriodSupervisionType.DUAL.""" if not supervision_types_str: return None supervision_type_strs = supervision_types_str.split(" ") supervision_types = set() for supervision_type_str in supervision_type_strs: supervision_type = STR_TO_SUPERVISION_PERIOD_SUPERVISION_TYPE_MAPPINGS.get( supervision_type_str, None) if not supervision_type: raise ValueError( f"No mapping for supervision period supervision type {supervision_type}" ) supervision_types.add(supervision_type) return get_most_relevant_supervision_type(supervision_types)
def test_getMostRelevantSupervisionType_allEnums(self) -> None: for supervision_type in StateSupervisionPeriodSupervisionType: types = {supervision_type} self.assertEqual(supervision_type, get_most_relevant_supervision_type(types))