diff options
Diffstat (limited to 'python/run-tests')
-rwxr-xr-x | python/run-tests | 35 |
1 files changed, 35 insertions, 0 deletions
diff --git a/python/run-tests b/python/run-tests new file mode 100755 index 0000000000..a3a9ff5dcb --- /dev/null +++ b/python/run-tests @@ -0,0 +1,35 @@ +#!/usr/bin/env bash + +# Figure out where the Scala framework is installed +FWDIR="$(cd `dirname $0`; cd ../; pwd)" + +FAILED=0 + +$FWDIR/pyspark pyspark/rdd.py +FAILED=$(($?||$FAILED)) + +$FWDIR/pyspark pyspark/context.py +FAILED=$(($?||$FAILED)) + +$FWDIR/pyspark -m doctest pyspark/broadcast.py +FAILED=$(($?||$FAILED)) + +$FWDIR/pyspark -m doctest pyspark/accumulators.py +FAILED=$(($?||$FAILED)) + +$FWDIR/pyspark -m unittest pyspark.tests +FAILED=$(($?||$FAILED)) + +if [[ $FAILED != 0 ]]; then + echo -en "\033[31m" # Red + echo "Had test failures; see logs." + echo -en "\033[0m" # No color + exit -1 +else + echo -en "\033[32m" # Green + echo "Tests passed." + echo -en "\033[0m" # No color +fi + +# TODO: in the long-run, it would be nice to use a test runner like `nose`. +# The doctest fixtures are the current barrier to doing this. |