40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import sqlite3
|
|
|
|
from scripts.iqpilot.package_xyz_tiles_to_mbtiles import build_mbtiles
|
|
from scripts.iqpilot.render_raster_tiles_from_vector_mbtiles import tile_range_for_bounds
|
|
|
|
|
|
PNG_1X1 = (
|
|
b"\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01\x00\x00\x00\x01"
|
|
b"\x08\x06\x00\x00\x00\x1f\x15\xc4\x89\x00\x00\x00\rIDATx\x9cc\xf8\xcf"
|
|
b"\xc0\xf0\x1f\x00\x05\x00\x01\xff\x89\x99=\x1d\x00\x00\x00\x00IEND\xaeB`\x82"
|
|
)
|
|
|
|
|
|
def test_tile_range_for_bounds_returns_non_empty_ranges():
|
|
x_range, y_range = tile_range_for_bounds((-88.15, 41.65, -88.02, 41.76), 14)
|
|
assert len(list(x_range)) > 0
|
|
assert len(list(y_range)) > 0
|
|
|
|
|
|
def test_build_mbtiles_from_xyz_tiles(tmp_path):
|
|
source = tmp_path / "xyz"
|
|
tile_dir = source / "14" / "2625"
|
|
tile_dir.mkdir(parents=True)
|
|
(tile_dir / "6335@2x.png").write_bytes(PNG_1X1)
|
|
|
|
output = tmp_path / "offline.mbtiles"
|
|
build_mbtiles(source, output, bounds="-88.15,41.65,-88.02,41.76")
|
|
|
|
conn = sqlite3.connect(output)
|
|
try:
|
|
fmt = conn.execute("SELECT value FROM metadata WHERE name='format'").fetchone()[0]
|
|
bounds = conn.execute("SELECT value FROM metadata WHERE name='bounds'").fetchone()[0]
|
|
count = conn.execute("SELECT COUNT(*) FROM tiles").fetchone()[0]
|
|
finally:
|
|
conn.close()
|
|
|
|
assert fmt == "png"
|
|
assert bounds == "-88.15,41.65,-88.02,41.76"
|
|
assert count == 1
|