X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/8bf15fa691e286c27c05619ae102fb4db16eda0d..2171b385519841af0e175d88e51741c5322fba25:/test/python/cursor.py?ds=inline diff --git a/test/python/cursor.py b/test/python/cursor.py index 46069020..73b83b1d 100644 --- a/test/python/cursor.py +++ b/test/python/cursor.py @@ -1,3 +1,9 @@ +# SPDX-License-Identifier: GPL-2.0-only +# +# This file is part of Nominatim. (https://nominatim.org) +# +# Copyright (C) 2022 by the Nominatim developer community. +# For a full list of authors see the git log. """ Specialised psycopg2 cursor with shortcut functions useful for testing. """ @@ -37,6 +43,15 @@ class CursorForTesting(psycopg2.extras.DictCursor): 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. """