Skip to content

Commit 7813dfe

Browse files
committed
chore: minor code style cleanups
1 parent ce01428 commit 7813dfe

File tree

2 files changed

+20
-19
lines changed

2 files changed

+20
-19
lines changed

openeo/metadata.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,10 @@ def rename_labels(self, target, source) -> 'Dimension':
4949
class SpatialDimension(Dimension):
5050
DEFAULT_CRS = 4326
5151

52-
def __init__(self, name: str, extent: Union[Tuple[float, float], List[float]],
53-
crs: Union[str, int, dict] = DEFAULT_CRS, step = None):
52+
def __init__(
53+
self, name: str, extent: Union[Tuple[float, float], List[float]],
54+
crs: Union[str, int, dict] = DEFAULT_CRS, step=None
55+
):
5456
"""
5557
5658
@param name:
@@ -213,7 +215,9 @@ def get_or_create(cls, metadata: Union[dict, 'CollectionMetadata', None]) -> 'Co
213215
else:
214216
return cls(metadata=metadata or {})
215217

216-
def _clone_and_update(self, metadata: dict = None, dimensions: List[Dimension] = None, **kwargs) -> 'CollectionMetadata':
218+
def _clone_and_update(
219+
self, metadata: dict = None, dimensions: List[Dimension] = None, **kwargs
220+
) -> 'CollectionMetadata':
217221
"""Create a new instance (of same class) with copied/updated fields."""
218222
cls = type(self)
219223
if dimensions == None:
@@ -436,9 +440,7 @@ def drop_dimension(self, name: str = None) -> 'CollectionMetadata':
436440
return self._clone_and_update(dimensions=[d for d in self._dimensions if not d.name == name])
437441

438442
def _repr_html_(self):
439-
return render_component('collection', data = self._orig_metadata)
443+
return render_component('collection', data=self._orig_metadata)
440444

441445
def __str__(self) -> str:
442446
return f"CollectionMetadata({self.extent} - {self.band_names} - {self.dimension_names()})"
443-
444-

tests/test_metadata.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -97,33 +97,32 @@ def test_band_dimension_rename_labels():
9797
b03 = Band("B03", "green", 0.560)
9898
b04 = Band("B04", "red", 0.665)
9999
bdim = BandDimension(name="bs", bands=[b02, b03, b04])
100-
metadata = CollectionMetadata({},dimensions=[bdim])
101-
newdim = metadata.rename_labels("bs",target=['1','2','3']).band_dimension
100+
metadata = CollectionMetadata({}, dimensions=[bdim])
101+
newdim = metadata.rename_labels("bs", target=['1', '2', '3']).band_dimension
102102

103-
assert metadata.band_dimension.band_names == ['B02','B03','B04']
104-
assert newdim.band_names == ['1','2','3']
103+
assert metadata.band_dimension.band_names == ['B02', 'B03', 'B04']
104+
assert newdim.band_names == ['1', '2', '3']
105105

106106

107107
def test_band_dimension_set_labels():
108-
109-
bdim = BandDimension(name="bs", bands=[Band('some_name',None,None)])
110-
metadata = CollectionMetadata({},dimensions=[bdim])
111-
newdim = metadata.rename_labels("bs",target=['1','2','3']).band_dimension
108+
bdim = BandDimension(name="bs", bands=[Band('some_name', None, None)])
109+
metadata = CollectionMetadata({}, dimensions=[bdim])
110+
newdim = metadata.rename_labels("bs", target=['1', '2', '3']).band_dimension
112111

113112
assert metadata.band_dimension.band_names == ['some_name']
114-
assert newdim.band_names == ['1','2','3']
113+
assert newdim.band_names == ['1', '2', '3']
115114

116115

117116
def test_band_dimension_rename_labels_with_source():
118117
b02 = Band("B02", "blue", 0.490)
119118
b03 = Band("B03", "green", 0.560)
120119
b04 = Band("B04", "red", 0.665)
121120
bdim = BandDimension(name="bs", bands=[b02, b03, b04])
122-
metadata = CollectionMetadata({},dimensions=[bdim])
123-
newdim = metadata.rename_labels("bs",target=['2'],source=['B03']).band_dimension
121+
metadata = CollectionMetadata({}, dimensions=[bdim])
122+
newdim = metadata.rename_labels("bs", target=['2'], source=['B03']).band_dimension
124123

125-
assert metadata.band_dimension.band_names == ['B02','B03','B04']
126-
assert newdim.band_names == ['B02','2','B04']
124+
assert metadata.band_dimension.band_names == ['B02', 'B03', 'B04']
125+
assert newdim.band_names == ['B02', '2', 'B04']
127126

128127

129128
def test_band_dimension_rename_labels_with_source_mismatch():

0 commit comments

Comments
 (0)