docs: update README and Dockerfiles to include pytest as a build dependency

Author: Artlesbol

README.md

@@ -178,10 +178,10 @@ package in this repository; CMake builds and installs the native library
 separately.
 
 The build environment must provide CMake, Ninja, a CUDA toolkit, SQLite3,
-Python 3.10+ development files, PyTorch, and pybind11. Runtime JIT generation
-and Python tests additionally require a preconfigured Triton/TLE-enabled
-Python environment and `pytest`. Install the codegen package into the Python
-environment that will execute JIT generation:
+Python 3.10+ development files, PyTorch, pybind11, and `pytest`. Runtime JIT
+generation and Python tests additionally require a preconfigured
+Triton/TLE-enabled Python environment. Install the codegen package into the
+Python environment that will execute JIT generation:
 
 ```sh
 python3 -m pip install .
@@ -196,8 +196,8 @@ cmake --install build --prefix /path/to/flagfft-install
 
 `docker/Dockerfile` builds a base Ubuntu 24.04 environment with Python 3.12,
 PyTorch CUDA wheels, CUDA toolkit 13.2, FlagTree, CMake, Ninja, pybind11, and
-other native build dependencies. The image only contains the environment; it
-does not compile this repository during `docker build`.
+`pytest`, plus other native build dependencies. The image only contains the
+environment; it does not compile this repository during `docker build`.
 
 ```sh
 docker build -f docker/Dockerfile -t flagfft-ubuntu2404:latest .

docker/Dockerfile

@@ -62,8 +62,8 @@ RUN pip uninstall -y triton || true && \
     RES="--index-url=https://resource.flagos.net/repository/flagos-pypi-hosted/simple" && \
     pip install --no-warn-conflicts flagtree===0.5.0 ${RES}
 
-# flagFFT build dependencies
-RUN pip install --no-warn-conflicts nanobind ninja cmake pybind11 && \
+# flagFFT build and test dependencies
+RUN pip install --no-warn-conflicts nanobind ninja cmake pybind11 pytest && \
     pip install --no-warn-conflicts -U pip "setuptools<82" wheel scikit-build-core
 
 WORKDIR /workspace

docker/Dockerfile.ci

@@ -0,0 +1,73 @@
+FROM ubuntu:24.04
+
+ARG DEBIAN_FRONTEND=noninteractive
+ARG GIT_URL_REWRITE_FROM=https://github.com/
+ARG GIT_URL_REWRITE_TO=
+
+SHELL ["/bin/bash", "-lc"]
+
+ENV PYENV_ROOT=/root/.pyenv
+ENV PATH=${PYENV_ROOT}/bin:${PYENV_ROOT}/shims:/usr/local/cuda/bin:${PATH}
+ENV CUDA_HOME=/usr/local/cuda
+ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
+ENV PIP_ROOT_USER_ACTION=ignore
+
+
+RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -yq --no-install-recommends tzdata && apt-get install -y openssh-server && mkdir -p /run/sshd
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    curl \
+    wget \
+    ca-certificates \
+    make \
+    build-essential \
+    libssl-dev \
+    zlib1g-dev \
+    libbz2-dev \
+    libreadline-dev \
+    libsqlite3-dev \
+    libncursesw5-dev \
+    xz-utils \
+    tk-dev \
+    libxml2-dev \
+    libxmlsec1-dev \
+    libffi-dev \
+    liblzma-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN if [ -n "${GIT_URL_REWRITE_TO}" ]; then \
+        git config --global url."${GIT_URL_REWRITE_TO}".insteadOf "${GIT_URL_REWRITE_FROM}"; \
+    fi
+
+# pyenv
+RUN curl -fsSL https://pyenv.run | bash
+
+RUN pyenv install 3.12 && \
+    pyenv global 3.12 && \
+    python --version && \
+    pip --version
+
+# torch
+RUN pip3 install --no-warn-conflicts torch torchvision --index-url https://download.pytorch.org/whl/cu132
+
+# cuda
+RUN wget https://developer.download.nvidia.com/compute/cuda/13.2.1/local_installers/cuda-repo-debian13-13-2-local_13.2.1-595.58.03-1_amd64.deb && \
+    dpkg -i cuda-repo-debian13-13-2-local_13.2.1-595.58.03-1_amd64.deb && \
+    cp /var/cuda-repo-debian13-13-2-local/cuda-*-keyring.gpg /usr/share/keyrings/ && \
+    apt-get update && \
+    apt-get -y install cuda-toolkit-13-2 && \
+    rm -f cuda-repo-debian13-13-2-local_13.2.1-595.58.03-1_amd64.deb && \
+    rm -rf /var/lib/apt/lists/*
+
+# flagtree
+RUN pip uninstall -y triton || true && \
+    RES="--index-url=https://resource.flagos.net/repository/flagos-pypi-hosted/simple" && \
+    pip install --no-warn-conflicts flagtree===0.5.0 ${RES}
+
+# flagFFT build and test dependencies
+RUN pip install --no-warn-conflicts nanobind ninja cmake pybind11 pytest && \
+    pip install --no-warn-conflicts -U pip "setuptools<82" wheel scikit-build-core
+
+WORKDIR /workspace
+
+CMD ["/bin/bash"]