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)
