StacDummyBuilder.asset: "data" role by default and add "proj:" helper arguments

Author: soxofaan

openeo/_version.py

@@ -1 +1 @@
-__version__ = "0.47.0a3"
+__version__ = "0.47.0a4"

openeo/testing/stac.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, Union
+from typing import Any, Dict, List, Optional, Sequence, Union
 
 
 class StacDummyBuilder:
@@ -117,13 +117,25 @@ def asset(
         cls,
         href: str = "https://stac.test/asset.tiff",
         type: str = "image/tiff; application=geotiff",
-        roles: Optional[List[str]] = None,
+        roles: Union[Sequence[str], None] = ("data",),
+        proj_code: Optional[str] = None,
+        proj_bbox: Optional[Sequence[float]] = None,
+        proj_shape: Optional[Sequence[int]] = None,
+        proj_transform: Optional[Sequence[float]] = None,
         **kwargs,
     ):
-        d = {"href": href, **kwargs}
+        d: Dict[str, Any] = {"href": href, **kwargs}
         if type:
             d["type"] = type
-        if roles is not None:
-            d["roles"] = roles
+        if roles:
+            d["roles"] = list(roles)
+        if proj_code:
+            d["proj:code"] = proj_code
+        if proj_bbox:
+            d["proj:bbox"] = list(proj_bbox)
+        if proj_shape:
+            d["proj:shape"] = list(proj_shape)
+        if proj_transform:
+            d["proj:transform"] = list(proj_transform)
 
         return d

tests/testing/test_stac.py

@@ -87,6 +87,32 @@ def test_asset_default(self):
         assert asset == {
             "href": "https://stac.test/asset.tiff",
             "type": "image/tiff; application=geotiff",
+            "roles": ["data"],
+        }
+        pystac.Asset.from_dict(asset)
+
+    def test_asset_no_roles(self):
+        asset = StacDummyBuilder.asset(roles=None)
+        assert asset == {
+            "href": "https://stac.test/asset.tiff",
+            "type": "image/tiff; application=geotiff",
+        }
+        pystac.Asset.from_dict(asset)
+
+    def test_asset_proj_fields(self):
+        asset = StacDummyBuilder.asset(
+            proj_code="EPSG:4326",
+            proj_bbox=(3, 51, 4, 52),
+            proj_shape=(100, 100),
+            proj_transform=(0.01, 0.0, 3.0, 0.0, -0.01, 52.0),
+        )
+        assert asset == {
+            "href": "https://stac.test/asset.tiff",
+            "type": "image/tiff; application=geotiff",
+            "roles": ["data"],
+            "proj:code": "EPSG:4326",
+            "proj:bbox": [3, 51, 4, 52],
+            "proj:shape": [100, 100],
+            "proj:transform": [0.01, 0.0, 3.0, 0.0, -0.01, 52.0],
         }
-        # Check if the default asset validates
         pystac.Asset.from_dict(asset)