Merge pull request #58 from argyle-engineering/floor-hyphen

Support hyphened floor numbers
argyle-engineering · Oct 9, 2024 · c159b0f · c159b0f
2 parents e81b9c3 + df6134b
commit c159b0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyap/source_US/data.py b/pyap/source_US/data.py
@@ -831,7 +831,7 @@ def street_type_list_to_regex(street_type_list: list[str]) -> str:
 floor = r"""
             (?P<floor>
                 (?:
-                    \d+[A-Za-z]{{0,2}}\.?\ {floor_indic}
+                    (?:\d+|\d-\d)[A-Za-z]{{0,2}}\.?\ {floor_indic}
                 )
                 |
                 (?:

diff --git a/tests/test_parser_us.py b/tests/test_parser_us.py
@@ -405,6 +405,7 @@ def test_po_box_positive(input, expected):
     "input,expected",
     [
         # positive assertions
+        ("14001 E. ILIFF AVE 5-7TH FLOOR", True),
         ("1111 WILSON BVD STE 2222", True),
         ("800 W EL CAMINO REAL\n350 STE *", True),
         ("899 HEATHROW PARK LN", True),