def test_most_recent_supervision_type_no_sentences_same_day_bound(self): supervision_period_supervision_type = \ us_id_get_most_recent_supervision_period_supervision_type_before_upper_bound_day( upper_bound_exclusive_date=self.upper_bound_date, lower_bound_inclusive_date=self.upper_bound_date, supervision_periods=[]) self.assertEqual(supervision_period_supervision_type, None)
def test_most_recent_supervision_type_supervision_starts_on_bound(self): preceding_supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=1, start_date=self.upper_bound_date, termination_date=self.upper_bound_date + relativedelta(years=1), supervision_period_supervision_type= StateSupervisionPeriodSupervisionType.PAROLE) supervision_period_supervision_type = \ us_id_get_most_recent_supervision_period_supervision_type_before_upper_bound_day( upper_bound_exclusive_date=self.upper_bound_date, lower_bound_inclusive_date=None, supervision_periods=[preceding_supervision_period]) self.assertEqual(supervision_period_supervision_type, None)
def test_most_recent_supervision_type_overlapping_supervision(self): preceding_supervision_period = StateSupervisionPeriod.new_with_defaults( supervision_period_id=1, start_date=self.upper_bound_date - relativedelta(days=100), termination_date=self.upper_bound_date + relativedelta(days=100), supervision_period_supervision_type= StateSupervisionPeriodSupervisionType.PROBATION) supervision_period_supervision_type = \ us_id_get_most_recent_supervision_period_supervision_type_before_upper_bound_day( upper_bound_exclusive_date=self.upper_bound_date, lower_bound_inclusive_date=None, supervision_periods=[preceding_supervision_period]) self.assertEqual(StateSupervisionPeriodSupervisionType.PROBATION, supervision_period_supervision_type)
def test_most_recent_supervision_type_ignore_before_lower_bound(self): preceding_supervision_period = StateSupervisionPeriod.new_with_defaults( state_code="US_ID", supervision_period_id=1, start_date=self.upper_bound_date - relativedelta(days=100), termination_date=self.upper_bound_date - relativedelta(days=10), supervision_period_supervision_type= StateSupervisionPeriodSupervisionType.PAROLE, status=StateSupervisionPeriodStatus.PRESENT_WITHOUT_INFO, ) supervision_period_supervision_type = us_id_get_most_recent_supervision_period_supervision_type_before_upper_bound_day( upper_bound_exclusive_date=self.upper_bound_date, lower_bound_inclusive_date=self.upper_bound_date - relativedelta(days=2), supervision_periods=[preceding_supervision_period], ) self.assertEqual(None, supervision_period_supervision_type)