Merge branch 'main' into poc/unified-chunk-grid

Author: maxrjones

.github/workflows/test.yml

@@ -155,14 +155,36 @@ jobs:
       run: |
         hatch run doctest:test
 
+  benchmarks:
+    name: Benchmark smoke test
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
+      with:
+        fetch-depth: 0
+        persist-credentials: false
+    - name: Set up Python
+      uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405 # v6.2.0
+      with:
+        python-version: '3.13'
+        cache: 'pip'
+    - name: Install Hatch
+      uses: pypa/hatch@257e27e51a6a5616ed08a39a408a21c35c9931bc
+      with:
+        version: '1.16.5'
+    - name: Run Benchmarks
+      run: |
+        hatch env run --env "test.py3.13-minimal" run-benchmark
+
   test-complete:
     name: Test complete
 
     needs:
       [
         test,
         test-upstream-and-min-deps,
-        doctests
+        doctests,
+        benchmarks
       ]
     if: always()
     runs-on: ubuntu-latest

TEAM.md

@@ -10,6 +10,7 @@
 - @dstansby (David Stansby)
 - @dcherian (Deepak Cherian)
 - @TomAugspurger (Tom Augspurger)
+- @maxrjones (Max Jones)
 
 ## Emeritus core-developers
 - @alimanfoo (Alistair Miles)

pyproject.toml

@@ -173,11 +173,11 @@ matrix.deps.dependency-groups = [
 
 [tool.hatch.envs.test.scripts]
 run-coverage = [
-    "coverage run --source=src -m pytest --junitxml=junit.xml -o junit_family=legacy {args:}",
+    "coverage run --source=src -m pytest --ignore tests/benchmarks --junitxml=junit.xml -o junit_family=legacy {args:}",
     "coverage xml",
 ]
 run-coverage-html = [
-    "coverage run --source=src -m pytest {args:}",
+    "coverage run --source=src -m pytest --ignore tests/benchmarks {args:}",
     "coverage html",
 ]
 run = "pytest --ignore tests/benchmarks"

tests/test_codecs/test_codecs.py

@@ -219,6 +219,7 @@ def test_morton_exact_order() -> None:
         (1, 1),
         (5, 1, 3),
         (1, 4, 1, 2),
+        (5, 5, 5),  # triggers argsort strategy (n_z/n_total > 4)
     ],
 )
 def test_morton_is_permutation(shape: tuple[int, ...]) -> None: