1 files changed, 14 insertions, 5 deletions

diff --git a/TestAufgabeFFP3.hs b/TestAufgabeFFP3.hs
index 065401d..30a915e 100644..100755
--- a/TestAufgabeFFP3.hs
+++ b/TestAufgabeFFP3.hs
@@ -1,10 +1,13 @@
+#!/usr/bin/runhugs +l
+
 module Main where
 
 import Prelude hiding (filter)
 
-
 import Test.HUnit
 import Control.Monad
+import System
+import Data.Array
 import AufgabeFFP3
 
 cases1 = TestLabel "foo" $ TestList [
@@ -33,11 +36,17 @@ cases4 = TestLabel "bar" $ TestList [
   TestCase $ assertEqual "a" ( [binomS (n, k) | k <- [0..50], n <- [k..50] ] ) ( [binomM (n, k) | k <- [0..50], n <- [k..50] ] ) 
   ]
 
-
 tests :: [Test]
 tests = [cases1, cases2, cases3, cases4]
 
-main = do
-  forM tests $ \test ->
-    runTestTT test
+isSuccess :: Counts -> Bool
+isSuccess Counts{ cases = _, tried = _, errors = 0, failures = 0 } = True
+isSuccess Counts{ cases = _, tried = _, errors = _, failures = _ } = False
+
+runTest :: Test -> IO ()
+runTest test = do
+  result <- runTestTT test
+  unless (isSuccess result) exitFailure
 
+main = do
+  forM tests $ (\test -> runTest test)