metrust 0.4.2__tar.gz → 0.4.3__tar.gz

{metrust-0.4.2 → metrust-0.4.3}/.github/workflows/ci.yml

@@ -5,23 +5,31 @@ on:
     branches: [master, main]
   pull_request:
 
+env:
+  METPY_VERSION: "1.7.1"
+
 jobs:
-  test:
+  smoke:
+    name: Smoke (${{ matrix.os }}, py${{ matrix.python-version }})
     runs-on: ${{ matrix.os }}
     strategy:
+      fail-fast: false
       matrix:
-        os: [ubuntu-latest, windows-latest, macos-latest]
-        python-version: ["3.10", "3.12"]
+        os: [ubuntu-latest, windows-latest, macos-14]
+        python-version: ["3.12"]
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
 
-      - uses: actions/setup-python@v5
+      - uses: actions/setup-python@v6
         with:
           python-version: ${{ matrix.python-version }}
 
       - name: Install Rust
         uses: dtolnay/rust-toolchain@stable
 
+      - name: Rust cache
+        uses: Swatinem/rust-cache@v2
+
       - name: Install package and test deps
         run: |
           python -m pip install --upgrade pip
@@ -33,11 +41,47 @@ jobs:
       - name: Run Python compatibility tests
         run: python -m pytest tests/test_python_compat.py -q
 
+  differential:
+    name: Differential (${{ matrix.mode }}, py${{ matrix.python-version }})
+    runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.10", "3.11", "3.12", "3.13"]
+        mode: ["metrust-only", "metpy", "metpy-xarray"]
+    steps:
+      - uses: actions/checkout@v5
+
+      - uses: actions/setup-python@v6
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install Rust
+        uses: dtolnay/rust-toolchain@stable
+
+      - name: Rust cache
+        uses: Swatinem/rust-cache@v2
+
+      - name: Install test dependencies
+        shell: bash
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install -e "." pytest numpy pint pandas
+          if [[ "${{ matrix.mode }}" != "metrust-only" ]]; then
+            python -m pip install "MetPy==${METPY_VERSION}"
+          fi
+          if [[ "${{ matrix.mode }}" == "metpy-xarray" ]]; then
+            python -m pip install xarray
+          fi
+
+      - name: Run differential test mode
+        run: python tests/run_ci_mode.py ${{ matrix.mode }}
+
   bench-rust:
     name: Rust Benchmarks
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
 
       - name: Install Rust
         uses: dtolnay/rust-toolchain@stable
@@ -49,7 +93,7 @@ jobs:
         run: cargo bench --package metrust --bench calc_bench
 
       - name: Upload criterion artifacts
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v7
         if: always()
         with:
           name: criterion-report
@@ -60,9 +104,9 @@ jobs:
     name: Python Benchmarks
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
 
-      - uses: actions/setup-python@v5
+      - uses: actions/setup-python@v6
         with:
           python-version: "3.12"
 
@@ -75,13 +119,13 @@ jobs:
       - name: Install metrust and MetPy
         run: |
           python -m pip install --upgrade pip
-          python -m pip install -e "." numpy pint metpy
+          python -m pip install -e "." numpy pint "MetPy==${METPY_VERSION}"
 
       - name: Run three-tier benchmark
         run: python benches/bench_python.py --json
 
       - name: Upload benchmark results
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v7
         if: always()
         with:
           name: python-bench-results

{metrust-0.4.2 → metrust-0.4.3}/.github/workflows/docs.yml

@@ -18,13 +18,13 @@ jobs:
   build:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: actions/checkout@v5
+      - uses: actions/setup-python@v6
         with:
           python-version: "3.12"
       - run: pip install mkdocs-material
       - run: mkdocs build
-      - uses: actions/upload-pages-artifact@v3
+      - uses: actions/upload-pages-artifact@v4
         with:
           path: site
 
@@ -36,4 +36,4 @@ jobs:
       url: ${{ steps.deployment.outputs.page_url }}
     steps:
       - id: deployment
-        uses: actions/deploy-pages@v4
+        uses: actions/deploy-pages@v5

{metrust-0.4.2 → metrust-0.4.3}/.github/workflows/release.yml

@@ -22,8 +22,8 @@ jobs:
           - runner: ubuntu-22.04
             target: aarch64
     steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: actions/checkout@v5
