Xarray io tests

.github/workflows/ci.yml

@@ -61,7 +61,8 @@ jobs:
       fail-fast: false
       matrix:
         os: [ ubuntu-22.04, macos-13, windows-2022 ]
-        python: [ "311", "312", "313" ]
+        # Disable py313 until numba is available for python 3.13
+        python: [ "311", "312" ]
     env:
       GCC_V: 11
     steps:

.vscode/settings.json

@@ -10,5 +10,6 @@
         "editor.codeActionsOnSave": {
             "source.fixAll": "explicit"
         }
-    }
-}
+    },
+    "files.insertFinalNewline": true,
+}

flopy4/xarray_jinja/filters.py

@@ -0,0 +1,12 @@
+import numpy as np
+import xarray as xr
+
+
+def dask_expand(data: xr.DataArray):
+    for block in data.data.to_delayed():
+        block_data = block.compute()
+        yield block_data
+
+
+def nparray2string(data: np.ndarray):
+    return np.array2string(data, separator=" ")[1:-1]  # remove brackets

flopy4/xarray_jinja/templates/disu_template.disu.jinja

@@ -0,0 +1,5 @@
+BEGIN GRIDDATA
+{% for chunk in data | dask_expand %}
+{{chunk | nparray2string }}
+{% endfor %}
+END GRIDDATA

pyproject.toml

@@ -45,6 +45,7 @@ dependencies = [
     "networkx>=3.4.2,<4",
     "xarray[parallel,io]>=2024.11.0,<2025",
     "scipy>=1.14.1,<2",
+    "xarray-extras>=0.6.0",
 ]
 dynamic = ["version"]
 
@@ -120,13 +121,15 @@ python = "3.11"
 [tool.pixi.feature.py312.dependencies]
 python = "3.12"
 
-[tool.pixi.feature.py313.dependencies]
-python = "3.13"
+# Disable py313 until numba is available for python 3.13
+# [tool.pixi.feature.py313.dependencies]
+# python = "3.13"
 
 [tool.pixi.environments]
 test311 = { features = ["py311", "test"], solve-group = "py311" }
 test312 = { features = ["py312", "test"], solve-group = "py312" }
-test313 = { features = ["py313", "test"], solve-group = "py313" }
+# Disable py313 until numba is available for python 3.13
+# test313 = { features = ["py313", "test"], solve-group = "py313" }
 dev = { features = ["py311", "test", "lint", "build"], solve-group = "py311" }
 
 [tool.pixi.feature.build.tasks]

test/pytest.ini

@@ -3,4 +3,6 @@ addopts = --color=yes
 python_files =
     test_*.py
 markers =
-    slow: tests not completing in a few seconds
+    slow: tests not completing in a few seconds
+    timing: mark test as time profiling test
+    memory: mark test as memory profiling test