X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/252fe42612f60641abbe44a24e460c989859a148..9734bbf2403146e914c34311beb314e36c735beb:/test/python/api/test_warm.py

diff --git a/test/python/api/test_warm.py b/test/python/api/test_warm.py
index af48732a..304943f1 100644
--- a/test/python/api/test_warm.py
+++ b/test/python/api/test_warm.py
@@ -2,14 +2,14 @@
 #
 # This file is part of Nominatim. (https://nominatim.org)
 #
-# Copyright (C) 2023 by the Nominatim developer community.
+# Copyright (C) 2024 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):
@@ -27,7 +27,7 @@ def setup_database_with_context(apiobj, table_factory):
 
 @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(module_dir='MODULE NOT AVAILABLE',
+                                           osm2pgsql_path='OSM2PGSQL NOT AVAILABLE',
+                                           cli_args=['admin', '--project-dir', str(tmp_path),
+                                                     '--warm'] + args)