tests.yaml

name: Tests

on: [ pull_request ]

env:
  MODULE_NAME: thunder

jobs:
  test:
    runs-on: ubuntu-latest
    strategy:
      matrix:
        python-version: [ '3.8', '3.9', '3.10', '3.11' ]
        torch-version: [ '1.13', '2.0.1' ]
        exclude:
          - torch-version: '1.13'
            python-version: '3.11'

    steps:
      - uses: actions/checkout@v3
      - name: Set up Python ${{ matrix.python-version }}
        uses: actions/setup-python@v4
        with:
          python-version: ${{ matrix.python-version }}

      - name: Build the package
        run: |
          python setup.py sdist

      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          python -m pip install -e .
          python -m pip install -r tests/dev-requirements.txt
          python -m pip install torch==${{ matrix.torch-version }}
          cd tests
          export MODULE_PARENT=$(python -c "import $MODULE_NAME, os; print(os.path.dirname($MODULE_NAME.__path__[0]))")
          export MODULE_PARENT=${MODULE_PARENT%"/"}
          cd ..
          echo $MODULE_PARENT
          echo "MODULE_PARENT=$(echo $MODULE_PARENT)" >> $GITHUB_ENV

      - name: Test with pytest
        run: |
          pytest tests --junitxml=reports/junit-${{ matrix.python-version }}.xml --cov="$MODULE_PARENT/$MODULE_NAME" --cov-report=xml --cov-branch
      - name: Generate coverage report
        run: |
          coverage xml -o reports/coverage-${{ matrix.python-version }}.xml
          sed -i -e "s|$MODULE_PARENT/||g" reports/coverage-${{ matrix.python-version }}.xml
          sed -i -e "s|$(echo $MODULE_PARENT/ | tr "/" .)||g" reports/coverage-${{ matrix.python-version }}.xml

      - name: Upload artifacts
        uses: actions/upload-artifact@v3
        with:
          name: reports-${{ matrix.python-version }}
          path: reports/*-${{ matrix.python-version }}.xml

      - name: Upload coverage results
        uses: codecov/codecov-action@v3
        with:
          files: reports/coverage-${{ matrix.python-version }}.xml
          verbose: true