Add xwrf/wrf DataArray and Dataset Accessors (#44)

Co-authored-by: jthielen <[email protected]>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: 3 people

.pre-commit-config.yaml

@@ -16,11 +16,12 @@ repos:
     hooks:
       - id: pyupgrade
         args:
-          - '--py37-plus'
+          - '--py38-plus'
 
   - repo: https://github.com/psf/black
     rev: 22.1.0
     hooks:
+      - id: black
       - id: black-jupyter
 
   - repo: https://github.com/keewis/blackdoc

ci/environment-upstream-dev.yml

@@ -9,7 +9,6 @@ dependencies:
   - pre-commit
   - pytest-cov
   - pytest-sugar
-  - wrf-python
   - xarray>=0.18,!=0.20.0,!=0.20.1
   - pooch
   - pip:

ci/environment.yml

@@ -9,7 +9,6 @@ dependencies:
   - pre-commit
   - pytest-cov
   - pytest-sugar
-  - wrf-python
   - xarray>=0.18,!=0.20.0,!=0.20.1
   - pooch
   - pip:

setup.py

@@ -26,7 +26,7 @@
 
 setup(
     name='xwrf',
-    description='A lightweight interface for reading in output from the Weather Research and Forecasting (WRF) model into xarray Dataset.',
+    description='A lightweight interface for working with the Weather Research and Forecasting (WRF) model output in Xarray.',
     long_description=long_description,
     long_description_content_type='text/markdown',
     python_requires='>=3.7',
@@ -36,21 +36,16 @@
     url='https://xwrf.readthedocs.io',
     project_urls={
         'Documentation': 'https://xwrf.readthedocs.io',
-        'Source': 'https://github.com/NCAR/xwrf',
-        'Tracker': 'https://github.com/NCAR/xwrf/issues',
+        'Source': 'https://github.com/ncar-xdev/xwrf',
+        'Tracker': 'https://github.com/ncar-xdev/xwrf/issues',
+        'Discussions/Support': 'https://github.com/ncar-xdev/xwrf/discussions',
     },
     packages=find_packages(exclude=('tests',)),
-    package_dir={'xwrf': 'xwrf'},
     include_package_data=True,
     install_requires=install_requires,
     license='Apache 2.0',
     zip_safe=False,
-    entry_points={
-        'xarray.backends': [
-            'xwrf=xwrf.io_plugin:WRFBackendEntrypoint',
-            'wrf=xwrf.io_plugin:WRFBackendEntrypoint',
-        ]
-    },
+    entry_points={},
     keywords='wrf, xarray',
     use_scm_version={'version_scheme': 'post-release', 'local_scheme': 'dirty-tag'},
 )

tests/__init__.py

@@ -0,0 +1,26 @@
+from __future__ import annotations
+
+import importlib
+import typing
+
+import packaging
+import pytest
+
+
+def importorskip(module_name, minversion: str = None) -> typing.Callable:
+    try:
+        module = importlib.import_module(module_name)
+        has = True
+
+        if minversion and packaging.version.parse(module.__version__) < packaging.version.parse(
+            minversion
+        ):
+            raise ValueError(
+                f'{module_name} version {module.__version__} is less than {minversion}'
+            )
+    except (ImportError, ValueError):
+        has = False
+
+    return pytest.mark.skipif(
+        not has, reason=f'requires {module_name} with minimum version={minversion}'
+    )

tests/sample-data/wrfout_d03_2012-04-22_23_00_00_subset.nc

@@ -0,0 +1,25 @@
+import pandas as pd
+import pytest
+
+import xwrf
+
+from . import importorskip
+
+
+@importorskip('cf_xarray')
+@pytest.mark.parametrize('name', ['lambert_conformal'])
+def test_postprocess(name):
+
+    raw_ds = xwrf.tutorial.open_dataset(name)
+    assert pd.api.types.is_string_dtype(raw_ds.Times.dtype)
+    assert pd.api.types.is_numeric_dtype(raw_ds.Time.dtype)
+    assert 'time' not in raw_ds.cf.coordinates
+    assert raw_ds.cf.standard_names == {}
+    ds = raw_ds.xwrf.postprocess()
+    assert pd.api.types.is_datetime64_dtype(ds.Time.dtype)
+    assert 'time' in ds.cf.coordinates
+    standard_names = ds.cf.standard_names
+
+    assert standard_names['time'] == ['Time']
+    assert standard_names['humidity_mixing_ratio'] == ['Q2', 'QVAPOR']
+    assert standard_names['air_temperature'] == ['T2']