Fix bay occupancy

argyle-engineering · Oct 9, 2024 · 3c4c03f · 3c4c03f
1 parent c159b0f
commit 3c4c03f
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 1 deletion.
diff --git a/pyap/source_US/data.py b/pyap/source_US/data.py
@@ -230,7 +230,7 @@ def str_list_to_upper_lower_regex(str_list: List[str]) -> str:
     "Avenue",
     "Avn",
     "Avnue",
-    "Bay",
+    "Bay(?!\ [1-9])",
     "Bayoo",
     "Bayou",
     "Bch",

diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -207,6 +207,18 @@ def test_combine_results():
                 "postal_code": "77030-3411",
             },
         ),
+        (
+            "2817 PETERS ROAD BAY 52, Amazeville, AL 12345",
+            {
+                "street_number": "2817",
+                "street_type": "ROAD",
+                "street_name": "PETERS",
+                "occupancy": "BAY 52",
+                "city": "Amazeville",
+                "region1": "AL",
+                "postal_code": "12345",
+            },
+        ),
     ],
 )
 def test_parse_address(input: str, expected):

diff --git a/tests/test_parser_us.py b/tests/test_parser_us.py
@@ -482,6 +482,7 @@ def test_po_box_positive(input, expected):
         ("P.O. BOX 41256, One Velvet Drive", True),
         ("666 Hell ST PMB 29700", True),
         ("817 S.E. 55TH", True),
+        ("2817 PETERS ROAD BAY 52", True),
         # negative assertions
         (", 666 Hell ST PMB 29700", False),
     ],