def test_find_supervision_periods_overlapping_with_date_no_termination(self):
referral_date = date(2013, 3, 1)
supervision_period = StateSupervisionPeriod.new_with_defaults(
supervision_period_id=111,
status=StateSupervisionPeriodStatus.TERMINATED,
state_code='US_XX',
start_date=date(2002, 11, 5),
supervision_type=StateSupervisionType.PAROLE
)
supervision_periods = [supervision_period]
supervision_periods_during_referral = \
identifier.find_supervision_periods_overlapping_with_date(
referral_date, supervision_periods
)
self.assertListEqual(supervision_periods,
supervision_periods_during_referral)
def test_find_supervision_periods_overlapping_with_date_no_overlap(self):
referral_date = date(2019, 3, 1)
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(2015, 5, 19),
termination_reason=StateSupervisionPeriodTerminationReason.DISCHARGE,
supervision_type=StateSupervisionType.PAROLE
)
supervision_periods = [supervision_period]
supervision_periods_during_referral = \
identifier.find_supervision_periods_overlapping_with_date(
referral_date, supervision_periods
)
self.assertListEqual([], supervision_periods_during_referral)