X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/ffc2d82b0ed150d52a718dc563f9399062e579a7..f3c4196c758a7720aefc26a7043bf1c6553028e2:/test/python/dummy_tokenizer.py diff --git a/test/python/dummy_tokenizer.py b/test/python/dummy_tokenizer.py index ceea4a7e..69202bc3 100644 --- a/test/python/dummy_tokenizer.py +++ b/test/python/dummy_tokenizer.py @@ -13,20 +13,26 @@ class DummyTokenizer: self.dsn = dsn self.data_dir = data_dir self.init_state = None + 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 + assert self.init_state is None self.init_state = "loaded" + @staticmethod + def finalize_import(_): + pass + + def name_analyzer(self): - return DummyNameAnalyzer() + return DummyNameAnalyzer(self.analyser_cache) class DummyNameAnalyzer: @@ -38,18 +44,28 @@ class DummyNameAnalyzer: self.close() + def __init__(self, cache): + self.analyser_cache = cache + cache['countries'] = [] + + def close(self): - """ Free all resources used by the analyzer. - """ pass - def add_postcodes_from_db(self): + @staticmethod + def normalize_postcode(postcode): + return postcode + + @staticmethod + def update_postcodes_from_db(): pass - def process_place(self, place): - """ Determine tokenizer information about the given place. + 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)) - Returns a JSON-serialisable structure that will be handed into - the database via the token_info field. - """ + @staticmethod + def process_place(place): return {}