Skip to content

Commit

Permalink
add inspect first version
Browse files Browse the repository at this point in the history
  • Loading branch information
ValentinaHutter committed Jul 1, 2024
1 parent 1de1088 commit 80710f2
Show file tree
Hide file tree
Showing 3 changed files with 49 additions and 0 deletions.
1 change: 1 addition & 0 deletions openeo_processes_dask/process_implementations/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
from .arrays import *
from .comparison import *
from .cubes import *
from .inspect import *
from .logic import *
from .math import *

Expand Down
28 changes: 28 additions & 0 deletions openeo_processes_dask/process_implementations/inspect.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
import logging
from typing import Any, Optional

__all__ = ["inspect"]

logger = logging.getLogger(__name__)


def inspect(
data: Any,
message: Optional[str] = "",
code: Optional[str] = "User",
level: Optional[str] = "info",
) -> Any:

if level == "info":
logger.info(f"{code}: {message} {data}.")

elif level == "warning":
logger.warning(f"{code}: {message} {data}.")

elif level == "error":
logger.error(f"{code}: {message} {data}.")

elif level == "debug":
logger.debug(f"{code}: {message} {data}.")

return data
20 changes: 20 additions & 0 deletions tests/test_inspect.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
import numpy as np
import pytest
import xarray as xr

from openeo_processes_dask.process_implementations.inspect import inspect
from tests.mockdata import create_fake_rastercube


@pytest.mark.parametrize("size", [(6, 5, 4, 4)])
@pytest.mark.parametrize("dtype", [np.float32])
def test_inspect(temporal_interval, bounding_box, random_raster_data):
raster_cube = create_fake_rastercube(
data=random_raster_data,
spatial_extent=bounding_box,
temporal_extent=temporal_interval,
bands=["B02", "B03", "B04", "B08"],
)

assert inspect(np.zeros((1))) == np.zeros((1))
xr.testing.assert_equal(raster_cube, inspect(raster_cube))

0 comments on commit 80710f2

Please sign in to comment.