runtests.sh 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. #!/bin/bash
  2. CUR=$(pwd)
  3. DATASET=bug.tar.gz
  4. DATASET_URL=http://www.ipe.fzk.de/~vogelgesang/$DATASET
  5. if [ ! -d "$CUR/data" ]; then
  6. echo "* Downloading data set ..."
  7. curl -o $CUR/$DATASET $DATASET_URL
  8. echo "* Extracting data set into data/ ..."
  9. mkdir -p $CUR/data/
  10. tar xfz $CUR/$DATASET -C $CUR/data
  11. echo "* Clean up ..."
  12. rm $CUR/$DATASET
  13. fi
  14. if [ ! -d "$CUR/venv" ]; then
  15. echo "* Creating virtualenv ..."
  16. virtualenv --system-site-packages -q $CUR/venv
  17. fi
  18. echo "* Activate virtualenv ..."
  19. source $CUR/venv/bin/activate
  20. # Install nose if not available
  21. pip freeze | grep nose > /dev/null
  22. RC=$?
  23. if [ "$RC" -ne "0" ]; then
  24. echo "* Installing nose ..."
  25. pip install nose
  26. fi
  27. # Install unittest2 if Python less than 2.7
  28. python -c 'import sys; v=sys.version_info; sys.exit((v[0], v[1]) < (2,7))'
  29. RC=$?
  30. if [ "$RC" -ne "0" ]; then
  31. pip freeze | grep unittest2 > /dev/null
  32. RC=$?
  33. if [ "$RC" -ne "0" ]; then
  34. echo "* Installing unittest2 ..."
  35. pip install unittest2
  36. fi
  37. fi
  38. echo "* Running tests ..."
  39. export CUDA_VISIBLE_DEVICES=0
  40. nosetests -s tests.py