1 # SPDX-License-Identifier: GPL-2.0-only
3 # This file is part of Nominatim. (https://nominatim.org)
5 # Copyright (C) 2022 by the Nominatim developer community.
6 # For a full list of authors see the git log.
8 Tests for methods of the SPWikiLoader class.
11 from nominatim.tools.special_phrases.sp_wiki_loader import SPWikiLoader
14 def xml_wiki_content(src_dir):
16 return the content of the static xml test file.
18 xml_test_content = src_dir / 'test' / 'testdata' / 'special_phrases_test_content.txt'
19 return xml_test_content.read_text()
23 def sp_wiki_loader(monkeypatch, def_config, xml_wiki_content):
25 Return an instance of SPWikiLoader.
27 monkeypatch.setenv('NOMINATIM_LANGUAGES', 'en')
28 loader = SPWikiLoader(def_config)
29 monkeypatch.setattr('nominatim.tools.special_phrases.sp_wiki_loader._get_wiki_content',
30 lambda lang: xml_wiki_content)
34 def test_generate_phrases(sp_wiki_loader):
36 Test objects returned from the next() method.
37 It should return all SpecialPhrases objects of
38 the 'en' special phrases.
40 phrases = list(sp_wiki_loader.generate_phrases())
41 check_phrases_content(phrases)
43 def check_phrases_content(phrases):
45 Asserts that the given phrases list contains
46 the right phrases of the 'en' special phrases.
48 assert set((p.p_label, p.p_class, p.p_type, p.p_operator) for p in phrases) ==\
49 {('Zip Line', 'aerialway', 'zip_line', '-'),
50 ('Zip Lines', 'aerialway', 'zip_line', '-'),
51 ('Zip Line in', 'aerialway', 'zip_line', 'in'),
52 ('Zip Lines in', 'aerialway', 'zip_line', 'in'),
53 ('Zip Line near', 'aerialway', 'zip_line', 'near'),
54 ('Animal shelter', 'amenity', 'animal_shelter', '-'),
55 ('Animal shelters', 'amenity', 'animal_shelter', '-'),
56 ('Animal shelter in', 'amenity', 'animal_shelter', 'in'),
57 ('Animal shelters in', 'amenity', 'animal_shelter', 'in'),
58 ('Animal shelter near', 'amenity', 'animal_shelter', 'near'),
59 ('Animal shelters near', 'amenity', 'animal_shelter', 'near'),
60 ('Drinking Water near', 'amenity', 'drinking_water', 'near'),
61 ('Water', 'amenity', 'drinking_water', '-'),
62 ('Water in', 'amenity', 'drinking_water', 'in'),
63 ('Water near', 'amenity', 'drinking_water', 'near'),
64 ('Embassy', 'amenity', 'embassy', '-'),
65 ('Embassys', 'amenity', 'embassy', '-'),
66 ('Embassies', 'amenity', 'embassy', '-')}