Merge remote-tracking branch 'upstream/master'

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

diff --git a/test/python/mocks.py b/test/python/mocks.py
index e95d57721357b7c5c8e7896feaf6434248f98b6f..f9faaa93c42e06fac7fdb11fe3fb6abeb132bd5e 100644 (file)
--- a/test/python/mocks.py
+++ b/test/python/mocks.py
@@ -98,6 +98,13 @@ class MockWordTable:
                            WHERE class = 'place' and type = 'postcode'""")
             return set((row[0] for row in cur))
 
                            WHERE class = 'place' and type = 'postcode'""")
             return set((row[0] for row in cur))
 

+    def get_partial_words(self):
+        with self.conn.cursor() as cur:
+            cur.execute("""SELECT word_token, search_name_count FROM word
+                           WHERE class is null and country_code is null
+                                 and not word_token like ' %'""")
+            return set((tuple(row) for row in cur))
+

 
 class MockPlacexTable:
     """ A placex table for testing.
 
 class MockPlacexTable:
     """ A placex table for testing.