mirror of
https://github.com/osm-search/Nominatim.git
synced 2024-11-29 08:36:24 +03:00
e828d0d3f7
The bad quotes around the type for special phrases specifically occure in the Wiki pages, so it should be removed by the loader and not in the generic SpecialPhrase object.
58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
# SPDX-License-Identifier: GPL-2.0-only
|
|
#
|
|
# This file is part of Nominatim. (https://nominatim.org)
|
|
#
|
|
# Copyright (C) 2022 by the Nominatim developer community.
|
|
# For a full list of authors see the git log.
|
|
"""
|
|
Tests for methods of the SPWikiLoader class.
|
|
"""
|
|
import pytest
|
|
from nominatim.tools.special_phrases.sp_wiki_loader import SPWikiLoader
|
|
|
|
|
|
@pytest.fixture
|
|
def sp_wiki_loader(src_dir, monkeypatch, def_config):
|
|
"""
|
|
Return an instance of SPWikiLoader.
|
|
"""
|
|
monkeypatch.setenv('NOMINATIM_LANGUAGES', 'en')
|
|
loader = SPWikiLoader(def_config)
|
|
|
|
def _mock_wiki_content(lang):
|
|
xml_test_content = src_dir / 'test' / 'testdata' / 'special_phrases_test_content.txt'
|
|
return xml_test_content.read_text()
|
|
|
|
monkeypatch.setattr('nominatim.tools.special_phrases.sp_wiki_loader._get_wiki_content',
|
|
_mock_wiki_content)
|
|
return loader
|
|
|
|
|
|
def test_generate_phrases(sp_wiki_loader):
|
|
"""
|
|
Test objects returned from the next() method.
|
|
It should return all SpecialPhrases objects of
|
|
the 'en' special phrases.
|
|
"""
|
|
phrases = list(sp_wiki_loader.generate_phrases())
|
|
|
|
assert set((p.p_label, p.p_class, p.p_type, p.p_operator) for p in phrases) ==\
|
|
{('Zip Line', 'aerialway', 'zip_line', '-'),
|
|
('Zip Lines', 'aerialway', 'zip_line', '-'),
|
|
('Zip Line in', 'aerialway', 'zip_line', 'in'),
|
|
('Zip Lines in', 'aerialway', 'zip_line', 'in'),
|
|
('Zip Line near', 'aerialway', 'zip_line', 'near'),
|
|
('Animal shelter', 'amenity', 'animal_shelter', '-'),
|
|
('Animal shelters', 'amenity', 'animal_shelter', '-'),
|
|
('Animal shelter in', 'amenity', 'animal_shelter', 'in'),
|
|
('Animal shelters in', 'amenity', 'animal_shelter', 'in'),
|
|
('Animal shelter near', 'amenity', 'animal_shelter', 'near'),
|
|
('Animal shelters near', 'amenity', 'animal_shelter', 'near'),
|
|
('Drinking Water near', 'amenity', 'drinking_water', 'near'),
|
|
('Water', 'amenity', 'drinking_water', '-'),
|
|
('Water in', 'amenity', 'drinking_water', 'in'),
|
|
('Water near', 'amenity', 'drinking_water', 'near'),
|
|
('Embassy', 'amenity', 'embassy', '-'),
|
|
('Embassys', 'amenity', 'embassy', '-'),
|
|
('Embassies', 'amenity', 'embassy', '-')}
|