name: Python Test
run-name: ${{ github.actor }} is running Riffusion CI

on:
  push:
    branches:
      - 'main'
  pull_request:
    types: [opened, synchronize, reopened, ready_for_review]

jobs:
  run:
    runs-on: ubuntu-latest

    strategy:
      matrix:
        python-version: ["3.9", "3.10"]

    steps:
      - name: Checkout
        uses: actions/checkout@v3

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

      - name: Install system packages
        run: |
          sudo apt-get update
          sudo apt-get install -y ffmpeg libsndfile1

      - name: Upgrade pip
        run: |
          python -m pip install --upgrade pip

      - name: Install pip packages
        run: |
          pip install -r requirements_all.txt

      - name: Test with unittest
        run: |
          RIFFUSION_TEST_DEVICE=cpu python -m unittest test/*_test.py