From 1a64432ba50904c3933d8a9539a619fc94b3b30b Mon Sep 17 00:00:00 2001
From: Josh Rosen <joshrosen@eecs.berkeley.edu>
Date: Wed, 9 Jan 2013 20:30:36 -0800
Subject: Indicate success/failure in PySpark test script.

---
 python/run-tests | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/python/run-tests b/python/run-tests
index da9e24cb1f..fcdd1e27a7 100755
--- a/python/run-tests
+++ b/python/run-tests
@@ -3,7 +3,24 @@
 # 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 -m doctest pyspark/broadcast.py
+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.
-- 
cgit v1.2.3