+      - uses: actions/setup-python@v6
         with:
           python-version: '3.12'
       - name: Build wheels
@@ -33,7 +33,7 @@ jobs:
           args: --release --out dist -i python3.10 python3.11 python3.12 python3.13
           manylinux: auto
       - name: Upload wheels
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v7
         with:
           name: wheels-linux-${{ matrix.platform.target }}
           path: dist
@@ -50,8 +50,8 @@ jobs:
           - runner: macos-14
             target: aarch64
     steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: actions/checkout@v5
+      - uses: actions/setup-python@v6
         with:
           python-version: '3.12'
       - name: Build wheels
@@ -60,7 +60,7 @@ jobs:
           target: ${{ matrix.platform.target }}
           args: --release --out dist
       - name: Upload wheels
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v7
         with:
           name: wheels-macos-${{ matrix.platform.target }}
           path: dist
@@ -72,8 +72,8 @@ jobs:
       matrix:
         target: [x64]
     steps:
-      - uses: actions/checkout@v4
-      - uses: actions/setup-python@v5
+      - uses: actions/checkout@v5
+      - uses: actions/setup-python@v6
         with:
           python-version: '3.12'
           architecture: ${{ matrix.target }}
@@ -82,7 +82,7 @@ jobs:
         with:
           args: --release --out dist
       - name: Upload wheels
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v7
         with:
           name: wheels-windows-${{ matrix.target }}
           path: dist
@@ -91,14 +91,14 @@ jobs:
   sdist:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - name: Build sdist
         uses: PyO3/maturin-action@v1
         with:
           command: sdist
           args: --out dist
       - name: Upload sdist
-        uses: actions/upload-artifact@v4
+        uses: actions/upload-artifact@v7
         with:
           name: wheels-sdist
           path: dist
@@ -115,7 +115,7 @@ jobs:
       id-token: write  # trusted publishing
     steps:
       - name: Download all wheels
-        uses: actions/download-artifact@v4
+        uses: actions/download-artifact@v8
         with:
           pattern: wheels-*
           merge-multiple: true

{metrust-0.4.2 → metrust-0.4.3}/Cargo.lock

@@ -802,7 +802,7 @@ dependencies = [
 
 [[package]]
 name = "metrust-py"
-version = "0.4.2"
+version = "0.4.3"
 dependencies = [
  "metrust",
  "numpy",

{metrust-0.4.2 → metrust-0.4.3}/Cargo.toml

@@ -8,7 +8,7 @@ license = "MIT"
 
 [package]
 name = "metrust-py"
-version = "0.4.2"
+version = "0.4.3"
 edition.workspace = true
 description = "Python bindings for metrust, a Rust-powered MetPy-compatible calculation layer"
 readme = "README.md"

{metrust-0.4.2 → metrust-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: metrust
-Version: 0.4.2
+Version: 0.4.3
 Classifier: Development Status :: 4 - Beta
 Classifier: Intended Audience :: Science/Research
 Classifier: License :: OSI Approved :: MIT License

{metrust-0.4.2 → metrust-0.4.3}/benches/bench_python.py

@@ -127,6 +127,7 @@ _raw_calc = None   # metrust._metrust.calc  (T1)
 _mr_calc = None    # metrust.calc            (T2 -- Pint wrapper)
 _mp_calc = None    # metpy.calc              (T3)
 _mp_units = None   # metpy.units.units
+_mr_units = None   # metrust.units.units
 
 
 def _import_raw():
@@ -145,6 +146,14 @@ def _import_metrust():
     return _mr_calc
 
 
+def _import_metrust_units():
+    global _mr_units
+    if _mr_units is None:
+        from metrust.units import units as mu
+        _mr_units = mu
+    return _mr_units
+
+
 def _import_metpy():
     global _mp_calc, _mp_units
     if _mp_calc is None:
@@ -345,8 +354,16 @@ def _bench_wind(tiers):
             lambda: rc.storm_relative_helicity(u_p, v_p, z_p, 1000.0, 10.0, 5.0)))
     if 2 in tiers:
         mc = _import_metrust()
+        mu = _import_metrust_units()
         results.append(run_bench(name, 2, "T2 metrust+Pint",
-            lambda: mc.storm_relative_helicity(u_p, v_p, z_p, 1000.0, 10.0, 5.0)))
+            lambda: mc.storm_relative_helicity(
+                z_p * mu.meter,
+                u_p * mu("m/s"),
+                v_p * mu("m/s"),
+                1000 * mu.meter,
+                storm_u=10 * mu("m/s"),
+                storm_v=5 * mu("m/s"),
+            )))
     if 3 in tiers:
         try:
             mpc, mu = _import_metpy()

