def test_get_violation_type_frequency_counter_us_mo_technical_only_no_conditions(
self,
) -> None:
violations = [
StateSupervisionViolation.new_with_defaults(
state_code="US_MO",
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.TECHNICAL,
)
],
)
]
violation_type_frequency_counter = (
violation_utils.get_violation_type_frequency_counter(
violations, UsMoViolationDelegate()
)
)
self.assertEqual(
[[StateSupervisionViolationType.TECHNICAL.value]],
violation_type_frequency_counter,
)
def test_get_violation_type_frequency_counter_us_mo_technical_only(self) -> None:
violations = [
StateSupervisionViolation.new_with_defaults(
state_code="US_MO",
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.TECHNICAL,
)
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_XX", condition="DRG"
)
],
)
]
violation_type_frequency_counter = (
violation_utils.get_violation_type_frequency_counter(
violations, UsMoViolationDelegate()
)
)
self.assertEqual([["SUBSTANCE_ABUSE"]], violation_type_frequency_counter)
def test_get_violation_type_frequency_counter_us_mo(self) -> None:
violations = [
StateSupervisionViolation.new_with_defaults(
state_code="US_MO",
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.ABSCONDED,
),
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.FELONY,
),
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_XX", condition="DRG"
)
],
)
]
violation_type_frequency_counter = (
violation_utils.get_violation_type_frequency_counter(
violations, UsMoViolationDelegate()
)
)
self.assertEqual(
[["ABSCONDED", "FELONY", "SUBSTANCE_ABUSE"]],
violation_type_frequency_counter,
)
def test_get_violation_and_response_history_us_mo_handle_law_technicals(self):
"""Tests that a US_MO violation report with a TECHNICAL type and a LAW condition is not treated like a
citation with a LAW condition."""
supervision_violation = StateSupervisionViolation.new_with_defaults(
supervision_violation_id=123455,
state_code="US_MO",
violation_date=datetime.date(2009, 1, 7),
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_MO",
violation_type=StateSupervisionViolationType.TECHNICAL,
),
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_MO", condition="LAW"
),
],
)
supervision_violation_response = StateSupervisionViolationResponse.new_with_defaults(
state_code="US_MO",
supervision_violation_response_id=_DEFAULT_SSVR_ID,
response_type=StateSupervisionViolationResponseType.VIOLATION_REPORT,
response_subtype="ITR",
response_date=datetime.date(2009, 1, 7),
is_draft=False,
supervision_violation_response_decisions=[
StateSupervisionViolationResponseDecisionEntry.new_with_defaults(
state_code="US_MO",
decision=StateSupervisionViolationResponseDecision.REVOCATION,
revocation_type=StateSupervisionViolationResponseRevocationType.REINCARCERATION,
),
StateSupervisionViolationResponseDecisionEntry.new_with_defaults(
state_code="US_MO",
decision=StateSupervisionViolationResponseDecision.CONTINUANCE,
revocation_type=StateSupervisionViolationResponseRevocationType.SHOCK_INCARCERATION,
),
],
supervision_violation=supervision_violation,
)
revocation_date = datetime.date(2009, 2, 13)
violation_history = violation_utils.get_violation_and_response_history(
revocation_date, [supervision_violation_response], UsMoViolationDelegate()
)
expected_output = violation_utils.ViolationHistory(
most_severe_violation_type=StateSupervisionViolationType.TECHNICAL,
most_severe_violation_type_subtype=StateSupervisionViolationType.TECHNICAL.value,
most_severe_response_decision=StateSupervisionViolationResponseDecision.REVOCATION,
response_count=1,
violation_history_description="1tech",
violation_type_frequency_counter=[["LAW", "TECHNICAL"]],
)
self.assertEqual(expected_output, violation_history)
def test_get_violation_and_response_history_with_us_mo_subtype(self):
supervision_violation = StateSupervisionViolation.new_with_defaults(
supervision_violation_id=123455,
state_code="US_MO",
violation_date=datetime.date(2009, 1, 3),
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_MO",
violation_type=StateSupervisionViolationType.TECHNICAL,
),
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_MO", condition="DRG"
),
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_MO", condition="OTHER"
),
],
)
supervision_violation_response = StateSupervisionViolationResponse.new_with_defaults(
supervision_violation_response_id=_DEFAULT_SSVR_ID,
response_type=StateSupervisionViolationResponseType.VIOLATION_REPORT,
response_subtype="INI",
state_code="US_MO",
response_date=datetime.date(2009, 1, 7),
supervision_violation_response_decisions=[
StateSupervisionViolationResponseDecisionEntry.new_with_defaults(
state_code="US_MO",
decision=StateSupervisionViolationResponseDecision.REVOCATION,
revocation_type=StateSupervisionViolationResponseRevocationType.REINCARCERATION,
),
StateSupervisionViolationResponseDecisionEntry.new_with_defaults(
state_code="US_MO",
decision=StateSupervisionViolationResponseDecision.CONTINUANCE,
),
],
supervision_violation=supervision_violation,
)
revocation_date = datetime.date(2009, 2, 13)
violation_history = violation_utils.get_violation_and_response_history(
revocation_date, [supervision_violation_response], UsMoViolationDelegate()
)
expected_output = violation_utils.ViolationHistory(
most_severe_violation_type=StateSupervisionViolationType.TECHNICAL,
most_severe_violation_type_subtype="SUBSTANCE_ABUSE",
most_severe_response_decision=StateSupervisionViolationResponseDecision.REVOCATION,
response_count=1,
violation_history_description="1subs",
violation_type_frequency_counter=[["SUBSTANCE_ABUSE", "OTHER"]],
)
self.assertEqual(expected_output, violation_history)
def test_get_violation_type_frequency_counter_us_mo_multiple_violations(
self,
) -> None:
violations = [
StateSupervisionViolation.new_with_defaults(
state_code="US_MO",
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.ABSCONDED,
),
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.FELONY,
),
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_XX", condition="WEA"
)
],
),
StateSupervisionViolation.new_with_defaults(
state_code="US_MO",
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.MISDEMEANOR,
),
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code="US_XX",
violation_type=StateSupervisionViolationType.TECHNICAL,
),
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_XX", condition="DRG"
),
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code="US_XX", condition="EMP"
),
],
),
]
violation_type_frequency_counter = (
violation_utils.get_violation_type_frequency_counter(
violations, UsMoViolationDelegate()
)
)
self.assertEqual(
[["ABSCONDED", "FELONY", "WEA"], ["MISDEMEANOR", "SUBSTANCE_ABUSE", "EMP"]],
violation_type_frequency_counter,
)
def get_state_specific_violation_delegate(
state_code: str, ) -> StateSpecificViolationDelegate:
"""Returns the type of StateSpecificViolationDelegate that should be used for
violation calculations in a given |state_code|."""
if state_code == StateCode.US_ID.value:
return UsIdViolationDelegate()
if state_code == StateCode.US_ND.value:
return UsNdViolationDelegate()
if state_code == StateCode.US_MO.value:
return UsMoViolationDelegate()
if state_code == StateCode.US_PA.value:
return UsPaViolationDelegate()
raise ValueError(f"Unexpected state code [{state_code}]")
def setUp(self) -> None:
self.delegate = UsMoViolationDelegate()
class TestUsMoGetViolationTypeSubstringsForViolation(unittest.TestCase):
"""Tests the get_violation_type_subtype_strings_for_violation function when the UsMoViolationDelegate is provided."""
def setUp(self) -> None:
self.delegate = UsMoViolationDelegate()
def test_us_mo_get_violation_type_subtype_strings_for_violation(
self) -> None:
# Arrange
violation = StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code=_STATE_CODE,
violation_type=StateSupervisionViolationType.FELONY,
)
],
)
# Act
type_subtype_strings = (
self.delegate.get_violation_type_subtype_strings_for_violation(
violation))
# Assert
expected_type_subtype_strings = ["FELONY"]
self.assertEqual(expected_type_subtype_strings, type_subtype_strings)
def test_us_mo_get_violation_type_subtype_strings_for_violation_substance(
self, ) -> None:
# Arrange
violation = StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code=_STATE_CODE,
violation_type=StateSupervisionViolationType.TECHNICAL,
)
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code=_STATE_CODE, condition="DRG")
],
)
# Act
type_subtype_strings = (
self.delegate.get_violation_type_subtype_strings_for_violation(
violation))
# Assert
expected_type_subtype_strings = ["SUBSTANCE_ABUSE"]
self.assertEqual(expected_type_subtype_strings, type_subtype_strings)
def test_us_mo_get_violation_type_subtype_strings_for_violation_law_citation(
self, ) -> None:
# Arrange
violation = StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code=_STATE_CODE,
violation_type=StateSupervisionViolationType.TECHNICAL,
)
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code=_STATE_CODE, condition="LAW_CITATION")
],
)
# Act
type_subtype_strings = (
self.delegate.get_violation_type_subtype_strings_for_violation(
violation))
# Assert
expected_type_subtype_strings = ["LAW_CITATION"]
self.assertEqual(expected_type_subtype_strings, type_subtype_strings)
def test_us_mo_get_violation_type_subtype_strings_for_violation_technical(
self, ) -> None:
# Arrange
violation = StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
state_code=_STATE_CODE,
violation_type=StateSupervisionViolationType.TECHNICAL,
)
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
state_code=_STATE_CODE, condition="EMP")
],
)
# Act
type_subtype_strings = (
self.delegate.get_violation_type_subtype_strings_for_violation(
violation))
# Assert
expected_type_subtype_strings = ["EMP", "TECHNICAL"]
self.assertEqual(expected_type_subtype_strings, type_subtype_strings)
def test_us_mo_get_violation_type_subtype_strings_for_violation_no_types(
self, ) -> None:
# Arrange
violation = StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE)
# Act
type_subtype_strings = (
self.delegate.get_violation_type_subtype_strings_for_violation(
violation))
# Assert
expected_type_subtype_strings: List[str] = []
self.assertEqual(expected_type_subtype_strings, type_subtype_strings)
def test_violation_type_subtypes_with_violation_type_mappings(
self) -> None:
supported_violation_subtypes = (
violation_type_subtypes_with_violation_type_mappings(
self.delegate))
expected_violation_subtypes = {
"FELONY",
"MISDEMEANOR",
"LAW_CITATION",
"ABSCONDED",
"MUNICIPAL",
"ESCAPED",
"SUBSTANCE_ABUSE",
"TECHNICAL",
}
self.assertEqual(supported_violation_subtypes,
expected_violation_subtypes)