def test_vulnerability_class_success(score, class_): assert vulnerability_class(score) == class_
def test_vulnerability_class_bad(score): assert vulnerability_class(score) is None
def test_vulnerability_class_success(score_and_class): assert vulnerability_class(score_and_class[0]) == score_and_class[1]