before_merge.yaml

name: "Tests"

on:
  workflow_call:
  workflow_dispatch:
  merge_group:

permissions:
  actions: read
  contents: read
  pages: write
  id-token: write

jobs:
  validate-pr:
    env:
      ARCH_NAME: wormhole_b0
    runs-on: ["in-service", "n150"]
    steps:
      - name: Checkout Repo
        uses: actions/checkout@v4
      - name: Install dependencies
        run: |
          python3 -m venv venv
          source venv/bin/activate
          python3 -m pip config set global.extra-index-url https://download.pytorch.org/whl/cpu
          python3 -m pip install --upgrade pip                    
          python3 -m pip install -r requirements-dev.txt
          python3 -m pip install pytest-github-report
      - name: Run Tools Tests
        env:
          pytest_verbosity: 2
          pytest_report_title: "⭐️ Tools Tests"
        run: | 
          source venv/bin/activate
          python3 -m pytest --github-report tests/tools/ -s      
      - name: Run Lowering Tests
        env:
          pytest_verbosity: 2
          pytest_report_title: "⭐️ Aten → TTNN Lowering Tests"
        run: | 
          source venv/bin/activate
          python3 -m pytest --github-report tests/lowering/ -s
      - name: Run Model Tests
        env:
          pytest_verbosity: 2
          pytest_report_title: "⭐️ Model Tests"
        run: | 
          source venv/bin/activate
          python3 -m pytest --github-report tests/models/ -s