CI: reusable chapter test workflow + chapters matrix

Author: srikanthbaride

.github/workflows/_chapter-tests.yml

@@ -1,44 +1,92 @@
-name: chapter tests (reusable)
+name: Chapter Tests (Reusable)
 
 on:
   workflow_call:
     inputs:
       chapter:
-        description: "Folder name of the chapter (e.g., ch2_rl_formulation)"
+        description: "Chapter folder name (e.g., ch7_td_control)"
         required: true
         type: string
+      python-version:
+        description: "Python version to use"
+        required: false
+        default: "3.10"
+        type: string
+      extra-deps:
+        description: "Extra pip packages to install (space-separated)"
+        required: false
+        default: ""
+        type: string
 
 jobs:
   test:
-    name: ${{ inputs.chapter }} (py${{ matrix.python }})
+    name: Run tests for ${{ inputs.chapter }}
     runs-on: ubuntu-latest
-    strategy:
-      fail-fast: false
-      matrix:
-        python: ["3.10", "3.11", "3.12"]
+    timeout-minutes: 15
+
+    env:
+      MPLBACKEND: Agg
+      PYTHONDONTWRITEBYTECODE: "1"
 
     steps:
-      - name: Check out repository
+      - name: Checkout
         uses: actions/checkout@v4
 
-      - name: Set up Python ${{ matrix.python }}
+      - name: Set up Python ${{ inputs.python-version }}
         uses: actions/setup-python@v5
         with:
-          python-version: ${{ matrix.python }}
+          python-version: ${{ inputs.python-version }}
           cache: "pip"
 
-      - name: Upgrade pip
-        run: python -m pip install -U pip
-
-      - name: Install dependencies (root requirements.txt)
-        run: pip install -r requirements.txt
+      - name: Install dependencies
+        shell: bash
+        run: |
+          python -m pip install --upgrade pip
+          # Root requirements if present
+          if [ -f requirements.txt ]; then
+            pip install -r requirements.txt
+          fi
+          # Minimal defaults + optional extras
+          pip install numpy matplotlib pytest
+          if [ -n "${{ inputs.extra-deps }}" ]; then
+            pip install ${{ inputs.extra-deps }}
+          fi
 
-      - name: Show environment
+      - name: Show environment (debug)
         run: |
-          python -V
+          python --version
+          pip --version
           pip list
 
-      - name: Run tests
+      - name: Run pytest (scoped to chapter if tests exist)
+        shell: bash
         run: |
-          echo "Running pytest in ${{ inputs.chapter }}/tests"
-          pytest -q ${{ inputs.chapter }}/tests
+          set -euo pipefail
+          CHAP="${{ inputs.chapter }}"
+          TARGET=""
+          if [ -d "tests/${CHAP}" ]; then
+            TARGET="tests/${CHAP}"
+          elif [ -f "tests/test_${CHAP}.py" ]; then
+            TARGET="tests/test_${CHAP}.py"
+          else
+            # Fallback: run full suite
+            TARGET=""
+          fi
+
+          echo "Chapter: ${CHAP}"
+          if [ -n "${TARGET}" ]; then
+            echo "Running targeted tests at: ${TARGET}"
+            pytest -q "${TARGET}"
+          else
+            echo "No chapter-specific tests found; running full suite."
+            pytest -q
+          fi
+
+      # Optional: upload pytest cache / reports if you later add them
+      # - name: Upload junit report
+      #   if: always()
+      #   uses: actions/upload-artifact@v4
+      #   with:
+      #     name: junit-${{ inputs.chapter }}
+      #     path: reports/junit-${{ inputs.chapter }}.xml
+      #     if-no-files-found: ignore

.github/workflows/python-tests.yml

@@ -1,31 +1,14 @@
-name: Python (Chapters)
-
-on:
-  push:
-    branches: [ "main" ]
-    paths:
-      - "ch2_rl_formulation/**"
-      - "ch4_dynamic_programming/**"
-      - "ch5_monte_carlo/**"
-      - ".github/workflows/python-tests.yml"
-      - ".github/workflows/_chapter-tests.yml"
-      - "requirements.txt"
-  pull_request:
-    branches: [ "main" ]
-    paths:
-      - "ch2_rl_formulation/**"
-      - "ch4_dynamic_programming/**"
-      - "ch5_monte_carlo/**"
-      - ".github/workflows/python-tests.yml"
-      - ".github/workflows/_chapter-tests.yml"
-      - "requirements.txt"
-
 jobs:
   ch2:
     uses: ./.github/workflows/_chapter-tests.yml
     with:
       chapter: ch2_rl_formulation
 
+  ch3:
+    uses: ./.github/workflows/_chapter-tests.yml
+    with:
+      chapter: ch3_multi_armed_bandits
+
   ch4:
     uses: ./.github/workflows/_chapter-tests.yml
     with:
@@ -35,3 +18,13 @@ jobs:
     uses: ./.github/workflows/_chapter-tests.yml
     with:
       chapter: ch5_monte_carlo
+
+  ch6:
+    uses: ./.github/workflows/_chapter-tests.yml
+    with:
+      chapter: ch6_td_learning
+
+  ch7:
+    uses: ./.github/workflows/_chapter-tests.yml
+    with:
+      chapter: ch7_td_control