1 files changed, 14 insertions, 1 deletions
diff --git a/test.sh b/test.sh
index a8c60bd..a27b321 100755
--- a/test.sh
+++ b/test.sh
@@ -5,19 +5,32 @@
 # tests for a given phase.                       #
 ##################################################
 
+function java() {
+    expected=$(bash runner.sh java "$1.java")
+}
+
+function match() {
+    [[ "$1" == *"error"* ]] && expected="Type error" ||
+            expected="Program type checked successfully"
+}
+
 testdir="./output"
 case "$1" in
     "heat")
         ext="java"
+        comp="match"
         ;;
     "boil")
         ext="java"
+        comp="java"
         ;;
     "vaporize")
         ext="vapor"
+        comp="java"
         ;;
     "condense")
         ext="vaporm"
+        comp="java"
         ;;
     *)
         echo "usage $0 [heat|boil|vaporize|condense]"
@@ -31,7 +44,7 @@ for file in $testdir/*.$ext; do
     [ -f "$file" ] || break
     base=${file%.*}
     echo -n "Processing file: $base "
-    expected=$(bash runner.sh java "${base}.java")
+    $comp $base
     actual=$(bash runner.sh $1 $file 2>/dev/null)
     [[ $expected == $actual ]] && echo 'PASSED' || echo -e "FAILED"
 done