release 4.5.0.post3

[nominatim.git] / test / python / dummy_tokenizer.py

diff --git a/test/python/dummy_tokenizer.py b/test/python/dummy_tokenizer.py
index 0a86ba8d1598752a00af622e3466f4deede31c0e..4739e3f0af831303f4860f601b802537e781d898 100644 (file)
--- 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.
 """
 """
 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.
 
 def create(dsn, data_dir):
     """ Create a new instance of the tokenizer provided by this module.


@@ -17,16 +25,18 @@ class DummyTokenizer:
 
 
     def init_new_db(self, *args, **kwargs):
 
 
     def init_new_db(self, *args, **kwargs):

-        assert self.init_state == None
+        assert self.init_state is None

         self.init_state = "new"
 
 
         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"
 
 
         self.init_state = "loaded"
 
 

-    def finalize_import(self, _):
+    @staticmethod
+    def finalize_import(_):

         pass
 
 
         pass
 
 


@@ -51,17 +61,21 @@ class DummyNameAnalyzer:
     def close(self):
         pass
 
     def close(self):
         pass
 

-    def normalize_postcode(self, postcode):
+    @staticmethod
+    def normalize_postcode(postcode):

         return postcode
 
         return postcode
 

-    def update_postcodes_from_db(self):
+    @staticmethod
+    def update_postcodes_from_db():

         pass
 
         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))
 
         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 {}
         return {}