File size: 3,243 Bytes
7a8b84d 4e35839 7a8b84d 4e35839 7a8b84d 2f08493 4e35839 7a8b84d 4e35839 7a8b84d 2f08493 7a8b84d 2f08493 7a8b84d 2f08493 7a8b84d 2f08493 7a8b84d 4e35839 7a8b84d 2f08493 4e35839 7a8b84d 4e35839 7a8b84d 2f08493 7a8b84d 2f08493 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import unittest
import numpy as np
from src import app_logger
from src.io.tms2geotiff import download_extent
from src.utilities.utilities import hash_calculate
from tests import LOCAL_URL_TILE, TEST_EVENTS_FOLDER
input_bbox = [[39.036252959636606, 15.040283203125002], [38.302869955150044, 13.634033203125002]]
class TestTms2geotiff(unittest.TestCase):
def test_download_extent(self):
from rasterio import Affine
from tests.local_tiles_http_server import LocalTilesHttpServer
listen_port = 8000
with LocalTilesHttpServer.http_server("localhost", listen_port, directory=TEST_EVENTS_FOLDER):
pt0, pt1 = input_bbox
zoom = 10
n_lat = pt0[0]
e_lng = pt0[1]
s_lat = pt1[0]
w_lng = pt1[1]
img, matrix = download_extent(w=w_lng, s=s_lat, e=e_lng, n=n_lat, zoom=zoom, source=LOCAL_URL_TILE)
app_logger.info(f"# DOWNLOAD ENDED, shape: {img.shape} #")
np_img = np.ascontiguousarray(img)
output_hash = hash_calculate(np_img)
assert output_hash == b'UmbkwbPJpRT1XXcLnLUapUDP320w7YhS/AmT3H7u+b4='
assert Affine.to_gdal(matrix) == (
1517657.1966021745, 152.8740565703525, 0.0, 4726942.266183584, 0.0, -152.87405657034955)
def test_download_extent_io_error1(self):
with self.assertRaises(Exception):
try:
pt0, pt1 = input_bbox
zoom = 10
n_lat = pt0[0]
e_lng = pt0[1]
s_lat = pt1[0]
w_lng = pt1[1]
download_extent(w=w_lng, s=s_lat, e=e_lng, n=n_lat, zoom=zoom, source=f"http://{LOCAL_URL_TILE}")
print("exception not raised")
except ConnectionError as ioe1:
app_logger.error(f"ioe1:{ioe1}.")
msg0 = "HTTPConnectionPool(host='localhost', port=8000): Max retries exceeded with url: /lambda_handler"
msg1 = "Caused by NewConnectionError"
msg2 = ": Failed to establish a new connection: [Errno 61] Connection refused'))"
assert msg0 in str(ioe1)
assert msg1 in str(ioe1)
assert msg2 in str(ioe1)
raise ioe1
def test_download_extent_io_error2(self):
from requests import HTTPError
from tests.local_tiles_http_server import LocalTilesHttpServer
listen_port = 8000
with LocalTilesHttpServer.http_server("localhost", listen_port, directory=TEST_EVENTS_FOLDER):
pt0, pt1 = input_bbox
zoom = 10
with self.assertRaises(HTTPError):
try:
n_lat = pt0[0]
e_lng = pt0[1]
s_lat = pt1[0]
w_lng = pt1[1]
download_extent(w=w_lng, s=s_lat, e=e_lng, n=n_lat, zoom=zoom,
source=LOCAL_URL_TILE + "_not_found_raster!")
except HTTPError as http_e:
app_logger.error(f"ae:{http_e}.")
assert "Tile URL resulted in a 404 error. Double-check your tile url:" in str(http_e)
raise http_e
|