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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
|
"""Tests for pystac.tests.extensions.render"""
import json
import pytest
import pystac
import pystac.errors
from pystac.extensions.render import Render, RenderExtension
from tests.conftest import get_data_file
@pytest.fixture
def ext_collection_uri() -> str:
return get_data_file("render/collection.json")
@pytest.fixture
def ext_collection(ext_collection_uri: str) -> pystac.Collection:
return pystac.Collection.from_file(ext_collection_uri)
@pytest.fixture
def ext_item_uri() -> str:
return get_data_file("render/item.json")
@pytest.fixture
def ext_item(ext_item_uri: str) -> pystac.Item:
return pystac.Item.from_file(ext_item_uri)
@pytest.fixture
def render() -> Render:
return Render.create(
assets=["B4", "B3", "B2"],
title="RGB",
rescale=[[0, 1000], [0, 1000], [0, 1000]],
nodata=-9999,
colormap_name="viridis",
colormap={
"0": "#e5f5f9",
"10": "#99d8c9",
"255": "#2ca25f",
},
color_formula="gamma rg 1.3, sigmoidal rgb 22 0.1, saturation 1.5",
resampling="bilinear",
expression="(B08-B04)/(B08+B04)",
minmax_zoom=[2, 18],
)
@pytest.fixture
def thumbnail_render(ext_item: pystac.Item) -> Render:
return RenderExtension.ext(ext_item).renders["thumbnail"]
def test_collection_stac_extensions(ext_collection: pystac.Collection) -> None:
assert RenderExtension.has_extension(ext_collection)
def test_item_stac_extensions(ext_item: pystac.Item) -> None:
assert RenderExtension.has_extension(ext_item)
def test_ext_raises_if_item_does_not_conform(item: pystac.Item) -> None:
with pytest.raises(pystac.errors.ExtensionNotImplemented):
RenderExtension.ext(item)
def test_ext_raises_if_collection_does_not_conform(
collection: pystac.Collection,
) -> None:
with pytest.raises(pystac.errors.ExtensionNotImplemented):
RenderExtension.ext(collection)
def test_ext_raises_on_catalog(catalog: pystac.Catalog) -> None:
with pytest.raises(
pystac.errors.ExtensionTypeError,
match="RenderExtension does not apply to type 'Catalog'",
):
RenderExtension.ext(catalog) # type: ignore
def test_item_to_from_dict(ext_item_uri: str, ext_item: pystac.Item) -> None:
with open(ext_item_uri) as f:
d = json.load(f)
actual = ext_item.to_dict(include_self_link=False)
assert actual == d
def test_collection_to_from_dict(
ext_collection_uri: str, ext_collection: pystac.Collection
) -> None:
with open(ext_collection_uri) as f:
d = json.load(f)
actual = ext_collection.to_dict(include_self_link=False)
assert actual == d
def test_add_to_item(item: pystac.Item) -> None:
assert not RenderExtension.has_extension(item)
RenderExtension.add_to(item)
assert RenderExtension.has_extension(item)
def test_add_to_collection(collection: pystac.Collection) -> None:
assert not RenderExtension.has_extension(collection)
RenderExtension.add_to(collection)
assert RenderExtension.has_extension(collection)
def test_get_render_values(thumbnail_render: Render) -> None:
assert thumbnail_render.title == "Thumbnail"
assert thumbnail_render.assets == ["B04", "B03", "B02"]
assert thumbnail_render.rescale == [[0, 150]]
assert thumbnail_render.colormap_name == "rainbow"
assert thumbnail_render.resampling == "bilinear"
assert thumbnail_render.properties.get("bidx") == [1]
assert thumbnail_render.properties.get("width") == 1024
assert thumbnail_render.properties.get("height") == 1024
assert thumbnail_render.properties.get("bands") == ["B4", "B3", "B2"]
def test_apply_renders_to_item(item: pystac.Item, render: Render) -> None:
RenderExtension.add_to(item)
RenderExtension.ext(item).apply({"render": render})
assert item.ext.render.renders["render"].assets == ["B4", "B3", "B2"]
assert item.ext.render.renders["render"].title == "RGB"
assert item.ext.render.renders["render"].rescale == [
[0, 1000],
[0, 1000],
[0, 1000],
]
assert item.ext.render.renders["render"].nodata == -9999
assert item.ext.render.renders["render"].colormap_name == "viridis"
assert item.ext.render.renders["render"].colormap == {
"0": "#e5f5f9",
"10": "#99d8c9",
"255": "#2ca25f",
}
assert (
item.ext.render.renders["render"].color_formula
== "gamma rg 1.3, sigmoidal rgb 22 0.1, saturation 1.5"
)
assert item.ext.render.renders["render"].resampling == "bilinear"
assert item.ext.render.renders["render"].expression == "(B08-B04)/(B08+B04)"
assert item.ext.render.renders["render"].minmax_zoom == [2, 18]
assert item.ext.render.renders["render"] == render
def test_get_unset_properties(item: pystac.Item) -> None:
RenderExtension.add_to(item)
RenderExtension.ext(item).apply(
{
"render": Render.create(
assets=["B4", "B3", "B2"],
)
}
)
assert item.ext.render.renders["render"].title is None
assert item.ext.render.renders["render"].rescale is None
assert item.ext.render.renders["render"].nodata is None
assert item.ext.render.renders["render"].colormap_name is None
assert item.ext.render.renders["render"].colormap is None
assert item.ext.render.renders["render"].color_formula is None
assert item.ext.render.renders["render"].resampling is None
assert item.ext.render.renders["render"].expression is None
assert item.ext.render.renders["render"].minmax_zoom is None
def test_equality_check_with_unexpected_type_raises_notimplemented_error() -> None:
render = Render.create(
assets=["B4", "B3", "B2"],
)
with pytest.raises(NotImplementedError):
_ = render == 1
def test_item_repr(ext_item: pystac.Item) -> None:
ext = RenderExtension.ext(ext_item)
assert ext.__repr__() == f"<ItemRenderExtension Item id={ext_item.id}>"
def test_collection_repr(ext_collection: pystac.Collection) -> None:
ext = RenderExtension.ext(ext_collection)
assert (
ext.__repr__()
== f"<CollectionRenderExtension Collection id={ext_collection.id}>"
)
def test_render_repr() -> None:
render = Render.create(
assets=["B4", "B3", "B2"],
title="RGB",
rescale=[[0, 1000], [0, 1000], [0, 1000]],
nodata=-9999,
colormap_name="viridis",
colormap={
"0": "#e5f5f9",
"10": "#99d8c9",
"255": "#2ca25f",
},
color_formula="gamma rg 1.3, sigmoidal rgb 22 0.1, saturation 1.5",
resampling="bilinear",
expression="(B08-B04)/(B08+B04)",
minmax_zoom=[2, 18],
)
assert render.__repr__() == (
"<Render assets=['B4', 'B3', 'B2'] "
"title=RGB rescale=[[0, 1000], [0, 1000], [0, 1000]] "
"nodata=-9999 colormap_name=viridis "
"colormap={'0': '#e5f5f9', '10': '#99d8c9', '255': '#2ca25f'} "
"color_formula=gamma rg 1.3, sigmoidal rgb 22 0.1, saturation 1.5 "
"resampling=bilinear expression=(B08-B04)/(B08+B04) minmax_zoom=[2, 18]>"
)
@pytest.mark.vcr
def test_item_validate(ext_item: pystac.Item) -> None:
assert ext_item.validate()
@pytest.mark.vcr
def test_collection_validate(ext_collection: pystac.Collection) -> None:
assert ext_collection.validate()
|
