Skip to content

Commit 1e2f9a5

Browse files
committed
Add basic __repr__ to DriverDataCube
1 parent 735a2df commit 1e2f9a5

File tree

3 files changed

+27
-1
lines changed

3 files changed

+27
-1
lines changed

openeo_driver/datacube.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,9 @@ def __init__(self, metadata: Optional[CubeMetadata] = None):
6363
metadata = CollectionMetadata(metadata=metadata)
6464
self.metadata: CubeMetadata = metadata or CubeMetadata()
6565

66+
def __repr__(self) -> str:
67+
return f"{self.__class__.__name__}(metadata={self.metadata!r})"
68+
6669
def __eq__(self, o: object) -> bool:
6770
if o.__class__ == self.__class__:
6871
if o.metadata == self.metadata:

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
"flask>=2.0.0",
5757
"werkzeug>=3.0.3", # https://github.com/Open-EO/openeo-python-driver/issues/243
5858
"requests>=2.28.0",
59-
"openeo>=0.42.0.dev",
59+
"openeo>=0.43.0.a2.dev",
6060
"openeo_processes==0.0.4", # 0.0.4 is special build/release, also see https://github.com/Open-EO/openeo-python-driver/issues/152
6161
"gunicorn>=20.0.1",
6262
"numpy>=1.22.0",

tests/test_datacube.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
import openeo.metadata
2+
from openeo_driver.datacube import DriverDataCube
3+
import dirty_equals
4+
5+
6+
class TestDriverDatacCube:
7+
def test_repr_default(self):
8+
cube = DriverDataCube()
9+
assert repr(cube) == "DriverDataCube(metadata=CubeMetadata(dimensions=None))"
10+
11+
def test_repr_simple(self):
12+
metadata = openeo.metadata.CubeMetadata(
13+
dimensions=[
14+
openeo.metadata.SpatialDimension(name="x", extent=[12, 34]),
15+
openeo.metadata.BandDimension(
16+
name="bands", bands=[openeo.metadata.Band("B02"), openeo.metadata.Band("B03")]
17+
),
18+
]
19+
)
20+
cube = DriverDataCube(metadata=metadata)
21+
assert repr(cube) == dirty_equals.IsStr(
22+
regex=r"DriverDataCube\(metadata=CubeMetadata\(dimensions=\[SpatialDimension\(.*BandDimension\(.*B02.*B03.*"
23+
)

0 commit comments

Comments
 (0)