add pdm run e2e:test:docker

this commit adds a new `Dockerfile.e2e-test` just for running the
end-to-end tests inside a Docker container. This is useful if you don't
have Nix in your local environment.

Author: diamondburned

.gitignore

@@ -25,6 +25,7 @@ share/python-wheels/
 .installed.cfg
 *.egg
 MANIFEST
+/result
 
 # PyInstaller
 #  Usually these files are written by a python script from a template
@@ -174,4 +175,4 @@ cython_debug/
 .pypirc
 
 # Test Reports (directories)
-reports-*/
+reports-*/

Dockerfile.e2e-test

@@ -0,0 +1,21 @@
+FROM docker.nix-community.org/nixpkgs/nix-flakes AS builder
+
+# Don't stay in the root directory for this.
+# https://github.com/NixOS/nix/issues/11217
+WORKDIR /workspace
+
+# Copy just enough for Nix pdm install without bringing in the entire project.
+# This ensures we don't run the nix develop too often.
+COPY flake.nix flake.lock pyproject.toml pdm.lock .
+
+# Build Nix shell and install all dev PDM dependencies.
+RUN nix develop path:///workspace -c pdm sync -d
+
+# Copy the actual module and E2E tests.
+# This is what will be edited during development, so we run it last.
+COPY inference_perf e2e .
+
+# Default command is to run the E2E tests, but you may override this with any
+# pdm or bash command (e.g., docker run -it <TAG> -- bash).
+ENTRYPOINT ["nix", "develop", "path:///workspace", "-c"]
+CMD ["pdm", "run", "test:e2e"]

flake.nix

@@ -60,6 +60,13 @@
                 ];
 
               venvDir = "venv";
+
+              shellHook = ''
+                # This isn't needed anywhere except for inside the Dockerfile environment, since we
+                # try to do `pdm install` without actually having any actual code present for better
+                # image layering.
+                export PYTHONPATH="$PYTHONPATH:$PWD"
+              '';
             };
 
             packages = rec {

pyproject.toml

@@ -65,11 +65,15 @@ distribution = true
 [tool.pdm.scripts]
 format = "ruff format"
 lint = "ruff check"
-test = "pytest tests"
-"test:e2e" = "pytest e2e"
 type-check = "mypy --strict ./inference_perf ./tests"
 validate = {composite = ["format", "lint", "type-check"]}
 
+"test" = "pytest tests"
+"test:e2e" = "pytest e2e"
+"test:e2e:docker" = "docker:e2e-test:run"
+"docker:e2e-test:build".cmd = "docker buildx b -f Dockerfile.e2e-test {args:-t inference-perf-e2e-test} ."
+"docker:e2e-test:run".shell = "pdm run docker:e2e-test:build --iidfile result && docker run --rm -it -v $PWD:/workspace $(< result)"
+
 [tool.ruff]
 # The GitHub editor is 127 chars wide
 line-length = 127