diff --git a/test_team_futureassured.py b/test_team_futureassured.py index 3b48edb..19a1170 100644 --- a/test_team_futureassured.py +++ b/test_team_futureassured.py @@ -2,7 +2,6 @@ import unittest from pii_scan import anonymize_text - class TestTeamFutureAssured(unittest.TestCase): """Test the Team TeamFutureAssured PII functions""" def test_anonymize_text(self): @@ -78,8 +77,23 @@ def test_it_driver_license(self): expected = 'My driver license code is R1053121X' actual = anonymize_text(test_string, ['IT_DRIVER_LICENSE']) self.assertEqual(expected, actual) + def test_au_abn(self): - """Test au_abn functionality""" + """Test case for AU_ABN""" + #positive testcase + test_string = "My ABN is 51824753556" + expected = "My ABN is " + actual = anonymize_text(test_string, ["AU_ABN"]) + self.assertEqual(expected,actual) + + #negative testcase + test_string = "My ABN is 1234334345678" + expected = "My ABN is " + actual = anonymize_text(test_string, ["AU_ABN"]) + self.assertNotEqual(expected,actual) + + + def test_au_medicare(self): """Test au_medicare functionality"""