Merge pull request #3467 from mtmail/docs-datrie

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

diff --git a/test/python/cursor.py b/test/python/cursor.py
index 46069020b1cc07ba05ac6ef3d6535e502bbc4e50..7d586b3c1520d9e0fea75cbd144c6111b3c0e58a 100644 (file)
--- a/test/python/cursor.py
+++ b/test/python/cursor.py
@@ -1,3 +1,9 @@
+# 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.

 """
 Specialised psycopg2 cursor with shortcut functions useful for testing.
 """
 """
 Specialised psycopg2 cursor with shortcut functions useful for testing.
 """


@@ -37,6 +43,15 @@ class CursorForTesting(psycopg2.extras.DictCursor):
         return num == 1
 
 
         return num == 1
 
 

+    def index_exists(self, table, index):
+        """ Check that an indexwith the given name exists on the given table.
+        """
+        num = self.scalar("""SELECT count(*) FROM pg_indexes
+                             WHERE tablename = %s and indexname = %s""",
+                          (table, index))
+        return num == 1
+
+

     def table_rows(self, table, where=None):
         """ Return the number of rows in the given table.
         """
     def table_rows(self, table, where=None):
         """ Return the number of rows in the given table.
         """