def test_cwe_should_be_ok(self): cwe_str = "CWE-399" self.assertTrue(CWEID.correct_cwe_str(cwe_str)) cwe = CWEID(cwe_str) self.assertEqual(str(cwe), cwe_str)
def test_cwe_should_be_bad_4(self): cwe_str = "CWE-333A" self.assertFalse(CWEID.correct_cwe_str(cwe_str)) self.assertRaises(ValueError, CWEID, cwe_str)