mirror of
https://github.com/osm-search/Nominatim.git
synced 2024-11-26 13:27:52 +03:00
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
# This file is part of Nominatim. (https://nominatim.org)
|
|
#
|
|
# Copyright (C) 2024 by the Nominatim developer community.
|
|
# For a full list of authors see the git log.
|
|
"""
|
|
Tests for methods of the SPCsvLoader class.
|
|
"""
|
|
import pytest
|
|
|
|
from nominatim_db.errors import UsageError
|
|
from nominatim_db.tools.special_phrases.sp_csv_loader import SPCsvLoader
|
|
from nominatim_db.tools.special_phrases.special_phrase import SpecialPhrase
|
|
|
|
@pytest.fixture
|
|
def sp_csv_loader(src_dir):
|
|
"""
|
|
Return an instance of SPCsvLoader.
|
|
"""
|
|
csv_path = (src_dir / 'test' / 'testdata' / 'sp_csv_test.csv').resolve()
|
|
loader = SPCsvLoader(csv_path)
|
|
return loader
|
|
|
|
|
|
def test_generate_phrases(sp_csv_loader):
|
|
"""
|
|
Test method parse_csv()
|
|
Should return the right SpecialPhrase objects.
|
|
"""
|
|
phrases = list(sp_csv_loader.generate_phrases())
|
|
|
|
assert len(phrases) == 42
|
|
assert len(set(phrases)) == 41
|
|
|
|
assert SpecialPhrase('Billboard', 'advertising', 'billboard', '-') in phrases
|
|
assert SpecialPhrase('Zip Lines', 'aerialway', 'zip_line', '-') in phrases
|
|
|
|
|
|
def test_invalid_cvs_file():
|
|
"""
|
|
Test method check_csv_validity()
|
|
It should raise an exception when file with a
|
|
different exception than .csv is given.
|
|
"""
|
|
loader = SPCsvLoader('test.wrong')
|
|
|
|
with pytest.raises(UsageError, match='not a csv file'):
|
|
next(loader.generate_phrases())
|