enable flake for Python tests

[nominatim.git] / test / python / tools / test_refresh_create_functions.py

diff --git a/test/python/tools/test_refresh_create_functions.py b/test/python/tools/test_refresh_create_functions.py
index 8d26e7554dd6d1613621133485226dad0015037d..bd8724d60138787eec910724642e78f88aea307a 100644 (file)
--- a/test/python/tools/test_refresh_create_functions.py
+++ b/test/python/tools/test_refresh_create_functions.py
@@ -1,15 +1,16 @@
-# SPDX-License-Identifier: GPL-2.0-only
+# SPDX-License-Identifier: GPL-3.0-or-later
 #
 # This file is part of Nominatim. (https://nominatim.org)
 #
-# Copyright (C) 2022 by the Nominatim developer community.
+# Copyright (C) 2025 by the Nominatim developer community.
 # For a full list of authors see the git log.
 """
 Tests for creating PL/pgSQL functions for Nominatim.
 """
 import pytest
 
-from nominatim.tools.refresh import create_functions
+from nominatim_db.tools.refresh import create_functions
+
 
 class TestCreateFunctions:
     @pytest.fixture(autouse=True)
@@ -18,12 +19,10 @@ class TestCreateFunctions:
         self.config = def_config
         def_config.lib_dir.sql = tmp_path
 
-
     def write_functions(self, content):
         sqlfile = self.config.lib_dir.sql / 'functions.sql'
         sqlfile.write_text(content)
 
-
     def test_create_functions(self, temp_db_cursor):
         self.write_functions("""CREATE OR REPLACE FUNCTION test() RETURNS INTEGER
                               AS $$
@@ -37,7 +36,6 @@ class TestCreateFunctions:
 
         assert temp_db_cursor.scalar('SELECT test()') == 43
 
-
     @pytest.mark.parametrize("dbg,ret", ((True, 43), (False, 22)))
     def test_create_functions_with_template(self, temp_db_cursor, dbg, ret):
         self.write_functions("""CREATE OR REPLACE FUNCTION test() RETURNS INTEGER