Create dynamic GitHub Actions matrix to test against PyBaMM

.github/workflows/scheduled_tests.yaml

@@ -10,53 +10,82 @@ on:
   schedule:
     - cron: '0 9 * * *'
 
+# Check noxfile.py for associated environment variables
+env:
+  PYBOP_SCHEDULED: 1
+
 jobs:
+  # Dynamically create a matrix of OS, Python, and PyBaMM versions
+  create_pybamm_matrix:
+    name: Dynamically create GitHub Actions matrix
+    runs-on: ubuntu-latest
+    steps:
+      - name: Check out PyBOP repository
+        uses: actions/checkout@v4
+        with:
+          sparse-checkout-cone-mode: false
+          sparse-checkout: |
+            scripts/ci/build_matrix.sh
+
+      - name: Run script to create matrix
+        id: set-matrix
+        run: |
+          echo "matrix=$(bash scripts/ci/build_matrix.sh)" >> "$GITHUB_OUTPUT"
+    outputs:
+      pybop_matrix: ${{ steps.set-matrix.outputs.matrix }}
+
   build:
+    needs: [create_pybamm_matrix]
+    name: Build (${{ matrix.os }}, Python ${{ matrix.python_version }}, PyBaMM ${{ matrix.pybamm_version }})
     runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false
-      matrix:
-        os: [ubuntu-latest, windows-latest, macos-latest]
-        python-version: ["3.8", "3.9", "3.10", "3.11"]
+      matrix: ${{fromJson(needs.create_pybamm_matrix.outputs.pybop_matrix)}}
+    env:
+      PYBAMM_VERSION: ${{ matrix.pybamm_version }}
 
     steps:
       - uses: actions/checkout@v4
-      - name: Set up Python ${{ matrix.python-version }}
+      - name: Set up Python ${{ matrix.python_version }}
         uses: actions/setup-python@v4
         with:
-          python-version: ${{ matrix.python-version }}
+          python-version: ${{ matrix.python_version }}
+
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip nox
+
       - name: Unit tests with nox
-        run: |
-          python -m nox -s unit
-          python -m nox -s notebooks
+        run: python -m nox -s unit
+
+      - name: Run notebooks with nox
+        run: python -m nox -s notebooks
 
-  #M-series Mac Mini
+  # M-series Mac Mini
   build-apple-mseries:
     runs-on: [self-hosted, macOS, ARM64]
+    if: github.repository == 'pybop-team/PyBOP'
     env:
       GITHUB_PATH: ${PYENV_ROOT/bin:$PATH}
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.10"]
+        python_version: ["3.10"]
 
     steps:
       - uses: actions/checkout@v4
       - name: Install python & create virtualenv
         shell: bash
         run: |
           eval "$(pyenv init -)"
-          pyenv install ${{ matrix.python-version }} -s
-          pyenv virtualenv ${{ matrix.python-version }} pybop-${{ matrix.python-version }}
+          pyenv install ${{ matrix.python_version }} -s
+          pyenv virtualenv ${{ matrix.python_version }} pybop-${{ matrix.python_version }}
 
       - name: Install dependencies & run unit tests
         shell: bash
         run: |
           eval "$(pyenv init -)"
-          pyenv activate pybop-${{ matrix.python-version }}
+          pyenv activate pybop-${{ matrix.python_version }}
           python -m pip install --upgrade pip wheel setuptools nox
           python -m nox -s unit
           python -m nox -s notebooks
@@ -66,5 +95,5 @@ jobs:
         shell: bash
         run: |
           eval "$(pyenv init -)"
-          pyenv activate pybop-${{ matrix.python-version }}
+          pyenv activate pybop-${{ matrix.python_version }}
           pyenv uninstall -f $( python --version )

noxfile.py

@@ -1,20 +1,31 @@
+import os
 import nox
 
+
 # nox options
 nox.options.reuse_existing_virtualenvs = True
+nox.options.venv_backend = "virtualenv"
+
+# Environment variables to control CI behaviour for nox sessions
+PYBOP_SCHEDULED = int(os.environ.get("PYBOP_SCHEDULED", 0))
+PYBAMM_VERSION = os.environ.get("PYBAMM_VERSION", None)
 
 
 @nox.session
 def unit(session):
-    session.run_always("pip", "install", "-e", ".[all]")
-    session.install("pytest", "pytest-mock")
+    session.install("-e", ".[all]", silent=False)
+    if PYBOP_SCHEDULED:
+        session.run("pip", "install", f"pybamm=={PYBAMM_VERSION}", silent=False)
+    session.install("pytest", "pytest-mock", silent=False)
     session.run("pytest", "--unit")
 
 
 @nox.session
 def coverage(session):
-    session.run_always("pip", "install", "-e", ".[all]")
-    session.install("pytest", "pytest-cov", "pytest-mock")
+    session.install("-e", ".[all]", silent=False)
+    if PYBOP_SCHEDULED:
+        session.run("pip", "install", f"pybamm=={PYBAMM_VERSION}", silent=False)
+    session.install("pytest", "pytest-cov", "pytest-mock", silent=False)
     session.run(
         "pytest",
         "--unit",
@@ -27,8 +38,10 @@ def coverage(session):
 @nox.session
 def notebooks(session):
     """Run the examples tests for Jupyter notebooks."""
-    session.run_always("pip", "install", "-e", ".[all]")
-    session.install("pytest", "nbmake")
+    session.install("-e", ".[all]", silent=False)
+    if PYBOP_SCHEDULED:
+        session.run("pip", "install", f"pybamm=={PYBAMM_VERSION}", silent=False)
+    session.install("pytest", "nbmake", silent=False)
     session.run("pytest", "--nbmake", "--examples", "examples/", external=True)
 
 
@@ -39,7 +52,7 @@ def docs(session):
     Credit: PyBaMM Team
     """
     envbindir = session.bin
-    session.install("-e", ".[all,docs]")
+    session.install("-e", ".[all,docs]", silent=False)
     session.chdir("docs")
     # Local development
     if session.interactive:

scripts/ci/build_matrix.sh

@@ -0,0 +1,43 @@
+#!/bin/bash
+
+# This helper script generates a matrix for further use in the
+# scheduled/nightly builds for PyBOP, i.e., in scheduled_tests.yaml
+# It generates a matrix of all combinations of the following variables:
+# - python_version: 3.X
+# - os: ubuntu-latest, windows-latest, macos-latest
+# - pybamm_version: the last X versions of PyBaMM from PyPI, excluding release candidates
+
+# To update the matrix, the variables below can be modified as needed.
+
+python_version=("3.8" "3.9" "3.10" "3.11")
+os=("ubuntu-latest" "windows-latest" "macos-latest")
+# This command fetches the last three PyBaMM versions excluding release candidates from PyPI
+pybamm_version=($(curl -s https://pypi.org/pypi/pybamm/json | jq -r '.releases | keys[]' | grep -v rc | tail -n 3 | awk '{print "\"" $1 "\"" }' | paste -sd " " -))
+
+# open dict
+json='{
+  "include": [
+'
+
+# loop through each combination of variables to generate matrix components
+for py_ver in "${python_version[@]}"; do
+  for os_type in "${os[@]}"; do
+    for pybamm_ver in "${pybamm_version[@]}"; do
+      json+='{
+        "os": "'$os_type'",
+        "python_version": "'$py_ver'",
+        "pybamm_version": '$pybamm_ver'
+      },'
+    done
+  done
+done
+
+# fix structure, removing trailing comma
+json=${json%,}
+
+# close dict
+json+='
+  ]
+}'
+
+echo "$json" | jq -c .