pytest-codspeed/.github/workflows/codspeed.yml at 7f56309474ebe36e2dcbb6c0103ea4cd6d326a1b · CodSpeedHQ/pytest-codspeed · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
name: CodSpeed
on:
  push:
    branches: [master]
  pull_request:
    branches: [master]
  workflow_dispatch:

env:
  PYTHON_VERSION: "3.14"
  SHARDS: 4

jobs:
  pytest-benchmarks:
    strategy:
      matrix:
        shard: [1, 2, 3, 4]
        mode: ["instrumentation", "walltime"]

    name: "pytest ${{ matrix.mode }} benchmarks (Shard #${{ matrix.shard }})"
    runs-on: ${{ matrix.mode == 'instrumentation' && 'ubuntu-24.04' || 'codspeed-macro' }}
    steps:
      - uses: actions/checkout@v5
        with:
          submodules: "recursive"
      - uses: astral-sh/setup-uv@v7
      - uses: actions/setup-python@v6
        with:
          python-version: ${{ env.PYTHON_VERSION }}
      - name: Install local version of pytest-codspeed
        run: |
          sudo apt-get update
          sudo apt-get install valgrind -y
          uv sync --dev
          sudo apt-get remove valgrind -y
      - name: Run benchmarks
        uses: CodSpeedHQ/action@main
        with:
          mode: ${{ matrix.mode }}
          run: uv run pytest pytest-codspeed/tests/benchmarks/ --codspeed --ignore=pytest-codspeed/tests/benchmarks/TheAlgorithms --test-group=${{ matrix.shard }} --test-group-count=${{ env.SHARDS }}
          token: ${{ secrets.CODSPEED_TOKEN }}

  asv-benchmarks:
    strategy:
      matrix:
        mode: ["instrumentation", "walltime"]

    name: "asv ${{ matrix.mode }} benchmarks"
    runs-on: ${{ matrix.mode == 'instrumentation' && 'ubuntu-24.04' || 'codspeed-macro' }}
    steps:
      - uses: actions/checkout@v5
        with:
          submodules: "recursive"
      - uses: astral-sh/setup-uv@v7
      - uses: actions/setup-python@v6
        with:
          python-version: ${{ env.PYTHON_VERSION }}
      - name: Install dependencies
        run: |
          sudo apt-get update
          sudo apt-get install valgrind -y
          uv sync --dev
          sudo apt-get remove valgrind -y
      - name: Run ASV benchmarks
        uses: CodSpeedHQ/action@main
        with:
          mode: ${{ matrix.mode }}
          run: uv run asv-codspeed run asv-codspeed/tests/sample_benchmarks
          token: ${{ secrets.CODSPEED_TOKEN }}

  all-checks:
    runs-on: ubuntu-latest
    steps:
      - run: echo "All CI checks passed."
    needs:
      - pytest-benchmarks
      - asv-benchmarks