use custom result formatters in CLI commands

[nominatim.git] / test / python / api / test_warm.py

diff --git a/test/python/api/test_warm.py b/test/python/api/test_warm.py
index af48732ac3d14af68fe6645c79ef5edd0d284cff..304943f1be27347edef3c189f6a916f1b42d3edd 100644 (file)
--- 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)