update puclic CI workflow and newly added tests

Author: Vahid Tavanashad

.github/workflows/conda-package.yml

@@ -16,6 +16,7 @@ env:
   CONDA_BUILD_INDEX_ENV_PY_VER: '3.12' # conda does not support python 3.13
   CONDA_BUILD_VERSION: '25.1.1'
   CONDA_INDEX_VERSION: '0.5.0'
+  LATEST_PYTHON: '3.13'
   RERUN_TESTS_ON_FAILURE: 'true'
   RUN_TESTS_MAX_ATTEMPTS: 2
   TEST_ENV_NAME: 'test'
@@ -209,12 +210,19 @@ jobs:
           python -c "import dpnp; print(dpnp.__version__)"
 
       - name: Run tests
-        if: env.RERUN_TESTS_ON_FAILURE != 'true'
+        if: env.RERUN_TESTS_ON_FAILURE != 'true' && matrix.python != env.LATEST_PYTHON
         run: |
           python -m pytest -n auto -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
 
+      - name: Run tests for all dtypes
+        if: env.RERUN_TESTS_ON_FAILURE != 'true' && matrix.python == env.LATEST_PYTHON
+        env:
+          DPNP_TEST_ALL_INT_TYPES: 1
+        run: |
+          python -m pytest -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
+
       - name: Run tests
-        if: env.RERUN_TESTS_ON_FAILURE == 'true'
+        if: env.RERUN_TESTS_ON_FAILURE == 'true' && matrix.python != env.LATEST_PYTHON
         id: run_tests_linux
         uses: nick-fields/retry@7152eba30c6575329ac0576536151aca5a72780e # v3.0.0
         with:
@@ -228,110 +236,22 @@ jobs:
 
             python -m pytest -n auto -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
 
-  test_linux_all_dtypes:
-    name: Test_All_dtypes
-
-    needs: build
-
-    runs-on: ${{ matrix.os }}
-
-    defaults:
-      run:
-        shell: bash -el {0}
-
-    strategy:
-      fail-fast: false
-      matrix:
-        python: ['3.12']
-        os: [ubuntu-latest]
-
-    env:
-      channel-path: '${{ github.workspace }}/channel/'
-      pkg-path-in-channel: '${{ github.workspace }}/channel/linux-64/'
-      ver-json-path: '${{ github.workspace }}/version.json'
-
-    steps:
-      - name: Download artifact
-        uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8
-        with:
-          name: ${{ env.PACKAGE_NAME }} ${{ runner.os }} Python ${{ matrix.python }}
-          path: ${{ env.pkg-path-in-channel }}
-
-      - name: Setup miniconda
-        id: setup_miniconda
-        continue-on-error: true
-        uses: conda-incubator/setup-miniconda@505e6394dae86d6a5c7fbb6e3fb8938e3e863830 # v3.1.1
-        with:
-          miniforge-version: latest
-          use-mamba: 'true'
-          channels: conda-forge
-          conda-remove-defaults: 'true'
-          python-version: ${{ env.CONDA_BUILD_INDEX_ENV_PY_VER}}
-          activate-environment: ${{ env.TEST_ENV_NAME }}
-
-      - name: ReSetup miniconda
-        if: steps.setup_miniconda.outcome == 'failure'
-        uses: conda-incubator/setup-miniconda@505e6394dae86d6a5c7fbb6e3fb8938e3e863830 # v3.1.1
-        with:
-          miniforge-version: latest
-          use-mamba: 'true'
-          channels: conda-forge
-          conda-remove-defaults: 'true'
-          python-version: ${{ env.CONDA_BUILD_INDEX_ENV_PY_VER}}
-          activate-environment: ${{ env.TEST_ENV_NAME }}
-
-      - name: Install conda-index
-        run: |
-          mamba install conda-index=${{ env.CONDA_INDEX_VERSION }}
-
-      - name: Create conda channel
-        run: |
-          python -m conda_index ${{ env.channel-path }}
-
-      - name: Test conda channel
-        run: |
-          conda search ${{ env.PACKAGE_NAME }} -c ${{ env.channel-path }} --override-channels --info --json > ${{ env.ver-json-path }}
-          cat ${{ env.ver-json-path }}
-
-      - name: Get package version
-        run: |
-          export PACKAGE_VERSION=$(python -c "${{ env.VER_SCRIPT1 }} ${{ env.VER_SCRIPT2 }}")
-
-          echo PACKAGE_VERSION=${PACKAGE_VERSION}
-          echo "PACKAGE_VERSION=$PACKAGE_VERSION" >> $GITHUB_ENV
-
-      # conda-index does not support python 3.13
-      - name: Remove conda-index
-        run: mamba remove conda-index
-
-      - name: Install dpnp
-        id: install_dpnp
-        continue-on-error: true
-        run: |
-          mamba install ${{ env.PACKAGE_NAME }}=${{ env.PACKAGE_VERSION }} pytest python=${{ matrix.python }} ${{ env.TEST_CHANNELS }}
-        env:
-          TEST_CHANNELS: '-c ${{ env.channel-path }} ${{ env.CHANNELS }}'
-
-      - name: ReInstall dpnp
-        if: steps.install_dpnp.outcome == 'failure'
-        run: |
-          mamba install ${{ env.PACKAGE_NAME }}=${{ env.PACKAGE_VERSION }} pytest python=${{ matrix.python }} ${{ env.TEST_CHANNELS }}
-        env:
-          TEST_CHANNELS: '-c ${{ env.channel-path }} ${{ env.CHANNELS }}'
-
-      - name: List installed packages
-        run: mamba list
-
-      - name: Smoke test
-        run: |
-          python -c "import dpctl; dpctl.lsplatform()"
-          python -c "import dpnp; print(dpnp.__version__)"
-
       - name: Run tests for all dtypes
