feat: add tests for Python package; bump min Python version; run in CI

.github/workflows/pytest.yml

@@ -0,0 +1,30 @@
+name: Python tests
+
+on:
+  push:
+    branches: ["main"]
+  pull_request:
+
+jobs:
+  test:
+    name: Python tests
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Install uv
+        uses: astral-sh/setup-uv@v3
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version-file: crates/pyaugurs/pyproject.toml
+      - name: Build wheel
+        uses: PyO3/maturin-action@v1
+        with:
+          command: develop
+          target: x86_64
+          args: --uv --manifest-path crates/pyaugurs/Cargo.toml
+          sccache: 'true'
+      - name: Install deps
+        run: uv sync --all-extras --dev
+      - name: Run tests
+        run: uv run pytest crates/pyaugurs/tests

crates/pyaugurs/.gitignore

@@ -1 +1,2 @@
 .python-version
+__pycache__

crates/pyaugurs/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "maturin"
 [project]
 name = "augurs"
 dependencies = ["numpy"]
-requires-python = ">=3.7"
+requires-python = ">=3.9"
 classifiers = [
     "Programming Language :: Rust",
     "Programming Language :: Python :: Implementation :: CPython",
@@ -14,3 +14,8 @@ classifiers = [
 
 [tool.maturin]
 features = ["pyo3/extension-module"]
+
+[dependency-groups]
+dev = [
+    "pytest>=8.3.3",
+]

crates/pyaugurs/tests/test_dtw.py

@@ -0,0 +1,48 @@
+import numpy as np
+import pytest
+
+from augurs import Dtw
+
+
+class TestDtw:
+    @pytest.mark.parametrize(
+        "opts, input, expected",
+        [
+            ({}, [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]], 5.0990195135926845),
+            ({"window": 2}, [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]], 5.0990195135926845),
+            ({"distance_fn": "euclidean"}, [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]], 5.0990195135926845),
+            ({"distance_fn": "manhattan"}, [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]], 9),
+        ],
+    )
+    def test_distance(self, opts, input, expected):
+        d = Dtw(**opts)
+        arrays = [np.array(x) for x in input]
+        np.testing.assert_allclose(d.distance(*arrays), expected)
+
+    @pytest.mark.parametrize(
+        "opts, input, expected",
+        [
+            (
+                {},
+                [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0], [6.0, 7.0, 8.0]],
+                [
+                    [0.0, 5.0990195135926845, 10.392304845413264],
+                    [5.0990195135926845, 0.0, 5.0990195135926845],
+                    [10.392304845413264, 5.0990195135926845, 0.0],
+                ]
+            ),
+            (
+                {"distance_fn": "manhattan"},
+                [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0], [6.0, 7.0, 8.0]],
+                [
+                    [0.0, 9.0, 18.0],
+                    [9.0, 0.0, 9.0],
+                    [18.0, 9.0, 0.0],
+                ]
+            ),
+        ],
+    )
+    def test_distance_matrix(self, opts, input, expected):
+        d = Dtw(**opts)
+        arrays = [np.array(x) for x in input]
+        np.testing.assert_allclose(d.distance_matrix(arrays).to_numpy(), expected)