From 23641470fbb80a1d6fdcd3f2ca713adc0c5dc3d7 Mon Sep 17 00:00:00 2001 From: TaLi-hub03 <114318763+TaLi-hub03@users.noreply.github.com> Date: Tue, 21 Oct 2025 18:41:53 +0000 Subject: [PATCH] Added AU_ABN test and detection for Sprint 2 --- test_team_aggie_annihilators.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/test_team_aggie_annihilators.py b/test_team_aggie_annihilators.py index 672d5bc..835760a 100644 --- a/test_team_aggie_annihilators.py +++ b/test_team_aggie_annihilators.py @@ -14,6 +14,21 @@ def test_aba_routing_number(self): def test_au_abn(self): """Test AU_ABN functionality""" + # Positive test case - valid ABN format + abn_text = "The company's ABN is 51824753556" + result = analyze_text(abn_text, ['AU_ABN']) + self.assertEqual(len(result), 1, "Expected one AU_ABN detection") + self.assertEqual(result[0].entity_type, 'AU_ABN') + + #Another pocitive case with ABN prefix + abn_text_prefix = "ABN 51824753556 belongs to Example Pty Ltd" + result = analyze_text(abn_text_prefix, ['AU_ABN']) + self.assertEqual(len(result),1, "Expected one AU_ABN detection with prefix") + + # negative test case no ABN present + no_abn_text = "The company registration was successful" + result = analyze_text(no_abn_text, ['AU_ABN']) + self.assertEqual(result, [], "Expected no AU_ABN detection in non-ABN text") def test_au_acn(self): """Test AU_ACN functionality"""