From f9841876dddaedcd3cd9df52c6ec3795b5cc4ce3 Mon Sep 17 00:00:00 2001
From: Fabian Gebhart <fabian.gebhart@gmail.com>
Date: Fri, 18 Feb 2022 20:11:40 +0100
Subject: [PATCH] TST: add test for merging and clipping tiffs to desired bbox

---
 tests/conftest.py  | 16 ++++++++++++++++
 tests/test_mapa.py | 30 +++++++++++++++++++++++++++++-
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index bb0105f..3901183 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -61,3 +61,19 @@ def geojson_bbox():
             ]
         ],
     }
+
+
+@pytest.fixture
+def geojson_bbox_two_stac_items():
+    yield {
+        "type": "Polygon",
+        "coordinates": [
+            [
+                [8.631413, 41.318388],
+                [8.631413, 41.762435],
+                [9.685828, 41.762435],
+                [9.685828, 41.318388],
+                [8.631413, 41.318388],
+            ]
+        ],
+    }
diff --git a/tests/test_mapa.py b/tests/test_mapa.py
index 0ac4000..dba4a0c 100644
--- a/tests/test_mapa.py
+++ b/tests/test_mapa.py
@@ -1,7 +1,12 @@
 import math
 
-from mapa import convert_bbox_to_stl
+import rasterio as rio
+
+from mapa import _fetch_merge_and_clip_tiffs, _get_tiff_for_bbox, convert_bbox_to_stl
+from mapa.caching import get_hash_of_geojson
+from mapa.stac import _turn_geojson_into_bbox
 from mapa.stl_file import get_dimensions_of_stl_file
+from mapa.utils import _path_to_clipped_tiff, _path_to_merged_tiff
 
 
 def test_create_stl_for_bbox__success(output_file, geojson_bbox) -> None:
@@ -94,3 +99,26 @@ def test_convert_bbox_to_stl__verify_x_y_dimensions(output_file) -> None:
     assert x == 2 * y == 200
     # z dimension should stay the same
     assert z1 == z2
+
+
+def test__get_tiff_for_bbox(geojson_bbox) -> None:
+    tiff = _get_tiff_for_bbox(geojson_bbox, allow_caching=True)
+    assert tiff.is_file()
+
+
+def test__fetch_merge_and_clip_tiffs(geojson_bbox_two_stac_items) -> None:
+    bbox_hash = get_hash_of_geojson(geojson_bbox_two_stac_items)
+    assert isinstance(bbox_hash, str)
+    tiff = _fetch_merge_and_clip_tiffs(geojson_bbox_two_stac_items, bbox_hash, allow_caching=True)
+    assert tiff.is_file()
+    assert _path_to_merged_tiff(bbox_hash).is_file()
+    assert _path_to_clipped_tiff(bbox_hash).is_file()
+
+    # verify coordinates of resulting tiff is in line with the coordinates of the input bbox
+    bbox = _turn_geojson_into_bbox(geojson_bbox_two_stac_items)
+    left, bottom, right, top = bbox[0], bbox[1], bbox[2], bbox[3]
+    data = rio.open(tiff)
+    assert math.isclose(data.bounds.left, left, rel_tol=0.0001)
+    assert math.isclose(data.bounds.bottom, bottom, rel_tol=0.0001)
+    assert math.isclose(data.bounds.right, right, rel_tol=0.0001)
+    assert math.isclose(data.bounds.top, top, rel_tol=0.0001)