+        if: env.RERUN_TESTS_ON_FAILURE == 'true' && matrix.python == env.LATEST_PYTHON
+        id: run_tests_linux_all_dtypes
+        uses: nick-fields/retry@7152eba30c6575329ac0576536151aca5a72780e # v3.0.0
         env:
           DPNP_TEST_ALL_INT_TYPES: 1
-        run: |
-          python -m pytest -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
+        with:
+          timeout_minutes: 20
+          max_attempts: ${{ env.RUN_TESTS_MAX_ATTEMPTS }}
+          retry_on: any
+          command: |
+            . $CONDA/etc/profile.d/conda.sh
+            . $CONDA/etc/profile.d/mamba.sh
+            mamba activate ${{ env.TEST_ENV_NAME }}
+
+            python -m pytest -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
 
   test_windows:
     name: Test
@@ -451,12 +371,19 @@ jobs:
           python -c "import dpnp; print(dpnp.__version__)"
 
       - name: Run tests
-        if: env.RERUN_TESTS_ON_FAILURE != 'true'
+        if: env.RERUN_TESTS_ON_FAILURE != 'true' && matrix.python != env.LATEST_PYTHON
         run: |
           python -m pytest -n auto -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
 
+      - name: Run tests for all dtypes
+        if: env.RERUN_TESTS_ON_FAILURE != 'true' && matrix.python == env.LATEST_PYTHON
+        env:
+          DPNP_TEST_ALL_INT_TYPES: 1
+        run: |
+          python -m pytest -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
+
       - name: Run tests
-        if: env.RERUN_TESTS_ON_FAILURE == 'true'
+        if: env.RERUN_TESTS_ON_FAILURE == 'true' && matrix.python != env.LATEST_PYTHON
         id: run_tests_win
         uses: nick-fields/retry@7152eba30c6575329ac0576536151aca5a72780e # v3.0.0
         with:
@@ -466,128 +393,18 @@ jobs:
           command: |
             python -m pytest -n auto -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
 
