Skip to content

Commit c9e666f

Browse files
authored
Fixup attrs (#40)
* Fixup attrs
1 parent f078c9b commit c9e666f

File tree

3 files changed

+11
-0
lines changed

3 files changed

+11
-0
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,3 +3,4 @@ __pycache__
33
htmlcov
44
dist
55
.direnv
6+
.venv

tests/test_xstac.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import json
12
import pyproj
23
import pytest
34
import pystac
@@ -164,3 +165,10 @@ def test_disable_infer_temporal_extent(ds, item_template):
164165
# del item_template["properties"]["datetime"]
165166
result = xarray_to_stac(ds, item_template, temporal_dimension=False)
166167
assert "start_datetime" not in result.properties
168+
169+
170+
def test_fixup_numpy_attrs_by_default(ds, item_template):
171+
ds.prcp.attrs["values"] = np.zeros(2)
172+
result = xarray_to_stac(ds, item_template, temporal_dimension=False)
173+
assert result.properties["cube:variables"]["prcp"]["attrs"]["values"] == [0.0, 0.0]
174+
json.dumps(result.to_dict())

xstac/_xstac.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -339,6 +339,8 @@ def xarray_to_stac(
339339
**additional_dimensions:
340340
A dictionary with keys ``extent``, ``values``, ``step``.
341341
"""
342+
ds = fix_attrs(ds)
343+
342344
temporal_dimension = maybe_use_cf_standard_axis(
343345
temporal_dimension, "temporal_dimension", ds
344346
)

0 commit comments

Comments
 (0)