Skip to content

Update au medicare test failures test_team_aggie_annihilators.py #83

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
jswcodes wants to merge 1 commit into
base: 17-detect-au_medicare
Choose a base branch
from
Open

Update au medicare test failures test_team_aggie_annihilators.py #83

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
96 changes: 63 additions & 33 deletions test_team_aggie_annihilators.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,62 +1,92 @@
"""Unit test file for team aggie_annihilators"""
import unittest
from pii_scan import analyze_text, show_aggie_pride # noqa
from presidio_analyzer import AnalyzerEngine
from pii_scan import analyze_text # assuming your project uses this helper
from au_medicare_recognizer import AuMedicareRecognizer # your recognizer class
from pii_scan import analyze_text, show_aggie_pride # noqa


class TestTeam_aggie_annihilators(unittest.TestCase):
"""Test team aggie_annihilators PII functions"""

def test_show_aggie_pride(self):
"""Test to make sure Aggie Pride is shown correctly"""
self.assertEqual(show_aggie_pride(), "Aggie Pride - Worldwide")

def test_aba_routing_number(self):
"""Test ABA_ROUTING_NUMBER functionality"""
# Positive: valid 9-digit ABA routing numbers (commonly used in docs)
positives = [
"Routing number 111000025",
"ABA: 021000021",
]
for text in positives:
with self.subTest(text=text):
res = analyze_text(text, ['ABA_ROUTING_NUMBER'])
self.assertGreaterEqual(len(res), 1, f"Should detect ABA_ROUTING_NUMBER in: {text}")
self.assertEqual(res[0].entity_type, 'ABA_ROUTING_NUMBER')

# Negative: wrong length / unrelated
negatives = [
"routing number 11100002", # 8 digits
"routing number 1110000250", # 10 digits
"credit card 4111 1111 1111 1111",
"Routing info not provided",
]
for text in negatives:
with self.subTest(text=text):
res = analyze_text(text, ['ABA_ROUTING_NUMBER'])
self.assertEqual(res, [], f"Should NOT detect ABA_ROUTING_NUMBER in: {text}")

def test_au_abn(self):
"""Test AU_ABN functionality"""
# Add specific ABN tests later if assigned
pass

def test_au_acn(self):
"""Test AU_ACN functionality"""
# Positive test case - basic format
acn = 'my australian company number is ACN 010 499 966'
result = analyze_text(acn, ['AU_ACN'])
self.assertEqual(len(result), 1)
self.assertEqual(result[0].entity_type, 'AU_ACN')

# Try without 'ACN' prefix
acn_no_prefix = 'my australian company number is 010 499 966'
result = analyze_text(acn_no_prefix, ['AU_ACN'])
self.assertEqual(len(result), 1)

# Negative test case
result = analyze_text('ACN is required', ['AU_ACN'])
self.assertEqual(result, [])

def test_au_medicare(self):
"""Test AU_MEDICARE functionality"""
# Initialize analyzer and add recognizer
analyzer = AnalyzerEngine()
analyzer.registry.add_recognizer(AuMedicareRecognizer())

# --- Positive (valid checksum) ---
valid_numbers = [
"My Medicare number is 2123 45670 1",
"Medicare: 3123 45670 2",
"Medicare card 5234 67890 3",
"Here’s my medicare number: 4123 45670 1",
"Medicare number 5123456703", # no spaces
]
for text in valid_numbers:
result = analyzer.analyze(text=text, entities=["AU_MEDICARE"], language="en")
self.assertGreater(len(result), 0, f"Expected AU_MEDICARE detected in: {text}")
self.assertEqual(result[0].entity_type, "AU_MEDICARE")

# --- Negative (invalid checksum or format) ---
invalid_numbers = [
"Medicare number 2123 45670 9", # wrong checksum
"My number is 7123 45670 1", # invalid prefix (7 not allowed)
"Medicare 12345678", # too short
"Medicare 2123-45670-1", # dashes (not supported in default patterns)
"My medicare info is hidden", # no number
# Use numbers that satisfy the AU Medicare checksum (10 digits total).
positives = [
"Medicare number: 1234 56788 1", # 1234567881
"AU Medicare: 2876543251", # compact
"my medicare is 2468 13576 1 please verify" # 2468135761
]
for text in invalid_numbers:
result = analyzer.analyze(text=text, entities=["AU_MEDICARE"], language="en")
self.assertEqual(result, [], f"Should NOT detect AU_MEDICARE in: {text}")


for text in positives:
with self.subTest(text=text):
result = analyze_text(text, ['AU_MEDICARE'])
self.assertGreaterEqual(len(result), 1, f"Should detect AU_MEDICARE in: {text}")
self.assertEqual(result[0].entity_type, 'AU_MEDICARE')

negatives = [
"Medicare 1234 5678 1", # too short
"Medicare 1234 56789 12", # too long
"Credit card: 4111 1111 1111 1111", # unrelated pattern
"Email: test@example.com", # unrelated text
"My number is 123456789" # wrong length/checks
]
for text in negatives:
with self.subTest(text=text):
result = analyze_text(text, ['AU_MEDICARE'])
self.assertEqual(result, [], f"Should NOT detect AU_MEDICARE in: {text}")

def test_au_tfn(self):
"""Test AU_TFN functionality"""
# Add specific TFN tests later if assigned
pass


if __name__ == '__main__':
Expand Down