add examples head snapshot checks

Author: mjreno

pixi.lock

@@ -48,6 +48,7 @@ dependencies = [
     "xarray[parallel,io]>=2025.10.1,<2026",
     "scipy>=1.16.2,<2",
     "sparse>=0.17.0,<0.18",
+    "syrupy<5",
     "dask[array]>=2025.10.0,<2026",
     "pyproj>=3.7.2,<4",
     "netcdf4>=1.7.3,<2",

pyproject.toml

@@ -6,7 +6,7 @@
 from modflow_devtools.download import download_and_unzip
 from modflow_devtools.misc import is_in_ci
 
-pytest_plugins = ["modflow_devtools.fixtures"]
+pytest_plugins = ["modflow_devtools.fixtures", "modflow_devtools.snapshots"]
 
 PROJ_ROOT_PATH = Path(__file__).parents[1]
 DOCS_PATH = PROJ_ROOT_PATH / "docs"

test/__snapshots__/test_examples/test_scripts[quickstart.py].npy

@@ -19,8 +19,21 @@ def pytest_generate_tests(metafunc):
         metafunc.parametrize("example_script", scripts.values(), ids=scripts.keys())
 
 
+def check(example_script, snapshot):
+    from pathlib import Path
+
+    from flopy.utils import HeadFile
+
+    check_path = Path(f"{example_script.parent}/{example_script.stem}")
+    for f in check_path.rglob("*.hds"):
+        hds = HeadFile(f, precision="double")
+        assert snapshot == hds.get_data()
+
+
+@pytest.mark.snapshot
 @pytest.mark.slow
-def test_scripts(example_script):
+def test_scripts(example_script, array_snapshot):
     args = [sys.executable, example_script]
     stdout, stderr, retcode = run_cmd(*args, verbose=True)
     assert not retcode, stdout + stderr
+    check(example_script, array_snapshot)