{metrust-0.4.2 → metrust-0.4.3}/docs/compatibility.md

@@ -17,13 +17,23 @@ calculation surface.
 | Tier | Description |
 |------|-------------|
 | **Native Rust** | Function is implemented entirely in Rust. The Python wrapper strips Pint units, calls the Rust function, and re-attaches units. No MetPy dependency. |
+| **Native + Optional MetPy Delegation** | Function uses the Rust path by default, but can delegate specific parity-sensitive call forms to MetPy when MetPy is installed. |
 | **Native + Rust Array Binding** | Same as Native Rust, but the function also exposes a vectorized `_array` variant in the Rust extension so that array inputs are processed in a single FFI call instead of element-wise Python loops. 28 functions have this optimization. |
 | **MetPy Shim** | Module forwards attribute lookups to the corresponding MetPy module via lazy import. MetPy must be installed separately. Used for `plots`, `xarray`, and `io.Level2File`. |
 
-When MetPy is not installed, all `metrust.calc` functions still work (they
-never import MetPy).  Only the shimmed surfaces (`metrust.plots`,
+When MetPy is not installed, all `metrust.calc` functions still work through
+their local metrust implementations. Only the shimmed surfaces (`metrust.plots`,
 `metrust.xarray`, `metrust.io.Level2File`) require MetPy at runtime.
 
+Current differential-CI target:
+
+- `MetPy 1.7.1`
+- Python `3.10` through `3.13`
+- dependency modes: `metrust-only`, `metrust + MetPy`, and `metrust + MetPy + xarray`
+
+For the exact list of parity-sensitive `metrust.calc` paths that may delegate
+to MetPy when available, see the [delegation ledger](delegation-ledger.md).
+
 ---
 
 ## 2. metrust.calc Function Matrix
@@ -69,15 +79,15 @@ never import MetPy).  Only the shimmed surfaces (`metrust.plots`,
 | `mixing_ratio_from_specific_humidity` | Native + Array | `metpy.calc.mixing_ratio_from_specific_humidity` | |
 | `specific_humidity_from_mixing_ratio` | Native + Array | `metpy.calc.specific_humidity_from_mixing_ratio` | |
 | `lcl` | Native | `metpy.calc.lcl` | Scalar only. Returns (p_lcl, t_lcl). Rust also exposes `lcl_pressure` and `lcl_pressure_array` internally. See section 5 for approximation differences. |
-| `lfc` | Native (profile) | `metpy.calc.lfc` | Operates on full sounding profiles. |
-| `el` | Native (profile) | `metpy.calc.el` | Operates on full sounding profiles. |
-| `cape_cin` | Native (profile) | `metpy.calc.cape_cin` | Extended signature: supports parcel_type, ml_depth, mu_depth, top_m, and returns (CAPE, CIN, LCL height, LFC height). |
+| `lfc` | Native (profile) | `metpy.calc.lfc` | Native by default; optionally delegates to MetPy for quantity-profile parity-sensitive cases. |
+| `el` | Native (profile) | `metpy.calc.el` | Native by default; optionally delegates to MetPy for quantity-profile parity-sensitive cases. |
+| `cape_cin` | Native (profile) | `metpy.calc.cape_cin` | Native by default; MetPy parcel-profile form can optionally delegate for exact parity. Extended signature supports parcel_type, ml_depth, mu_depth, top_m, and returns (CAPE, CIN, LCL height, LFC height). |
 | `surface_based_cape_cin` | Native (profile) | `metpy.calc.surface_based_cape_cin` | |
 | `mixed_layer_cape_cin` | Native (profile) | `metpy.calc.mixed_layer_cape_cin` | |
 | `most_unstable_cape_cin` | Native (profile) | `metpy.calc.most_unstable_cape_cin` | |
-| `downdraft_cape` | Native (profile) | `metpy.calc.downdraft_cape` | |
+| `downdraft_cape` | Native (profile) | `metpy.calc.downdraft_cape` | Native by default; optionally delegates on quantity-profile calls when MetPy is available. |
 | `parcel_profile` | Native (profile) | `metpy.calc.parcel_profile` | |
-| `parcel_profile_with_lcl` | Native (profile) | `metpy.calc.parcel_profile_with_lcl` | |
+| `parcel_profile_with_lcl` | Native (profile) | `metpy.calc.parcel_profile_with_lcl` | Native by default; optionally delegates on quantity-profile calls when MetPy is available. |
 | `dry_lapse` | Native (profile) | `metpy.calc.dry_lapse` | |
 | `moist_lapse` | Native (profile) | `metpy.calc.moist_lapse` | |
 | `ccl` | Native (profile) | `metpy.calc.ccl` | Convective Condensation Level. |
@@ -178,7 +188,7 @@ major moisture conversions:
 | `frontogenesis` | Native (grid) | `metpy.calc.frontogenesis` | Petterssen frontogenesis. |
 | `geostrophic_wind` | Native (grid) | `metpy.calc.geostrophic_wind` | |
 | `ageostrophic_wind` | Native (grid) | `metpy.calc.ageostrophic_wind` | |
-| `potential_vorticity_baroclinic` | Native (grid) | `metpy.calc.potential_vorticity_baroclinic` | Ertel PV. |
+| `potential_vorticity_baroclinic` | Native (grid) | `metpy.calc.potential_vorticity_baroclinic` | Ertel PV. Native by default with optional MetPy delegation on parity-sensitive quantity/DataArray calls. |
 | `potential_vorticity_barotropic` | Native (grid) | `metpy.calc.potential_vorticity_barotropic` | |
 | `normal_component` | Native (profile) | `metpy.calc.normal_component` | Cross-section decomposition. |
 | `tangential_component` | Native (profile) | `metpy.calc.tangential_component` | Cross-section decomposition. |
@@ -196,7 +206,7 @@ major moisture conversions:
 | `stretching_deformation` | Native (grid) | `metpy.calc.stretching_deformation` | |
 | `total_deformation` | Native (grid) | `metpy.calc.total_deformation` | |
 | `geospatial_gradient` | Native (grid) | (no direct MetPy equivalent) | Gradient on lat/lon grids with spherical corrections. |
-| `geospatial_laplacian` | Native (grid) | (no direct MetPy equivalent) | Laplacian on lat/lon grids with spherical corrections. |
+| `geospatial_laplacian` | Native (grid) | (no direct MetPy equivalent) | Laplacian on lat/lon grids with spherical corrections. Native by default with optional MetPy delegation on parity-sensitive quantity/DataArray calls. |
 | `advection_3d` | Native (grid) | (no direct MetPy equivalent) | 3-D advection including vertical term. |
 
 ### 2.5 Severe Weather Parameters

metrust-0.4.3/docs/delegation-ledger.md

@@ -0,0 +1,29 @@
+# Optional MetPy Delegation Ledger
+
+`metrust.calc` is Rust-first by default, but a small parity-sensitive subset can
+delegate to `metpy.calc` when MetPy is installed. This page is the explicit
+ledger for that behavior.
+
+Current compatibility target:
+
+- MetPy `1.7.1`
+- Python `3.10` to `3.13`
+- CI modes: `metrust-only`, `metrust + MetPy`, `metrust + MetPy + xarray`
+
+## Calc Delegations
+
+| Function | Delegates When | Local Fallback |
+|---|---|---|
+| `lfc` | Quantity profile inputs, especially MetPy's more complex `which=` handling | metrust native profile intersection solver |
+| `el` | Quantity profile inputs, especially MetPy's more complex `which=` handling | metrust native profile intersection solver |
+| `cape_cin` | MetPy parcel-profile form where the 4th positional argument is temperature-like | metrust native CAPE/CIN integration |
+| `downdraft_cape` | Quantity profile inputs when MetPy is available | metrust native DCAPE layer selection and integration |
+| `parcel_profile_with_lcl` | Quantity profile inputs in the MetPy profile-returning form | metrust native interpolation and parcel-trace construction |
+| `potential_vorticity_baroclinic` | Quantity/DataArray inputs in the MetPy-style baroclinic-PV form | metrust native PV computation with local dx/dy and latitude handling |
+| `geospatial_laplacian` | Quantity/DataArray geospatial-laplacian inputs when MetPy is available | metrust native geospatial gradient and derivative path |
+
+## CI Guarantees
+
+- `tests/test_delegation_ledger.py` verifies the ledger stays in sync with the code.
+- The same test file verifies delegated functions still run with MetPy blocked.
+- Differential CI runs the relevant suites against pinned MetPy `1.7.1`.

{metrust-0.4.2 → metrust-0.4.3}/mkdocs.yml

@@ -108,3 +108,4 @@ nav:
     - I/O Formats: methodology/io-formats.md
   - Performance: performance.md
   - Compatibility: compatibility.md
+  - Delegation Ledger: delegation-ledger.md

{metrust-0.4.2 → metrust-0.4.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "metrust"
-version = "0.4.2"
+version = "0.4.3"
 description = "Rust-powered MetPy-compatible calculation layer with optional GPU acceleration"
 requires-python = ">=3.9"
 license = {text = "MIT"}

{metrust-0.4.2 → metrust-0.4.3}/python/metrust/calc/__init__.py

@@ -43,6 +43,49 @@ class InvalidSoundingError(Exception):
 _BACKEND = "cpu"
 _GPU_CALC = None
 
+METPY_COMPATIBILITY_TARGET = {
+    "metpy": "1.7.1",
+    "python": ("3.10", "3.11", "3.12", "3.13"),
+}
+
+METPY_OPTIONAL_CALC_DELEGATIONS = (
+    {
+        "function": "lfc",
+        "trigger": "Quantity profile inputs, with MetPy-preferred handling for complex which selections.",
+        "fallback": "Native profile intersection solver in metrust.",
+    },
+    {
+        "function": "el",
+        "trigger": "Quantity profile inputs, with MetPy-preferred handling for complex which selections.",
+        "fallback": "Native profile intersection solver in metrust.",
+    },
+    {
+        "function": "cape_cin",
+        "trigger": "MetPy parcel-profile form where the 4th positional argument is temperature-like.",
+        "fallback": "Native CAPE/CIN integration path in metrust.",
+    },
+    {
+        "function": "downdraft_cape",
+        "trigger": "Quantity profile inputs when MetPy is available.",
+        "fallback": "Native metrust DCAPE layer-selection and integration path.",
+    },
+    {
+        "function": "parcel_profile_with_lcl",
+        "trigger": "Quantity profile inputs in the profile-returning MetPy form.",
+        "fallback": "Native metrust profile interpolation and parcel trace path.",
+    },
+    {
+        "function": "potential_vorticity_baroclinic",
+        "trigger": "Quantity/DataArray inputs in the MetPy-style baroclinic PV calling form.",
+        "fallback": "Native metrust PV computation with local dx/dy and latitude handling.",
+    },
+    {
+        "function": "geospatial_laplacian",
+        "trigger": "Quantity/DataArray geospatial laplacian inputs when MetPy is available.",
+        "fallback": "Native metrust geospatial gradient and derivative path.",
+    },
+)
+
 
 def _normalize_backend_name(backend):
     name = str(backend).strip().lower()
@@ -5386,6 +5429,12 @@ def geospatial_laplacian(f, *args, dx=None, dy=None, x_dim=-1, y_dim=-2,
         dx, dy = _resolve_dx_dy(f, dx=dx, dy=dy, latitude=latitude, longitude=longitude)
     if dx is None or dy is None:
         raise TypeError("geospatial_laplacian requires dx/dy or inferable latitude/longitude coordinates")
+    if hasattr(f, "magnitude"):
+        f_arr = np.asarray(f.magnitude, dtype=np.float64)
+    elif _is_dataarray_like(f):
+        f_arr = np.asarray(f.values, dtype=np.float64)
+    else:
+        f_arr = np.asarray(f, dtype=np.float64)
     if parallel_scale is None and meridional_scale is None:
         ps, ms = _get_scale_factors(f)
         parallel_scale = ps if ps is not None else parallel_scale
@@ -5403,13 +5452,13 @@ def geospatial_laplacian(f, *args, dx=None, dy=None, x_dim=-1, y_dim=-2,
             longitude=longitude,
             crs=crs,
         )
-        x_axis = x_dim % np.ndim(np.asarray(_strip(f) if hasattr(f, "to") else f))
-        y_axis = y_dim % np.ndim(np.asarray(_strip(f) if hasattr(f, "to") else f))
+        x_axis = x_dim % f_arr.ndim
+        y_axis = y_dim % f_arr.ndim
         term_x = first_derivative(grad_u, delta=dx, axis=x_axis)
         term_y = first_derivative(grad_v, delta=dy, axis=y_axis)
         return term_x + term_y
-    x_axis = x_dim % np.ndim(np.asarray(_strip(f) if hasattr(f, "to") else f))
-    y_axis = y_dim % np.ndim(np.asarray(_strip(f) if hasattr(f, "to") else f))
+    x_axis = x_dim % f_arr.ndim
+    y_axis = y_dim % f_arr.ndim
     return second_derivative(f, delta=dx, axis=x_axis) + second_derivative(f, delta=dy, axis=y_axis)
 
 
@@ -6164,12 +6213,15 @@ def smooth_window(scalar_grid, window, passes=1, normalize_weights=True):
         )
 
     data_units = getattr(scalar_grid, "units", None)
+    unit_str = str(data_units) if data_units is not None else None
+    if unit_str is None:
+        unit_str = getattr(getattr(scalar_grid, "attrs", None), "get", lambda *_: None)("units")
     data = np.array(getattr(scalar_grid, "magnitude", scalar_grid))
     for _ in range(int(passes)):
         data[inner_full_index] = sum(
             weights[index] * data[offset_full_index(index)] for index in weight_indexes
         )
-    return data * data_units if data_units is not None else data
+    return _wrap_result_like(scalar_grid, data, unit_str=unit_str)
 
 
 def _gradient_axes_and_positions(f, axes, coordinates, deltas):
@@ -8222,19 +8274,12 @@ class _MetPyCalcSignatureHook(importlib.abc.MetaPathFinder, importlib.abc.Loader
         if fullname != "metpy.calc":
             return None
 
-        for finder in sys.meta_path:
-            if finder is self:
-                continue
-            find_spec = getattr(finder, "find_spec", None)
-            if find_spec is None:
-                continue
-            spec = find_spec(fullname, path, target)
-            if spec is None:
-                continue
-            self._wrapped_loader = spec.loader
-            spec.loader = self
-            return spec
-        return None
+        spec = importlib.machinery.PathFinder.find_spec(fullname, path)
+        if spec is None:
+            return None
+        self._wrapped_loader = spec.loader
+        spec.loader = self
+        return spec
 
     def create_module(self, spec):
         if self._wrapped_loader is not None and hasattr(self._wrapped_loader, "create_module"):

metrust-0.4.3/tests/run_ci_mode.py

@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+import subprocess
+import sys
+
+
+MODE_TESTS = {
+    "metrust-only": [
+        "tests/test_python_compat.py",
+        "tests/test_delegation_ledger.py",
+    ],
+    "metpy": [
+        "tests/test_signature_parity.py",
+        "tests/test_runtime_parity.py",
+        "tests/test_runtime_parity_sounding_core.py",
+        "tests/test_runtime_parity_thermo_layers.py",
+        "tests/test_runtime_parity_wind_profiles.py",
+        "tests/test_runtime_parity_remaining.py",
+        "tests/test_delegation_ledger.py",
+        "tests/test_cookbook_replays.py::test_cookbook_sounding_workflow_replay",
+    ],
+    "metpy-xarray": [
+        "tests/test_metpy_dropin_compat.py",
+        "tests/test_runtime_parity_interp_dataset.py",
+        "tests/test_runtime_parity_kinematics_extra.py",
+        "tests/test_runtime_parity_utils_misc.py",
+        "tests/test_delegation_ledger.py",
+        "tests/test_cookbook_replays.py::test_cookbook_grid_workflow_replay",
+        "tests/test_cookbook_replays.py::test_cookbook_xarray_workflow_replay",
+    ],
+}
+
+
+def main() -> int:
+    if len(sys.argv) != 2 or sys.argv[1] not in MODE_TESTS:
+        valid = ", ".join(sorted(MODE_TESTS))
+        raise SystemExit(f"usage: {sys.argv[0]} <mode>; valid modes: {valid}")
+
+    mode = sys.argv[1]
+    cmd = [sys.executable, "-m", "pytest", "-q", *MODE_TESTS[mode]]
+    return subprocess.call(cmd)
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())