Merge pull request #2192 from lonvia/database-versioning

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

diff --git a/test/python/test_tools_refresh_create_functions.py b/test/python/test_tools_refresh_create_functions.py
index 4807e64f12e1b4d37c918d9ea5ed6b2179af8c77..ac2f221126f79c4da85249a8e5facb9c7bcd6f68 100644 (file)
--- a/test/python/test_tools_refresh_create_functions.py
+++ b/test/python/test_tools_refresh_create_functions.py
@@ -7,13 +7,12 @@ import pytest
 from nominatim.db.connection import connect
 from nominatim.tools.refresh import _get_standard_function_sql, _get_partition_function_sql
 
 from nominatim.db.connection import connect
 from nominatim.tools.refresh import _get_standard_function_sql, _get_partition_function_sql
 

-SQL_DIR = (Path(__file__) / '..' / '..' / '..' / 'sql').resolve()
+SQL_DIR = (Path(__file__) / '..' / '..' / '..' / 'lib-sql').resolve()

 
 @pytest.fixture
 def db(temp_db):
 
 @pytest.fixture
 def db(temp_db):

-    conn = connect('dbname=' + temp_db)
-    yield conn
-    conn.close()
+    with connect('dbname=' + temp_db) as conn:
+        yield conn

 
 @pytest.fixture
 def db_with_tables(db):
 
 @pytest.fixture
 def db_with_tables(db):