-  test_windows_all_dtypes:
-    name: Test_All_dtypes
-
-    needs: build
-
-    runs-on: ${{ matrix.os }}
-
-    defaults:
-      run:
-        shell: cmd /C CALL {0}
-
-    strategy:
-      fail-fast: false
-      matrix:
-        python: ['3.12']
-        os: [windows-2019]
-
-    env:
-      channel-path: '${{ github.workspace }}\channel\'
-      pkg-path-in-channel: '${{ github.workspace }}\channel\win-64\'
-      ver-json-path: '${{ github.workspace }}\version.json'
-      workdir: '${{ github.workspace }}'
-
-    steps:
-      - name: Download artifact
-        uses: actions/download-artifact@fa0a91b85d4f404e444e00e005971372dc801d16 # v4.1.8
-        with:
-          name: ${{ env.PACKAGE_NAME }} ${{ runner.os }} Python ${{ matrix.python }}
-          path: ${{ env.pkg-path-in-channel }}
-
-      - name: Store a path to package archive
-        run: |
-          @echo on
-
-          set SEARCH_SCRIPT="DIR ${{ env.pkg-path-in-channel }} /s/b | FINDSTR /r "dpnp-.*\.conda""
-          FOR /F "tokens=* USEBACKQ" %%F IN (`%SEARCH_SCRIPT%`) DO (
-            SET FULL_PACKAGE_PATH=%%F
-          )
-          echo FULL_PACKAGE_PATH: %FULL_PACKAGE_PATH%
-          (echo FULL_PACKAGE_PATH=%FULL_PACKAGE_PATH%) >> %GITHUB_ENV%
-
-      - name: Setup miniconda
-        uses: conda-incubator/setup-miniconda@505e6394dae86d6a5c7fbb6e3fb8938e3e863830 # v3.1.1
-        with:
-          miniforge-version: latest
-          use-mamba: 'true'
-          channels: conda-forge
-          conda-remove-defaults: 'true'
-          python-version: ${{ env.CONDA_BUILD_INDEX_ENV_PY_VER}}
-          activate-environment: ${{ env.TEST_ENV_NAME }}
-
-      - name: Store conda paths as envs
-        run: |
-          @echo on
-          (echo CONDA_LIB_PATH=%CONDA_PREFIX%\Library\lib\) >> %GITHUB_ENV%
-          (echo CONDA_LIB_BIN_PATH=%CONDA_PREFIX%\Library\bin\) >> %GITHUB_ENV%
-
-      - name: Install conda-index
-        run: |
-          mamba install conda-index=${{ env.CONDA_INDEX_VERSION }}
-
-      - name: Create conda channel
-        run: |
-          @echo on
-          python -m conda_index ${{ env.channel-path }}
-
-      - name: Test conda channel
-        run: |
-          @echo on
-          conda search ${{ env.PACKAGE_NAME }} -c ${{ env.channel-path }} --override-channels --info --json > ${{ env.ver-json-path }}
-
-      - name: Dump version.json
-        run: more ${{ env.ver-json-path }}
-
-      - name: Get package version
-        run: |
-          @echo on
-          set "SCRIPT=${{ env.VER_SCRIPT1 }} ${{ env.VER_SCRIPT2 }}"
-          FOR /F "tokens=* USEBACKQ" %%F IN (`python -c "%SCRIPT%"`) DO (
-            set PACKAGE_VERSION=%%F
-          )
-          echo PACKAGE_VERSION: %PACKAGE_VERSION%
-          (echo PACKAGE_VERSION=%PACKAGE_VERSION%) >> %GITHUB_ENV%
-
-      # conda-index does not support python 3.13
-      - name: Remove conda-index
-        run: mamba remove conda-index
-
-      - name: Install dpnp
-        run: |
-          @echo on
-          mamba install ${{ env.PACKAGE_NAME }}=${{ env.PACKAGE_VERSION }} pytest python=${{ matrix.python }} ${{ env.TEST_CHANNELS }}
-        env:
-          TEST_CHANNELS: '-c ${{ env.channel-path }} ${{ env.CHANNELS }}'
-          MAMBA_NO_LOW_SPEED_LIMIT: 1
-
-      - name: List installed packages
-        run: mamba list
-
-      - name: Activate OCL CPU RT
-        shell: pwsh
-        run: |
-          $script_path="$env:CONDA_PREFIX\Scripts\set-intel-ocl-icd-registry.ps1"
-          if (Test-Path $script_path) {
-            &$script_path
-          } else {
-            Write-Warning "File $script_path was NOT found!"
-          }
-          # Check the variable assisting OpenCL CPU driver to find TBB DLLs which are not located where it expects them by default
-          $cl_cfg="$env:CONDA_PREFIX\Library\lib\cl.cfg"
-          Get-Content -Tail 5 -Path $cl_cfg
-
-      - name: Smoke test
-        run: |
-          python -c "import dpctl; dpctl.lsplatform()"
-          python -c "import dpnp; print(dpnp.__version__)"
-
       - name: Run tests for all dtypes
