#!/bin/bash
set -e
set -x

suite=$1

# Install XGBoost Python package
wheel_found=0
for file in python-package/dist/*.whl
do
  if [ -e "${file}" ]
  then
    pip install --user "${file}"
    wheel_found=1
    break  # need just one
  fi
done
if [ "$wheel_found" -eq 0 ]
then
  pushd .
  cd python-package
  python setup.py install --user
  popd
fi

# Run specified test suite
case "$suite" in
  gpu)
    pytest -v -s --fulltrace -m "not mgpu" tests/python-gpu
    ;;

  mgpu)
    pytest -v -s --fulltrace -m "mgpu" tests/python-gpu
    cd tests/distributed
    ./runtests-gpu.sh
    cd -
    pytest -v -s --fulltrace -m "mgpu" tests/python-gpu/test_gpu_with_dask.py
    ;;

  cudf)
    source activate cudf_test
    pytest -v -s --fulltrace -m "not mgpu" tests/python-gpu/test_from_columnar.py tests/python-gpu/test_from_cupy.py
    ;;

  cpu)
    pytest -v -s --fulltrace tests/python
    cd tests/distributed
    ./runtests.sh
    ;;

  *)
    echo "Usage: $0 {gpu|mgpu|cudf|cpu}"
    exit 1
    ;;
esac