Initial implementation of GeoTIFF import functionality

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

diff --git a/test/python/tools/test_refresh.py b/test/python/tools/test_refresh.py
index ac52aa36091643923d1c6791038b78ddc8f8c8d9..c8ebdab8734eb94d0900811bd20d52ee28d1b3cb 100644 (file)
--- a/test/python/tools/test_refresh.py
+++ b/test/python/tools/test_refresh.py
@@ -17,6 +17,10 @@ def test_refresh_import_wikipedia_not_existing(dsn):
     assert refresh.import_wikipedia_articles(dsn, Path('.')) == 1
 
 
+def test_refresh_import_osm_views_geotiff_not_existing(dsn):
+    assert refresh.import_osm_views_geotiff(dsn, Path('.')) == 1
+
+
 @pytest.mark.parametrize("replace", (True, False))
 def test_refresh_import_wikipedia(dsn, src_dir, table_factory, temp_db_cursor, replace):
     if replace:
@@ -30,6 +34,17 @@ def test_refresh_import_wikipedia(dsn, src_dir, table_factory, temp_db_cursor, r
     assert temp_db_cursor.table_rows('wikipedia_redirect') > 0
 
 
+@pytest.mark.parametrize("replace", (True, False))
+def test_refresh_import_osm_views_geotiff(dsn, src_dir, table_factory, temp_db_cursor, replace):
+    if replace:
+        table_factory('osmviews')
+
+    # use the small osm views GeoTIFF file for the API testdb
+    assert refresh.import_osm_views_geotiff(dsn, src_dir / 'test' / 'testdb') == 0
+
+    assert temp_db_cursor.table_rows('osmviews') > 0
+
+
 def test_recompute_importance(placex_table, table_factory, temp_db_conn, temp_db_cursor):
     temp_db_cursor.execute("""CREATE OR REPLACE FUNCTION compute_importance(extratags HSTORE,
                                               country_code varchar(2),