CI: try using -compat-wrapped modules

Author: ev-br

.github/workflows/test.yml

@@ -9,7 +9,7 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.11"]
-        array-api-module: ["array_api_strict", "numpy", "torch", "jax.numpy"]
+        array-api-module: ["array_api_strict", "array_api_compat.numpy", "array_api_compat.torch", "jax.numpy"]
       fail-fast: false
 
     steps:
@@ -24,16 +24,17 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        if [ ${{ matrix.array-api-module }} = 'torch' ]
+        if [[ ${{ matrix.array-api-module }} == *"torch"* ]]
         then
           python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
-        elif [ ${{ matrix.array-api-module }} = 'jax.numpy' ]
+        elif [[ ${{ matrix.array-api-module }} == *"jax.numpy"* ]]
         then
           python -m pip install "jax[cpu]"
         else
           python -m pip install ${{ matrix.array-api-module }}
         fi
         python -m pip install -r requirements.txt
+        python -m pip install array-api-compat
     - name: Run the test suite
       env:
         ARRAY_API_TESTS_MODULE: ${{ matrix.array-api-module }}