mirror of
https://gitlvb.teallvbs.xyz/IQ.Lvbs/IQ.Pilot.git
synced 2026-08-05 08:15:41 +08:00
IQ.Lvbs CI [bot]
129 lines
5.4 KiB
Python
129 lines
5.4 KiB
Python
import os
|
|
import sqlite3
|
|
import json
|
|
|
|
from openpilot.iqpilot.ui.onroad import offline_tiles
|
|
|
|
|
|
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 _write_mbtiles(path):
|
|
conn = sqlite3.connect(path)
|
|
conn.execute("CREATE TABLE metadata (name text, value text)")
|
|
conn.execute("CREATE TABLE tiles (zoom_level integer, tile_column integer, tile_row integer, tile_data blob)")
|
|
conn.execute("INSERT INTO metadata (name, value) VALUES ('format', 'png')")
|
|
conn.execute("INSERT INTO metadata (name, value) VALUES ('minzoom', '1')")
|
|
conn.execute("INSERT INTO metadata (name, value) VALUES ('maxzoom', '3')")
|
|
conn.execute(
|
|
"INSERT INTO tiles (zoom_level, tile_column, tile_row, tile_data) VALUES (?, ?, ?, ?)",
|
|
(1, 1, offline_tiles.xyz_to_tms_y(1, 0), PNG_1X1),
|
|
)
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
def test_xyz_to_tms_y():
|
|
assert offline_tiles.xyz_to_tms_y(1, 0) == 1
|
|
assert offline_tiles.xyz_to_tms_y(1, 1) == 0
|
|
|
|
|
|
def test_find_offline_mbtiles_path_from_env(tmp_path, monkeypatch):
|
|
mbtiles = tmp_path / "demo.mbtiles"
|
|
_write_mbtiles(mbtiles)
|
|
monkeypatch.setenv(offline_tiles.OFFLINE_MBTILES_ENV, str(mbtiles))
|
|
assert offline_tiles.find_offline_mbtiles_path() == mbtiles
|
|
|
|
|
|
def test_find_offline_xyz_root(tmp_path, monkeypatch):
|
|
root = tmp_path / "tiles"
|
|
(root / "15" / "10500").mkdir(parents=True)
|
|
monkeypatch.setenv(offline_tiles.OFFLINE_TILE_ROOT_ENV, str(root))
|
|
assert offline_tiles.find_offline_xyz_root() == root
|
|
|
|
|
|
def test_load_raster_tile_blob_from_mbtiles(tmp_path):
|
|
mbtiles = tmp_path / "offline.mbtiles"
|
|
_write_mbtiles(mbtiles)
|
|
conn = offline_tiles.open_mbtiles(mbtiles)
|
|
try:
|
|
assert offline_tiles.mbtiles_is_raster(conn) is True
|
|
assert offline_tiles.mbtiles_zoom_bounds(conn) == (1, 3)
|
|
assert offline_tiles.load_raster_tile_blob(conn, 1, 1, 0) == PNG_1X1
|
|
finally:
|
|
conn.close()
|
|
|
|
|
|
def test_load_raster_tile_blob_from_xyz_dir(tmp_path, monkeypatch):
|
|
root = tmp_path / "offline_tiles"
|
|
tile_path = root / "14" / "2625"
|
|
tile_path.mkdir(parents=True)
|
|
(tile_path / "6335@2x.png").write_bytes(PNG_1X1)
|
|
monkeypatch.setenv(offline_tiles.OFFLINE_TILE_ROOT_ENV, str(root))
|
|
assert offline_tiles.xyz_zoom_bounds(root) == (14, 14)
|
|
assert offline_tiles.load_raster_xyz_tile_blob(root, 14, 2625, 6335) == PNG_1X1
|
|
|
|
|
|
def test_find_offline_region_root_by_bounds(tmp_path, monkeypatch):
|
|
root = tmp_path / "offline_maps"
|
|
il = root / "regions" / "illinois"
|
|
ca = root / "regions" / "california"
|
|
(il / "tiles").mkdir(parents=True)
|
|
(ca / "tiles").mkdir(parents=True)
|
|
(il / "manifest.json").write_text(json.dumps({"mbtiles": {"bounds": "-91.6,36.9,-87.4,42.6"}}))
|
|
(ca / "manifest.json").write_text(json.dumps({"mbtiles": {"bounds": "-124.5,32.4,-114.1,42.1"}}))
|
|
|
|
monkeypatch.setenv(offline_tiles.OFFLINE_TILE_ROOT_ENV, str(root / "tiles"))
|
|
assert offline_tiles.find_offline_region_root(41.88, -87.63) == il
|
|
assert offline_tiles.find_offline_region_root(34.05, -118.24) == ca
|
|
|
|
|
|
def test_find_offline_mbtiles_path_uses_selected_region(tmp_path, monkeypatch):
|
|
root = tmp_path / "offline_maps"
|
|
il = root / "regions" / "illinois"
|
|
ca = root / "regions" / "california"
|
|
(il / "tiles").mkdir(parents=True)
|
|
(ca / "tiles").mkdir(parents=True)
|
|
(il / "manifest.json").write_text(json.dumps({"mbtiles": {"bounds": "-91.6,36.9,-87.4,42.6"}}))
|
|
(ca / "manifest.json").write_text(json.dumps({"mbtiles": {"bounds": "-124.5,32.4,-114.1,42.1"}}))
|
|
_write_mbtiles(il / "tiles" / "offline.mbtiles")
|
|
_write_mbtiles(ca / "tiles" / "offline.mbtiles")
|
|
|
|
monkeypatch.setenv(offline_tiles.OFFLINE_TILE_ROOT_ENV, str(root / "tiles"))
|
|
assert offline_tiles.find_offline_mbtiles_path(41.88, -87.63) == il / "tiles" / "offline.mbtiles"
|
|
assert offline_tiles.find_offline_mbtiles_path(34.05, -118.24) == ca / "tiles" / "offline.mbtiles"
|
|
|
|
|
|
def test_day_variant_selection(tmp_path, monkeypatch):
|
|
root = tmp_path / "offline_maps"
|
|
region = root / "regions" / "us_state.IL"
|
|
(region / "tiles").mkdir(parents=True)
|
|
(region / "manifest.json").write_text(json.dumps({"mbtiles": {"bounds": "-88.3,41.5,-87.8,41.9"}}))
|
|
_write_mbtiles(region / "tiles" / "offline.mbtiles")
|
|
monkeypatch.setenv(offline_tiles.OFFLINE_TILE_ROOT_ENV, str(root / "tiles"))
|
|
offline_tiles._region_roots_cache = None
|
|
offline_tiles._region_bounds_cache.clear()
|
|
|
|
# no day variant yet: day request falls back to the night set
|
|
night = region / "tiles" / "offline.mbtiles"
|
|
assert offline_tiles.find_offline_mbtiles_path(41.7, -88.0, day=True) == night
|
|
assert offline_tiles.find_offline_mbtiles_path(41.7, -88.0, day=False) == night
|
|
|
|
# day variant installed: day requests prefer it, night unchanged
|
|
_write_mbtiles(region / "tiles" / "offline_day.mbtiles")
|
|
assert offline_tiles.find_offline_mbtiles_path(41.7, -88.0, day=True) == region / "tiles" / "offline_day.mbtiles"
|
|
assert offline_tiles.find_offline_mbtiles_path(41.7, -88.0, day=False) == night
|
|
|
|
|
|
def test_solar_elevation_day_night():
|
|
from openpilot.iqpilot.ui.onroad.nav_map_utils import solar_elevation_deg
|
|
# Chicago 2026-07-11: 18:00 UTC (1pm CDT) is day; 06:00 UTC (1am CDT) is night
|
|
noon_utc = 1783792800.0 # 2026-07-11 18:00:00 UTC
|
|
night_utc = noon_utc - 12 * 3600
|
|
assert solar_elevation_deg(41.88, -87.63, noon_utc) > 30.0
|
|
assert solar_elevation_deg(41.88, -87.63, night_utc) < -10.0
|