Skip to content

Commit 645b968

Browse files
committed
feat(fixers): stop names with double spaces before number, converted to comma+space before number
1 parent eecc5f7 commit 645b968

File tree

2 files changed

+10
-5
lines changed

2 files changed

+10
-5
lines changed

vigobus/datasources/fixers.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,11 @@ class Buses:
3232
@classmethod
3333
def stop_name(cls, original_name: str) -> str:
3434

35+
# Replace double spaces between a number, with a comma
36+
name = re.sub(r'\s{2}(\d)', r', \1', original_name)
37+
3538
# Remove double spaces
36-
name = re.sub(' +', ' ', original_name)
39+
name = re.sub(' +', ' ', name)
3740

3841
# Replace - with commas
3942
name = name.replace("-", ",")

vigobus/datasources/test_fixers.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -17,17 +17,19 @@ def test_fix_bus_line_route(line, route, expected_line, expected_route):
1717
@pytest.mark.parametrize("name, expected_name", [
1818
# TODO Update after implementing "Replace double spaces present before a number at the end, with a comma+space"
1919
# If expected_name=True, expect it to be equal to name
20-
("Rúa de Jenaro de la Fuente 29", "Rúa de Jenaro de la Fuente 29"),
21-
("Estrada de Moledo 6", "Estrada de Moledo 6"),
20+
("Rúa de Jenaro de la Fuente 29", "Rúa de Jenaro de la Fuente, 29"),
21+
("Estrada de Moledo 6", "Estrada de Moledo, 6"),
2222
("Estrada de Moledo, 6", True),
23-
("Rúa de Tomás Paredes 4", "Rúa de Tomás Paredes 4"),
23+
("Estrada de Moledo 6", "Estrada de Moledo, 6"),
24+
("Estrada de Moledo 6", True),
25+
("Rúa de Tomás Paredes 4", "Rúa de Tomás Paredes, 4"),
2426
("Rúa da Pedra Seixa (Colexio)", True),
2527
("Rúa de Pi i Margall (fronte 5)", "Rúa de Pi i Margall (Fronte 5)"),
2628
("Subida á Madroa (fronte Campo Fútbol)", "Subida á Madroa (Fronte Campo Fútbol)"),
2729
("Rúa de Manuel Álvarez (fronte cruce Camiño Sulevada)", "Rúa de Manuel Álvarez (Fronte Cruce Camiño Sulevada)"),
2830
("Estrada de Fragoselo (cruce Camiño Río da Barxa)", "Estrada de Fragoselo (Cruce Camiño Río da Barxa)"),
2931
("Rúa Castañal (cruce Camiño das Presas)", "Rúa Castañal (Cruce Camiño das Presas)"),
30-
("Rúa das Teixugueiras 19-Portal 5", "Rúa das Teixugueiras 19, Portal 5"),
32+
("Rúa das Teixugueiras 19-Portal 5", "Rúa das Teixugueiras, 19, Portal 5"),
3133
("Rúa de Santo Amaro (Praza de España)", True),
3234
("Rúa do Gaiteiro de Ricardo Portela (fronte Pavillón)", "Rúa do Gaiteiro de Ricardo Portela (Fronte Pavillón)"),
3335
("Telecomunicacións (CUVI)", "Telecomunicacións (Universidade)"),

0 commit comments

Comments
 (0)