X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/efafa5271957fb54b356ec1c90e8613f14de40d4..a79c1bda9b77635e31f228e031ecbb6465f8dff4:/test/python/db/test_connection.py?ds=sidebyside

diff --git a/test/python/db/test_connection.py b/test/python/db/test_connection.py
index a89d9289..ed0537c8 100644
--- a/test/python/db/test_connection.py
+++ b/test/python/db/test_connection.py
@@ -26,6 +26,16 @@ def test_connection_table_exists(db, table_factory):
     assert db.table_exists('foobar')
 
 
+def test_has_column_no_table(db):
+    assert not db.table_has_column('sometable', 'somecolumn')
+
+
+@pytest.mark.parametrize('name,result', [('tram', True), ('car', False)])
+def test_has_column(db, table_factory, name, result):
+    table_factory('stuff', 'tram TEXT')
+
+    assert db.table_has_column('stuff', name) == result
+
 def test_connection_index_exists(db, table_factory, temp_db_cursor):
     assert not db.index_exists('some_index')