X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/1e9f37ab82db1758235bedf83c659693f4ca6c3e..fc038261b9fe7f7d89a4000496185e72ac100a2a:/test/python/mock_legacy_word_table.py diff --git a/test/python/mock_legacy_word_table.py b/test/python/mock_legacy_word_table.py index 8baf3adc..40fe1044 100644 --- a/test/python/mock_legacy_word_table.py +++ b/test/python/mock_legacy_word_table.py @@ -20,6 +20,14 @@ class MockLegacyWordTable: conn.commit() + def add_full_word(self, word_id, word, word_token=None): + with self.conn.cursor() as cur: + cur.execute("""INSERT INTO word (word_id, word_token, word) + VALUES (%s, %s, %s) + """, (word_id, ' ' + (word_token or word), word)) + self.conn.commit() + + def add_special(self, word_token, word, cls, typ, oper): with self.conn.cursor() as cur: cur.execute("""INSERT INTO word (word_token, word, class, type, operator)