diff --git a/test_team_1.py b/test_team_1.py index 613313b..de203e5 100644 --- a/test_team_1.py +++ b/test_team_1.py @@ -11,7 +11,24 @@ def test_show_aggie_pride(self): def test_it_driver_license(self): """Test IT_DRIVER_LICENSE functionality""" - + prefixes = ["AB", "CD", "EF", "U1"] + numbers = ["1234567", "7654321"] + suffixes = ["A", "B"] + + # positive test cases + for prefix in prefixes: + for num in numbers: + for suffix in suffixes: + candidate = f"{prefix}{num}{suffix}" + with self.subTest(candidate=candidate): + result = analyze_text(candidate, ["IT_DRIVER_LICENSE"]) + if result: + self.assertEqual(result[0].entity_type, "IT_DRIVER_LICENSE") + + # negative test case + result = analyze_text("my driver license number is hidden", ["IT_DRIVER_LICENSE"]) + self.assertEqual(result, []) + def test_it_fiscal_code(self): """Test IT_FISCAL_CODE functionality"""