def test_ip_is_nested_in_previous_period_bad_sort(self):
# This period should not have been sorted before ip
previous_ip = StateIncarcerationPeriod.new_with_defaults(
external_id="1",
incarceration_period_id=1111,
status=StateIncarcerationPeriodStatus.NOT_IN_CUSTODY,
state_code="US_XX",
admission_date=date(2002, 2, 5),
admission_reason=StateIncarcerationPeriodAdmissionReason.NEW_ADMISSION,
release_date=date(2002, 9, 11),
release_reason=StateIncarcerationPeriodReleaseReason.SENTENCE_SERVED,
)
ip = StateIncarcerationPeriod.new_with_defaults(
external_id="2",
incarceration_period_id=2222,
status=StateIncarcerationPeriodStatus.NOT_IN_CUSTODY,
state_code="US_XX",
admission_date=date(2002, 1, 1),
admission_reason=StateIncarcerationPeriodAdmissionReason.TRANSFER,
release_date=date(2002, 5, 22),
release_reason=StateIncarcerationPeriodReleaseReason.TRANSFER,
)
with pytest.raises(ValueError):
ip_is_nested_in_previous_period(ip, previous_ip)
def test_ip_is_nested_in_previous_period_share_release(self):
previous_ip = StateIncarcerationPeriod.new_with_defaults(
external_id="1",
incarceration_period_id=1111,
status=StateIncarcerationPeriodStatus.NOT_IN_CUSTODY,
state_code="US_XX",
admission_date=date(2002, 2, 5),
admission_reason=StateIncarcerationPeriodAdmissionReason.NEW_ADMISSION,
release_date=date(2002, 9, 11),
release_reason=StateIncarcerationPeriodReleaseReason.SENTENCE_SERVED,
)
ip = StateIncarcerationPeriod.new_with_defaults(
external_id="2",
incarceration_period_id=2222,
status=StateIncarcerationPeriodStatus.NOT_IN_CUSTODY,
state_code="US_XX",
admission_date=date(2002, 3, 13),
admission_reason=StateIncarcerationPeriodAdmissionReason.TRANSFER,
release_date=date(2002, 9, 11),
release_reason=StateIncarcerationPeriodReleaseReason.TRANSFER,
)
self.assertTrue(ip_is_nested_in_previous_period(ip, previous_ip))
def test_ip_is_nested_in_previous_period_zero_day_period_on_release(self):
previous_ip = StateIncarcerationPeriod.new_with_defaults(
external_id="1",
incarceration_period_id=1111,
status=StateIncarcerationPeriodStatus.NOT_IN_CUSTODY,
state_code="US_XX",
admission_date=date(2002, 1, 13),
admission_reason=StateIncarcerationPeriodAdmissionReason.TRANSFER,
release_date=date(2002, 3, 13),
release_reason=StateIncarcerationPeriodReleaseReason.TRANSFER,
)
ip = StateIncarcerationPeriod.new_with_defaults(
external_id="2",
incarceration_period_id=2222,
status=StateIncarcerationPeriodStatus.NOT_IN_CUSTODY,
state_code="US_XX",
admission_date=date(2002, 3, 13),
admission_reason=StateIncarcerationPeriodAdmissionReason.TRANSFER,
release_date=date(2002, 3, 13),
release_reason=StateIncarcerationPeriodReleaseReason.TRANSFER,
)
self.assertFalse(ip_is_nested_in_previous_period(ip, previous_ip))