test_height_maps.py

from pathlib import Path

from srtm.height_maps import Srtm3HeightMap
from srtm.utilities import get_srtm3_file_path


def test_get_altitude_for_pixed():
    height_map = Srtm3HeightMap(path=get_srtm3_file_path("N40W008"))
    # Values taken from raster parsed by postgis, loaded with raster2pgsql
    assert height_map.get_altitude_for_pixel(x=1, y=1) == 1130
    assert height_map.get_altitude_for_pixel(x=1, y=2) == 1137
    assert height_map.get_altitude_for_pixel(x=1201, y=1) == 317
    assert height_map.get_altitude_for_pixel(x=1, y=1201) == 620
    assert height_map.get_altitude_for_pixel(x=1201, y=1201) == 390


def test_latitude_and_longitude_to_coordinates_corner1():
    height_map = Srtm3HeightMap(path=Path("/dummy/N40W008.hgt.zip"))
    # x,y = 1, 1
    assert height_map._latitude_and_longitude_to_coordinates(
        latitude=41, longitude=-8
    ) == (1, 1)


def test_latitude_and_longitude_to_coordinates_corner2():
    height_map = Srtm3HeightMap(path=Path("/dummy/N40W008.hgt.zip"))
    # x,y = 1201, 1201
    assert height_map._latitude_and_longitude_to_coordinates(
        latitude=40, longitude=-7
    ) == (1201, 1201)


def test_latitude_and_longitude_to_coordinates_ne():
    height_map = Srtm3HeightMap(path=Path("/dummy/N40E015.hgt.zip"))
    assert height_map._latitude_and_longitude_to_coordinates(
        latitude=40.6208333, longitude=15.101666
    ) == (123, 456)


def test_latitude_and_longitude_to_coordinates_nw():
    height_map = Srtm3HeightMap(path=Path("/dummy/N40W008.hgt.zip"))
    assert height_map._latitude_and_longitude_to_coordinates(
        latitude=40.6208333, longitude=-7.898333
    ) == (123, 456)


def test_latitude_and_longitude_to_coordinates_se():
    height_map = Srtm3HeightMap(path=Path("/dummy/S34E018.hgt.zip"))
    assert height_map._latitude_and_longitude_to_coordinates(
        latitude=-33.37916666, longitude=18.1016666
    ) == (123, 456)


def test_latitude_and_longitude_to_coordinates_sw():
    height_map = Srtm3HeightMap(path=Path("/dummy/S08W015.hgt.zip"))
    assert height_map._latitude_and_longitude_to_coordinates(
        latitude=-7.37916666666667, longitude=-14.8983333333333
    ) == (123, 456)


def test_get_altitude():
    height_map = Srtm3HeightMap(path=get_srtm3_file_path("N40W008"))
    assert (
        height_map.get_altitude_for_latitude_and_longitude(latitude=40, longitude=-7)
        == 390
    )