custom comparison for SpecialPhrase

Duplicate elemination only works when a custom hash/equal function
is implemented that is based on the members.
This commit is contained in:
Sarah Hoffmann 2022-05-30 16:30:41 +02:00
parent e828d0d3f7
commit 46689df668
4 changed files with 18 additions and 15 deletions

View File

@ -10,7 +10,7 @@
This class is a model used to transfer a special phrase through
the process of load and importation.
"""
class SpecialPhrase():
class SpecialPhrase:
"""
Model representing a special phrase.
"""
@ -22,3 +22,15 @@ class SpecialPhrase():
# Needed if some operator in the wiki are not written in english
p_operator = p_operator.strip().lower()
self.p_operator = '-' if p_operator not in ('near', 'in') else p_operator
def __eq__(self, other):
if not isinstance(other, SpecialPhrase):
return False
return self.p_label == other.p_label \
and self.p_class == other.p_class \
and self.p_type == other.p_type \
and self.p_operator == other.p_operator
def __hash__(self):
return hash((self.p_label, self.p_class, self.p_type, self.p_operator))

View File

@ -17,11 +17,6 @@ from nominatim.errors import UsageError
from cursor import CursorForTesting
@pytest.fixture
def testfile_dir(src_dir):
return src_dir / 'test' / 'testfiles'
@pytest.fixture
def sp_importer(temp_db_conn, def_config, monkeypatch):
"""

View File

@ -11,6 +11,7 @@ import pytest
from nominatim.errors import UsageError
from nominatim.tools.special_phrases.sp_csv_loader import SPCsvLoader
from nominatim.tools.special_phrases.special_phrase import SpecialPhrase
@pytest.fixture
def sp_csv_loader(src_dir):
@ -29,17 +30,11 @@ def test_generate_phrases(sp_csv_loader):
"""
phrases = list(sp_csv_loader.generate_phrases())
assert len(phrases) == 41
assert len(phrases) == 42
assert len(set(phrases)) == 41
assert any(p.p_label == 'Billboard'
and p.p_class == 'advertising'
and p.p_type == 'billboard'
and p.p_operator == '-' for p in phrases)
assert any(p.p_label == 'Zip Lines'
and p.p_class == 'aerialway'
and p.p_type == 'zip_line'
and p.p_operator == '-' for p in phrases)
assert SpecialPhrase('Billboard', 'advertising', 'billboard', '-') in phrases
assert SpecialPhrase('Zip Lines', 'aerialway', 'zip_line', '-') in phrases
def test_invalid_cvs_file():

View File

@ -18,6 +18,7 @@ Zipline near,aerialway,zip_line,near,N
Ziplines near,aerialway,zip_line,near,Y
Zipwire,aerialway,zip_line,-,N
Zipwires,aerialway,zip_line,-,Y
Zipwires,aerialway,zip_line,name,Y
Zipwire in,aerialway,zip_line,in,N
Zipwires in,aerialway,zip_line,in,Y
Zipwire near,aerialway,zip_line,near,N

1 phrase class type operator plural
18 Ziplines near aerialway zip_line near Y
19 Zipwire aerialway zip_line - N
20 Zipwires aerialway zip_line - Y
21 Zipwires aerialway zip_line name Y
22 Zipwire in aerialway zip_line in N
23 Zipwires in aerialway zip_line in Y
24 Zipwire near aerialway zip_line near N