+        if: env.RERUN_TESTS_ON_FAILURE == 'true' && matrix.python == env.LATEST_PYTHON
+        id: run_tests_win_all_dtypes
+        uses: nick-fields/retry@7152eba30c6575329ac0576536151aca5a72780e # v3.0.0
         env:
           DPNP_TEST_ALL_INT_TYPES: 1
-        run: |
-          python -m pytest -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
+        with:
+          timeout_minutes: 30
+          max_attempts: ${{ env.RUN_TESTS_MAX_ATTEMPTS }}
+          retry_on: any
+          command: |
+            python -m pytest -ra --pyargs ${{ env.PACKAGE_NAME }}.tests
 
   upload:
     name: Upload

dpnp/tests/test_indexing.py

@@ -473,7 +473,7 @@ def test_input_1d(self, a_dt, indices, ind_dt, ivals, mode):
             b.put(ind, vals, mode=mode)
             ib.put(iind, ivals, mode=mode)
             assert_array_equal(ib, b)
-        elif numpy.issubdtype(ind_dt, numpy.uint64):
+        elif ind_dt == numpy.uint64:
             # For this special case, NumPy raises an error but dpnp works
             assert_raises(TypeError, numpy.put, a, ind, vals, mode=mode)
             assert_raises(TypeError, b.put, ind, vals, mode=mode)
@@ -508,7 +508,7 @@ def test_input_2d(self, a_dt, indices, ind_dt, mode):
         ind = get_abs_array(indices, ind_dt)
         iind = dpnp.array(ind)
 
-        if numpy.issubdtype(ind_dt, numpy.uint64):
+        if ind_dt == numpy.uint64:
             # For this special case, NumPy raises an error but dpnp works
             assert_raises(TypeError, numpy.put, a, ind, vals, mode=mode)
 
@@ -656,7 +656,7 @@ def test_broadcast(self, arr_dt, idx_dt):
         ind = numpy.arange(10, dtype=idx_dt).reshape((1, 2, 5)) % 4
         ia, iind = dpnp.array(a), dpnp.array(ind)
 
-        if numpy.issubdtype(idx_dt, numpy.uint64):
+        if idx_dt == numpy.uint64:
             numpy.put_along_axis(a, ind, 20, axis=1)
             dpnp.put_along_axis(ia, iind, 20, axis=1)
             assert_array_equal(ia, a)
@@ -701,7 +701,7 @@ def test_1d(self, a_dt, ind_dt, indices, mode):
             result = dpnp.take(ia, iind, mode=mode)
             expected = numpy.take(a, ind, mode=mode)
             assert_array_equal(result, expected)
-        elif numpy.issubdtype(ind_dt, numpy.uint64):
+        elif ind_dt == numpy.uint64:
             # For this special case, although casting `ind_dt` to numpy.intp
             # is not safe, both NumPy and dpnp work properly
             # NumPy < "2.2.0" raises an error
@@ -726,7 +726,7 @@ def test_2d(self, a_dt, ind_dt, indices, mode, axis):
         ind = get_abs_array(indices, ind_dt)
         ia, iind = dpnp.array(a), dpnp.array(ind)
 
-        if numpy.issubdtype(ind_dt, numpy.uint64):
+        if ind_dt == numpy.uint64:
             # For this special case, NumPy raises an error on Windows
             result = ia.take(iind, axis=axis, mode=mode)
             expected = a.take(ind.astype(numpy.int64), axis=axis, mode=mode)
@@ -1461,6 +1461,16 @@ def test_choose_inds_all_dtypes(self, dtype):
             chcs = dpnp.ones(1, dtype=dtype)
             with pytest.raises(TypeError):
                 dpnp.choose(inds, chcs)
+        elif dtype == numpy.uint64:
+            # For this special case, NumPy raises an error but dpnp works
+            inds_np = numpy.array([1, 0, 1], dtype=dtype)
+            inds = dpnp.array(inds_np)
+            chcs_np = numpy.array([1, 2, 3], dtype=dtype)
+            chcs = dpnp.array(chcs_np)
+            assert_raises(TypeError, numpy.choose, inds_np, chcs_np)
+            expected = numpy.choose(inds_np.astype(numpy.int64), chcs_np)
+            result = dpnp.choose(inds, chcs)
+            assert_array_equal(expected, result)
         else:
             inds_np = numpy.array([1, 0, 1], dtype=dtype)
             inds = dpnp.array(inds_np)