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)