CI: parametrize over the Array API modules

data-apis · Nov 14, 2024 · f742ae2 · f742ae2
1 parent 9db56ec
commit f742ae2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -9,6 +9,7 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.10", "3.11"]
+        array-api-module: ["array_api_strict", "numpy", "torch"]
 
     steps:
     - name: Checkout array-api-tests
@@ -22,11 +23,11 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install array-api-strict
+        python -m pip install ${{ matrix.array-api-module }}
         python -m pip install -r requirements.txt
     - name: Run the test suite
       env:
-        ARRAY_API_TESTS_MODULE: array_api_strict
+        ARRAY_API_TESTS_MODULE: ${ {matrix.array-api-module }}
         ARRAY_API_STRICT_API_VERSION: 2023.12
       run: |
         pytest -v -rxXfE --skips-file array-api-strict-skips.txt array_api_tests/