X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/9448c5e16f5220d1580f65a9ba800efc2e344ff5..c70dfccacac90a90bfc193f49262abe7505d7d75:/test/python/api/test_warm.py

diff --git a/test/python/api/test_warm.py b/test/python/api/test_warm.py
index af48732a..94081d00 100644
--- a/test/python/api/test_warm.py
+++ b/test/python/api/test_warm.py
@@ -2,14 +2,15 @@
 #
 # This file is part of Nominatim. (https://nominatim.org)
 #
-# Copyright (C) 2023 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 warm-up CLI function.
 """
 import pytest
 
-import nominatim.cli
+import nominatim_db.cli
+
 
 @pytest.fixture(autouse=True)
 def setup_database_with_context(apiobj, table_factory):
@@ -21,13 +22,13 @@ def setup_database_with_context(apiobj, table_factory):
     apiobj.add_data('properties',
                     [{'property': 'tokenizer', 'value': 'icu'},
                      {'property': 'tokenizer_import_normalisation', 'value': ':: lower();'},
-                     {'property': 'tokenizer_import_transliteration', 'value': "'1' > '/1/'; 'ä' > 'ä '"},
-                    ])
+                     {'property': 'tokenizer_import_transliteration',
+                      'value': "'1' > '/1/'; 'ä' > 'ä '"}
+                     ])
 
 
 @pytest.mark.parametrize('args', [['--search-only'], ['--reverse-only']])
 def test_warm_all(tmp_path, args):
-    assert 0 == nominatim.cli.nominatim(module_dir='MODULE NOT AVAILABLE',
-                                        osm2pgsql_path='OSM2PGSQL NOT AVAILABLE',
-                                        cli_args=['admin', '--project-dir', str(tmp_path),
-                                                  '--warm'] + args)
+    assert 0 == nominatim_db.cli.nominatim(osm2pgsql_path='OSM2PGSQL NOT AVAILABLE',
+                                           cli_args=['admin', '--project-dir', str(tmp_path),
+                                                     '--warm'] + args)