pytest.yaml

name: Test

on:
  push:
    branches:
      - main
      - release/*
  pull_request:
    branches:
      - main
      - release/*
  workflow_call:
  workflow_dispatch: {}
jobs:
  pytest:
    timeout-minutes: 30
    runs-on: ubuntu-latest
    strategy:
      matrix:
        python_version:
          - "3.9"
        install_version:
          - "[all]"
    steps:
      - uses: actions/checkout@v3
      - uses: actions/setup-python@v4
        with:
          python-version: ${{ matrix.python_version }}
      - name: Cache pip
        uses: actions/cache@v3
        with:
          # This path is specific to Ubuntu
          path: ~/.cache/pip
          # Look to see if there is a cache hit for the corresponding requirements file
          key: ${{ runner.os }}-pip-${{ hashFiles('setup.py') }}
          restore-keys: |
            ${{ runner.os }}-pip-
            ${{ runner.os }}-
      - name: Setup
        run: |
          set -ex
          python -m pip install --upgrade pip wheel
          python -m pip install -e .${{ matrix.install_version }}
      - name: Run Tests
        # TODO: stop using pytest-split and remove low coverage requirement once C++ shared mem impl is done!
        # Jira link for above: https://mosaicml.atlassian.net/browse/STR-107
        id: tests
        run: |
          set -ex
          pytest --splits 7 --group 1 --cov-fail-under=10
          pytest --splits 7 --group 2 --cov-fail-under=10
          pytest --splits 7 --group 3 --cov-fail-under=10
          pytest --splits 7 --group 4 --cov-fail-under=10
          pytest --splits 7 --group 5 --cov-fail-under=10
          pytest --splits 7 --group 6 --cov-fail-under=10
          pytest --splits 7 --group 7 --cov-fail-under=10