X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/7cb7cf848d69e818cb3cccd656c4a34713316264..67462e095319038fa53636decb69dcf40f0c6dee:/test/python/dummy_tokenizer.py diff --git a/test/python/dummy_tokenizer.py b/test/python/dummy_tokenizer.py index 0868b57d..4739e3f0 100644 --- a/test/python/dummy_tokenizer.py +++ b/test/python/dummy_tokenizer.py @@ -1,6 +1,14 @@ +# 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. """ Tokenizer for testing. """ +from nominatim_db.data.place_info import PlaceInfo +from nominatim_db.config import Configuration def create(dsn, data_dir): """ Create a new instance of the tokenizer provided by this module. @@ -16,16 +24,22 @@ class DummyTokenizer: self.analyser_cache = {} - def init_new_db(self, config): - assert self.init_state == None + def init_new_db(self, *args, **kwargs): + assert self.init_state is None self.init_state = "new" - def init_from_project(self): - assert self.init_state == None + def init_from_project(self, config): + assert isinstance(config, Configuration) + assert self.init_state is None self.init_state = "loaded" + @staticmethod + def finalize_import(_): + pass + + def name_analyzer(self): return DummyNameAnalyzer(self.analyser_cache) @@ -47,14 +61,21 @@ class DummyNameAnalyzer: def close(self): pass - def add_postcodes_from_db(self): + @staticmethod + def normalize_postcode(postcode): + return postcode + + @staticmethod + def update_postcodes_from_db(): pass - def update_special_phrases(self, phrases): + def update_special_phrases(self, phrases, should_replace): self.analyser_cache['special_phrases'] = phrases def add_country_names(self, code, names): self.analyser_cache['countries'].append((code, names)) - def process_place(self, place): + @staticmethod + def process_place(place): + assert isinstance(place, PlaceInfo) return {}