Merge pull request #2462 from rochisha0/add-jax-en #37
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # The name is short because we mostly care how it appears in the pull request | |
| # "checks" dialogue box - it looks like | |
| # Tests / ubuntu-latest, python-3.9, defaults | |
| # or similar. | |
| name: Tests | |
| on: | |
| [push, pull_request] | |
| defaults: | |
| run: | |
| # The slightly odd shell call is to force bash to read .bashrc, which is | |
| # necessary for having conda behave sensibly. We use bash as the shell even | |
| # on Windows, since we don't run anything much complicated, and it makes | |
| # things much simpler. | |
| shell: bash -l -e {0} | |
| jobs: | |
| cases: | |
| name: ${{ matrix.os }}, python${{ matrix.python-version }}, ${{ matrix.case-name }} | |
| runs-on: ${{ matrix.os }} | |
| env: | |
| MPLBACKEND: Agg # Explicitly define matplotlib backend for Windows tests | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| os: [ubuntu-latest] | |
| # Test other versions of Python in special cases to avoid exploding the | |
| # matrix size; make sure to test all supported versions in some form. | |
| python-version: ["3.11"] | |
| case-name: [defaults] | |
| # Version 2 not yet available on conda's default channel | |
| condaforge: [1] | |
| numpy-build: [""] | |
| numpy-requirement: [""] | |
| scipy-requirement: [">=1.9"] | |
| coverage-requirement: ["==6.5"] | |
| # Extra special cases. In these, the new variable defined should always | |
| # be a truth-y value (hence 'nomkl: 1' rather than 'mkl: 0'), because | |
| # the lack of a variable is _always_ false-y, and the defaults lack all | |
| # the special cases. | |
| include: | |
| - case-name: p312 numpy 2 | |
| os: ubuntu-latest | |
| python-version: "3.12" | |
| numpy-build: ">=2.0.0" | |
| numpy-requirement: ">=2.0.0" | |
| pypi: 1 | |
| - case-name: p310 numpy 1.22 | |
| os: ubuntu-latest | |
| python-version: "3.10" | |
| numpy-build: ">=1.22.0,<1.23.0" | |
| numpy-requirement: ">=1.22.0,<1.23.0" | |
| scipy-requirement: ">=1.10,<1.11" | |
| semidefinite: 1 | |
| oldcython: 1 | |
| pypi: 1 | |
| pytest-extra-options: "-W ignore:dep_util:DeprecationWarning -W \"ignore:The 'renderer' parameter of do_3d_projection\"" | |
| # Python 3.12 and latest numpy | |
| # Use conda-forge to provide Python 3.11 and latest numpy | |
| - case-name: p312, numpy fallback | |
| os: ubuntu-latest | |
| python-version: "3.12" | |
| numpy-requirement: ">=1.26,<1.27" | |
| scipy-requirement: ">=1.11,<1.12" | |
| condaforge: 1 | |
| # Install mpi4py to test mpi_pmap | |
| # Should be enough to include this in one of the runs | |
| includempi: 1 | |
| # Python 3.10, no mkl, scipy 1.9, numpy 1.23 | |
| # Scipy 1.9 did not support cython 3.0 yet. | |
| # cython#17234 | |
| - case-name: p310 no mkl | |
| os: ubuntu-latest | |
| python-version: "3.10" | |
| numpy-requirement: ">=1.23,<1.24" | |
| scipy-requirement: ">=1.9,<1.10" | |
| semidefinite: 1 | |
| condaforge: 1 | |
| oldcython: 1 | |
| nomkl: 1 | |
| pytest-extra-options: "-W ignore:dep_util:DeprecationWarning -W \"ignore:The 'renderer' parameter of do_3d_projection\"" | |
| # Mac | |
| # Mac has issues with MKL since september 2022. | |
| - case-name: macos | |
| # setup-miniconda not compatible with macos-latest presently. | |
| # https://github.com/conda-incubator/setup-miniconda/issues/344 | |
| os: macos-12 | |
| python-version: "3.12" | |
| numpy-build: ">=2.0.0" | |
| numpy-requirement: ">=2.0.0" | |
| condaforge: 1 | |
| nomkl: 1 | |
| - case-name: macos - numpy fallback | |
| os: macos-12 | |
| python-version: "3.11" | |
| numpy-build: ">=2.0.0" | |
| numpy-requirement: ">=1.25,<1.26" | |
| condaforge: 1 | |
| nomkl: 1 | |
| - case-name: Windows | |
| os: windows-latest | |
| python-version: "3.11" | |
| numpy-build: ">=2.0.0" | |
| numpy-requirement: ">=2.0.0" | |
| pypi: 1 | |
| - case-name: Windows - numpy fallback | |
| os: windows-latest | |
| python-version: "3.10" | |
| numpy-build: ">=2.0.0" | |
| numpy-requirement: ">=1.24,<1.25" | |
| semidefinite: 1 | |
| oldcython: 1 | |
| nocython: 1 | |
| condaforge: 1 | |
| pytest-extra-options: "-W ignore:dep_util:DeprecationWarning -W \"ignore:The 'renderer' parameter of do_3d_projection\"" | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: conda-incubator/setup-miniconda@v3 | |
| with: | |
| auto-update-conda: true | |
| python-version: ${{ matrix.python-version }} | |
| channels: ${{ matrix.condaforge == 1 && 'conda-forge' || 'defaults' }} | |
| - name: Install QuTiP and dependencies | |
| # In the run, first we handle any special cases. We do this in bash | |
| # rather than in the GitHub Actions file directly, because bash gives us | |
| # a proper programming language to use. | |
| # We install without build isolation so qutip is compiled with the | |
| # version of cython, scipy, numpy in the test matrix, not a temporary | |
| # version use in the installation virtual environment. | |
| run: | | |
| # Install the extra requirement | |
| python -m pip install pytest>=5.2 pytest-rerunfailures # tests | |
| python -m pip install ipython # ipython | |
| python -m pip install loky tqdm # extras | |
| python -m pip install "coverage${{ matrix.coverage-requirement }}" chardet | |
| python -m pip install pytest-cov coveralls pytest-fail-slow | |
| if [[ "${{ matrix.pypi }}" ]]; then | |
| pip install "numpy${{ matrix.numpy-build }}" | |
| pip install "scipy${{ matrix.scipy-requirement }}" | |
| elif [[ -z "${{ matrix.nomkl }}" ]]; then | |
| conda install blas=*=mkl "numpy${{ matrix.numpy-build }}" "scipy${{ matrix.scipy-requirement }}" | |
| elif [[ "${{ matrix.os }}" =~ ^windows.*$ ]]; then | |
| # Conda doesn't supply forced nomkl builds on Windows, so we rely on | |
| # pip not automatically linking to MKL. | |
| pip install "numpy${{ matrix.numpy-build }}" "scipy${{ matrix.scipy-requirement }}" | |
| else | |
| conda install nomkl "numpy${{ matrix.numpy-build }}" "scipy${{ matrix.scipy-requirement }}" | |
| fi | |
| if [[ -n "${{ matrix.conda-extra-pkgs }}" ]]; then | |
| conda install "${{ matrix.conda-extra-pkgs }}" | |
| fi | |
| if [[ "${{ matrix.includempi }}" ]]; then | |
| # Use openmpi because mpich causes problems. Note, environment variable names change in v5 | |
| conda install "openmpi<5" mpi4py | |
| fi | |
| if [[ "${{ matrix.oldcython }}" ]]; then | |
| python -m pip install cython==0.29.36 filelock matplotlib==3.5 | |
| else | |
| python -m pip install cython filelock | |
| fi | |
| python -m pip install -e . -v --no-build-isolation | |
| if [[ "${{ matrix.nocython }}" ]]; then | |
| python -m pip uninstall cython -y | |
| fi | |
| if [[ "${{ matrix.pypi }}" ]]; then | |
| python -m pip install "numpy${{ matrix.numpy-requirement }}" | |
| elif [[ -z "${{ matrix.nomkl }}" ]]; then | |
| conda install "numpy${{ matrix.numpy-requirement }}" | |
| elif [[ "${{ matrix.os }}" =~ ^windows.*$ ]]; then | |
| python -m pip install "numpy${{ matrix.numpy-requirement }}" | |
| else | |
| conda install nomkl "numpy${{ matrix.numpy-requirement }}" | |
| fi | |
| if [[ -n "${{ matrix.semidefinite }}" ]]; then | |
| python -m pip install cvxpy>=1.0 cvxopt | |
| fi | |
| python -m pip install matplotlib>=1.2.1 # graphics | |
| - name: Package information | |
| run: | | |
| conda list | |
| python -c "import qutip; qutip.about()" | |
| python -c "import qutip; print(qutip.settings)" | |
| - name: Environment information | |
| run: | | |
| uname -a | |
| if [[ "ubuntu-latest" == "${{ matrix.os }}" ]]; then | |
| hostnamectl | |
| lscpu | |
| free -h | |
| fi | |
| - name: Run tests | |
| # If our tests are running for longer than an hour, _something_ is wrong | |
| # somewhere. The GitHub default is 6 hours, which is a bit long to wait | |
| # to see if something hung. | |
| timeout-minutes: 60 | |
| run: | | |
| if [[ -n "${{ matrix.openmp }}" ]]; then | |
| # Force OpenMP runs to use more threads, even if there aren't | |
| # actually that many CPUs. We have to check any dispatch code is | |
| # truly being executed. | |
| export QUTIP_NUM_PROCESSES=2 | |
| fi | |
| if [[ "${{ matrix.includempi }}" ]]; then | |
| # By default, the max. number of allowed worker processes in openmpi is | |
| # (number of physical cpu cores) - 1. | |
| # We only have 2 physical cores, but we want to test mpi_pmap with 2 workers. | |
| export OMPI_MCA_rmaps_base_oversubscribe=true | |
| fi | |
| pytest -Werror --strict-config --strict-markers --fail-slow=300 --durations=0 --durations-min=1.0 --verbosity=1 --cov=qutip --cov-report= --color=yes ${{ matrix.pytest-extra-options }} qutip/tests | |
| # Above flags are: | |
| # -Werror | |
| # treat warnings as errors | |
| # --strict-config | |
| # error out if the configuration file is not parseable | |
| # --strict-markers | |
| # error out if a marker is used but not defined in the | |
| # configuration file | |
| # --timeout=300 | |
| # error any individual test that goes longer than the given time | |
| # --durations=0 --durations-min=1.0 | |
| # at the end, show a list of all the tests that took longer than a | |
| # second to run | |
| # --verbosity=1 | |
| # turn the verbosity up so pytest prints the names of the tests | |
| # it's currently working on | |
| # --cov=qutip | |
| # limit coverage reporting to code that's within the qutip package | |
| # --cov-report= | |
| # don't print the coverage report to the terminal---it just adds | |
| # cruft, and we're going to upload the .coverage file to Coveralls | |
| # --color=yes | |
| # force coloured output in the terminal | |
| - name: Upload to Coveralls | |
| env: | |
| GITHUB_TOKEN: ${{ secrets.github_token }} | |
| COVERALLS_FLAG_NAME: ${{ matrix.os }}-${{ matrix.python-version }}-${{ matrix.case-name }} | |
| COVERALLS_PARALLEL: true | |
| run: coveralls --service=github | |
| towncrier-check: | |
| name: Verify Towncrier entry added | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v4 | |
| with: | |
| fetch-depth: 0 | |
| - name: Install Towncrier | |
| run: | | |
| python -m pip install towncrier | |
| - name: Verify Towncrier entry added | |
| if: github.event_name == 'pull_request' | |
| env: | |
| BASE_BRANCH: ${{ github.base_ref }} | |
| run: | | |
| # Fetch the pull request' base branch so towncrier will be able to | |
| # compare the current branch with the base branch. | |
| # Source: https://github.com/actions/checkout/#fetch-all-branches. | |
| git fetch --no-tags origin +refs/heads/${BASE_BRANCH}:refs/remotes/origin/${BASE_BRANCH} | |
| towncrier check --compare-with origin/${BASE_BRANCH} | |
| towncrier build --version "$(cat VERSION)" --draft | |
| finalise: | |
| name: Finalise coverage reporting | |
| needs: cases | |
| runs-on: ubuntu-latest | |
| container: python:3-slim | |
| steps: | |
| - name: Finalise coverage reporting | |
| env: | |
| GITHUB_TOKEN: ${{ secrets.github_token }} | |
| run: | | |
| python -m pip install coveralls | |
| coveralls --service=github --finish |