def test_identifyViolationSubtype_technical_drg(self):
# Arrange
violations = [
StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
violation_type=StateSupervisionViolationType.TECHNICAL)
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
condition='DRG')
]),
StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
violation_type=StateSupervisionViolationType.FELONY)
])
]
# Act
subtype = identify_violation_subtype(
StateSupervisionViolationType.TECHNICAL, violations)
# Assert
self.assertEqual('SUBSTANCE_ABUSE', subtype)
def test_identifyViolationSubtype_technical_law_citation(self):
# Arrange
violations = [
StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
violation_type=StateSupervisionViolationType.TECHNICAL)
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
condition=_SUBSTANCE_ABUSE_CONDITION_STR),
StateSupervisionViolatedConditionEntry.new_with_defaults(
condition=_LAW_CITATION_SUBTYPE_STR)
]),
StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
violation_type=StateSupervisionViolationType.FELONY)
])
]
# Act
subtype = identify_violation_subtype(
StateSupervisionViolationType.TECHNICAL, violations)
# Assert
self.assertEqual(_LAW_CITATION_SUBTYPE_STR, subtype)
def test_identifyViolationSubtype_nonTechnical(self):
# Arrange
violations = [
StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
violation_type=StateSupervisionViolationType.TECHNICAL)
],
supervision_violated_conditions=[
StateSupervisionViolatedConditionEntry.new_with_defaults(
condition='DRG')
]),
StateSupervisionViolation.new_with_defaults(
state_code=_STATE_CODE,
supervision_violation_types=[
StateSupervisionViolationTypeEntry.new_with_defaults(
violation_type=StateSupervisionViolationType.FELONY)
])
]
# Act
subtype = identify_violation_subtype(
StateSupervisionViolationType.FELONY, violations)
self.assertIsNone(subtype)
