Merge pull request #20 from CU-ESIIL/codex/add-show_cube_lexcube-verb-for-pipe

Add Lexcube pipe verb

Author: ttuff

src/cubedynamics/tests/test_piping_ops.py

@@ -6,6 +6,7 @@
 import pandas as pd
 import xarray as xr
 
+import cubedynamics.viz as viz
 from cubedynamics import pipe, verbs as v
 
 
@@ -47,3 +48,21 @@ def test_to_netcdf_roundtrip(tmp_path):
     loaded = xr.load_dataarray(path)
     xr.testing.assert_identical(da, loaded)
     xr.testing.assert_identical(da, result)
+
+
+def test_show_cube_lexcube_returns_original_cube(monkeypatch):
+    da = _make_time_series()
+    captured = {}
+
+    def fake_show(cube, **kwargs):
+        captured["cube"] = cube
+        captured["kwargs"] = kwargs
+        return "widget"
+
+    monkeypatch.setattr(viz, "show_cube_lexcube", fake_show)
+
+    result = (pipe(da) | v.show_cube_lexcube(cmap="Blues")).unwrap()
+
+    assert result is da
+    assert captured["cube"] is da
+    assert captured["kwargs"] == {"cmap": "Blues"}

src/cubedynamics/verbs/__init__.py

@@ -2,11 +2,23 @@
 
 from __future__ import annotations
 
+from .. import viz
 from ..ops.io import to_netcdf
 from ..ops.ndvi import ndvi_from_s2
 from ..ops.stats import correlation_cube, variance, zscore
 from ..ops.transforms import anomaly, month_filter
 
+
+def show_cube_lexcube(**kwargs):
+    """Pipe verb that renders the current cube via Lexcube widgets."""
+
+    def _op(da):
+        viz.show_cube_lexcube(da, **kwargs)
+        return da
+
+    return _op
+
+
 __all__ = [
     "anomaly",
     "month_filter",
@@ -15,4 +27,5 @@
     "to_netcdf",
     "zscore",
     "ndvi_from_s2",
+    "show_cube_lexcube",
 ]

src/cubedynamics/viz/__init__.py

@@ -1 +1,8 @@
-"""Subpackage for cubedynamics."""
+"""Visualization helpers exposed at the package level."""
+
+from __future__ import annotations
+
+from .lexcube_viz import show_cube_lexcube
+from .qa_plots import plot_median_over_space
+
+__all__ = ["show_cube_lexcube", "plot_median_over_space"]