tests.yml

name: Tests

on:
  pull_request:
  push:
    branches:
      - master
      - dev
      - streamlined-backend


jobs:
  test:
    runs-on: ${{ matrix.os }}
    strategy:
      matrix:
        os: [ubuntu-latest, windows-latest]
        python-version: ["3.10", "3.11"]
        backend: ["jax", "tensorflow", "torch"]
    defaults:
      run:
        shell: bash -el {0}
    env:
      KERAS_BACKEND: ${{ matrix.backend }}

    steps:
      - name: Checkout code
        uses: actions/checkout@v4

      - name: Set up Conda
        uses: conda-incubator/setup-miniconda@v3
        with:
          environment-file: environment.yaml
          python-version: ${{ matrix.python-version }}

      - name: Install JAX
        if: ${{ matrix.backend == 'jax' }}
        run: |
          pip install -U "jax[cpu]"
      - name: Install NumPy
        if: ${{ matrix.backend == 'numpy' }}
        run: |
          conda install numpy
      - name: Install Tensorflow
        if: ${{ matrix.backend == 'tensorflow' }}
        run: |
          pip install -U tensorflow
      - name: Install PyTorch
        if: ${{ matrix.backend == 'torch' }}
        run: |
          conda install pytorch torchvision torchaudio cpuonly -c pytorch

      - name: Show Environment Info
        run: |
          conda info
          conda list
          conda config --show-sources
          conda config --show
          printenv | sort

      - name: Run JAX Tests
        if: ${{ matrix.backend == 'jax' }}
        run: |
          python -m pytest tests/ -v -m "not (numpy or tensorflow or torch)"

      - name: Run NumPy Tests
        if: ${{ matrix.backend == 'numpy' }}
        run: |
          python -m pytest tests/ -v -m "not (jax or tensorflow or torch)"

      - name: Run TensorFlow Tests
        if: ${{ matrix.backend == 'tensorflow' }}
        run: |
          python -m pytest tests/ -v -m "not (jax or numpy or torch)"

      - name: Run PyTorch Tests
        if: ${{ matrix.backend == 'torch' }}
        run: |
          python -m pytest tests/ -v -m "not (jax or numpy or tensorflow)"