diff --git a/test_team_seven_eleven.py b/test_team_seven_eleven.py index dc3df90..d7abef6 100644 --- a/test_team_seven_eleven.py +++ b/test_team_seven_eleven.py @@ -11,7 +11,27 @@ def test_show_aggie_pride(self): def test_in_aadhaar(self): """Test IN_AADHAAR functionality""" + valid_compact = ["323456789012", "223456789018"] # pass Verhoeff + samples = [ + f"My Aadhaar number is {valid_compact[0]}", # compact + f"My Aadhaar number is {valid_compact[1]}", # compact + f"Aadhaar: {valid_compact[0][:4]} {valid_compact[0][4:8]} {valid_compact[0][8:]}", # spaced + f"Aadhaar ID {valid_compact[1][:4]}-{valid_compact[1][4:8]}-{valid_compact[1][8:]}",# hyphenated + ] + for text in samples: + result = analyze_text(text, ['IN_AADHAAR']) + self.assertGreater(len(result), 0, f"No match for: {text}") + self.assertEqual(result[0].entity_type, 'IN_AADHAAR') + + # Negative + neg = analyze_text("My Aadhaar is hidden", ['IN_AADHAAR']) + self.assertListEqual(neg, []) + + # Negative + result = analyze_text("My Aadhaar is hidden", ['IN_AADHAAR']) + self.assertListEqual(result, []) + def test_in_pan(self): """Test IN_PAN functionality""" beginning = ['ABCDE', 'FFFFF', 'ABABA', 'CDCDC']