added updated regex for AU passports and updated tests

benjaminlhai · Jul 31, 2024 · ec4fe27 · ec4fe27
1 parent ffb23f7
commit ec4fe27
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pyvalidator/is_passport_number.py b/pyvalidator/is_passport_number.py
@@ -5,7 +5,7 @@
     "AM": r"^[A-Z]{2}\d{7}$",  # ARMENIA
     "AR": r"^[A-Z]{3}\d{6}$",  # ARGENTINA
     "AT": r"^[A-Z]\d{7}$",  # AUSTRIA
-    "AU": r"^[A-Z]\d{7}$",  # AUSTRALIA
+    "AU": r"^([A-Z]{2}\d{7}|[A-Z]{1}\d{7})$",  # AUSTRALIA
     "BA": r"^[A-Z]\d{7}$",  # BOSNIA AND HERZEGOVINA,
     "BE": r"^[A-Z]{2}\d{6}$",  # BELGIUM
     "BG": r"^\d{9}$",  # BULGARIA

diff --git a/test/test_is_passport_number.py b/test/test_is_passport_number.py
@@ -16,6 +16,7 @@ def test_valid_passport_number(self):
             ['P 1630837', "AT"],
             ['P 4366918', "AT"],
             ['N0995852', "AU"],
+            ['PN0995852', "AU"],
             ['L4819236', "AU"],
             ['A1234567', "BA"],
             ['346395366', "BG"],
@@ -105,6 +106,7 @@ def test_invalid_passport_number(self):
             ['A11811035', "AR"],
             ['0 1630837', "AT"],
             ['1A012345', "AU"],
+            ['1AP012345', "AU"],
             ['ABC123456', "BA"],
             ['ABC123456', "BG"],
             ['ABX29332', "BR"],