pytest.yaml

name: Test

on:
  pull_request:
    branches: [ main , "migrate**"]
  schedule:
  # 05:00 UTC = 06:00 CET = 07:00 CEST
  - cron: "0 5 * * *"

# Cancel previous runs that have not completed
concurrency:
  group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
  cancel-in-progress: true

jobs:
  warm-lfs-cache:
    strategy:
      matrix:
        os: [ macos-13, ubuntu-latest, windows-latest ]
    runs-on: ${{ matrix.os }}
    steps:
    - uses: nschloe/action-cached-lfs-checkout@v1

  pytest:
    needs: warm-lfs-cache

    strategy:
      matrix:
        os:
        - macos-13
        - ubuntu-latest
        - windows-latest
        upstream:
        # In each group:
        # - Versions of ixmp and message_ix to test.
        # - Latest supported Python version for those or other dependencies.
        # - Extra dependencies, in particular fixed/maximum versions to resolve conficts.
        #   - dask[dataframe] >= 2024.3.0 requires dask-expr and in turn pandas >= 2.0.
        #     https://github.com/iiasa/message-ix-models/pull/156#issuecomment-2020152360
        #   - genno: upstream versions < 3.8.0 import genno.computations, removed in 1.25.0.
        #     https://github.com/iiasa/message-ix-models/pull/156
        #   - pytest: upstream versions < 3.9.0 use a hook argument removed in pytest 8.1.0.
        #     https://github.com/iiasa/message-ix-models/pull/155
        #
        # Minimum version given in pyproject.toml
        - version: v3.4.0
          python-version: "3.11"
          extra-deps: '"dask < 2024.3.0" "genno < 1.25" "pandas < 2.0" "pytest == 8.0.0"' #
        - version: v3.5.0
          python-version: "3.11"
          extra-deps: '"dask < 2024.3.0" "genno < 1.25" "pandas < 2.0" "pytest == 8.0.0"' #
        - version: v3.6.0
          python-version: "3.11"
          extra-deps: '"dask < 2024.3.0" "genno < 1.25" "pandas < 2.0" "pytest == 8.0.0"' #
        - version: v3.7.0
          python-version: "3.11"
          extra-deps: 'dask[dataframe]   "genno < 1.25"                "pytest == 8.0.0"' #
        # Latest released version
        - version: v3.8.0
          python-version: "3.12"
          extra-deps: 'dask[dataframe]                                 "pytest == 8.0.0"' #
        # Development version
        - version: main
          python-version: "3.12"
          extra-deps: 'dask[dataframe]' #

      fail-fast: false

    runs-on: ${{ matrix.os }}
    name: ${{ matrix.os }}-py${{ matrix.upstream.python-version }}-upstream-${{ matrix.upstream.version }}

    steps:
    - name: Cache test data
      uses: actions/cache@v4
      with:
        path: message-local-data
        key: ${{ matrix.os }}

    - name: Check out message-ix-models
      uses: nschloe/action-cached-lfs-checkout@v1

    - uses: actions/setup-python@v5
      with:
        python-version: ${{ matrix.upstream.python-version }}
        cache: pip
        cache-dependency-path: "**/pyproject.toml"

    - uses: iiasa/actions/setup-gams@main
      with:
        version: 29.1.0
        license: ${{ secrets.GAMS_LICENSE }}

    - uses: ts-graphviz/setup-graphviz@v2
      # TEMPORARY Work around ts-graphviz/setup-graphviz#630
      if: ${{ ! startswith(matrix.os, 'macos-') }}

    - name: Install packages and dependencies
      # By default, install:
      # - ixmp, message_ix: from GitHub branches/tags per matrix.upstream-version (above)
      # - other dependencies including genno: from PyPI.
      #
      # To test against unreleased code (on `main`, or other branches
      # for open PRs), temporarily uncomment, add, or edit lines below
      # as needed. DO NOT merge such changes to `main`.
      run: |
        # pip install --upgrade "genno @ git+https://github.com/khaeru/genno.git@main"
        pip install --upgrade "ixmp @ git+https://github.com/iiasa/ixmp.git@${{ matrix.upstream.version }}"
        pip install --upgrade "message-ix @ git+https://github.com/iiasa/message_ix.git@${{ matrix.upstream.version }}"

        pip install .[docs,tests] ${{ matrix.upstream.extra-deps }}

        # TEMPORARY Work around hgrecco/pint#2007, unionai-oss/pandera#1685;
        # see https://github.com/khaeru/genno/issues/140
        pip install "pint != 0.24.0" "numpy < 2"

    - name: Configure local data path
      run: |
        mkdir -p message-local-data/cache
        mix-models config set "message local data" "$(realpath message-local-data)"
        mix-models config show
      shell: bash

    - name: Run test suite using pytest
      run: |
        pytest message_ix_models \
          -m "not (ece_db or snapshot)" \
          -rA --verbose --color=yes --durations=20 \
          --cov-report=term-missing --cov-report=xml \
          --numprocesses=auto \
          --local-cache --jvmargs="-Xmx6G"
      shell: bash

    - name: Upload test coverage to Codecov.io
      uses: codecov/codecov-action@v4
      with:
        token: ${{ secrets.CODECOV_TOKEN }} # required

  pre-commit:
    name: Code quality

    runs-on: ubuntu-latest

    steps:
    - uses: actions/checkout@v4
    - uses: actions/setup-python@v5
      with: { python-version: "3.12" }

    - name: Force recreation of pre-commit virtual environment for mypy
      if: github.event_name == 'schedule'  # Comment this line to run on a PR
      run: gh cache list -L 999 | cut -f2 | grep pre-commit | xargs -I{} gh cache delete "{}" || true
      env: { GH_TOKEN: "${{ github.token }}" }

    - uses: pre-commit/action@v3.0.1