mirror of
https://github.com/osm-search/Nominatim.git
synced 2026-02-15 02:47:59 +00:00
enable flake for Python tests
This commit is contained in:
@@ -2,13 +2,14 @@
|
||||
#
|
||||
# This file is part of Nominatim. (https://nominatim.org)
|
||||
#
|
||||
# Copyright (C) 2024 by the Nominatim developer community.
|
||||
# Copyright (C) 2025 by the Nominatim developer community.
|
||||
# For a full list of authors see the git log.
|
||||
"""
|
||||
Specialised psycopg cursor with shortcut functions useful for testing.
|
||||
"""
|
||||
import psycopg
|
||||
|
||||
|
||||
class CursorForTesting(psycopg.Cursor):
|
||||
""" Extension to the DictCursor class that provides execution
|
||||
short-cuts that simplify writing assertions.
|
||||
@@ -22,7 +23,6 @@ class CursorForTesting(psycopg.Cursor):
|
||||
assert self.rowcount == 1
|
||||
return self.fetchone()[0]
|
||||
|
||||
|
||||
def row_set(self, sql, params=None):
|
||||
""" Execute a query and return the result as a set of tuples.
|
||||
Fails when the SQL command returns duplicate rows.
|
||||
@@ -34,7 +34,6 @@ class CursorForTesting(psycopg.Cursor):
|
||||
|
||||
return result
|
||||
|
||||
|
||||
def table_exists(self, table):
|
||||
""" Check that a table with the given name exists in the database.
|
||||
"""
|
||||
@@ -42,7 +41,6 @@ class CursorForTesting(psycopg.Cursor):
|
||||
WHERE tablename = %s""", (table, ))
|
||||
return num == 1
|
||||
|
||||
|
||||
def index_exists(self, table, index):
|
||||
""" Check that an indexwith the given name exists on the given table.
|
||||
"""
|
||||
@@ -51,7 +49,6 @@ class CursorForTesting(psycopg.Cursor):
|
||||
(table, index))
|
||||
return num == 1
|
||||
|
||||
|
||||
def table_rows(self, table, where=None):
|
||||
""" Return the number of rows in the given table.
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user