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

Dockerfile.e2e-test

@@ -0,0 +1,24 @@
+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 -c pdm install --dev --frozen-lockfile --no-editable
+
+# Copy the rest of the source code.
+COPY . .
+
+# Ensure PYTHONPATH contains our current WORKDIR so that the main module can be
+# found.
+ENV PYTHONPATH="$PYTHONPATH:/workspace"
+
+# 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", "-c"]
+CMD ["pdm", "run", "test:e2e"]

flake.nix

@@ -64,6 +64,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,8 +65,14 @@ distribution = true
 [tool.pdm.scripts]
 format = "ruff format"
 lint = "ruff check"
-test = "pytest tests"
+
+"test" = "pytest tests"
 "test:e2e" = "pytest e2e"
+"test:e2e:docker".composite = ["docker:e2e-test:build", "docker:e2e-test:run"]
+
+"docker:e2e-test:build" = "docker build -f Dockerfile.e2e-test -t inference-perf-e2e-test ."
+"docker:e2e-test:run" = "docker run --rm -it inference-perf-e2e-test"
+
 type-check = "mypy --strict ./inference_perf ./tests"
 validate = {composite = ["format", "lint", "type-check"]}