1 files changed, 31 insertions, 9 deletions

diff --git a/examples/tests.hs b/examples/experiments/bigtests.hs
index cd923cd..a3dfad2 100644
--- a/examples/tests.hs
+++ b/examples/experiments/bigtests.hs
@@ -11,15 +11,41 @@ pseudorandomR seed (n,m) = reshape m $ fromList $ take (n*m) $ randomRs (-100,10
 
 pseudorandomC seed (n,m) = toComplex (pseudorandomR seed (n,m), pseudorandomR (seed+1) (n,m))
 
-bigmat = m + trans m :: RM
-    where m = pseudorandomR 18 (1000,1000)
-bigmatc = mc + ctrans mc ::CM
-    where mc = pseudorandomC 19 (1000,1000)
+bigmat = m + trans m
+    where m = pseudorandomR 18 (1000,1000) :: Matrix Double
+bigmatc = mc + ctrans mc
+    where mc = pseudorandomC 19 (1000,1000) :: Matrix (Complex Double)
 
 utest str b = TestCase $ assertBool str b
 
 feye n = flipud (ident n) :: Matrix Double
 
+infixl 4 |~|
+a |~| b = dist a b < 10^^(-10)
+
+dist a b = r
+    where norm = pnorm Infinity
+          na = norm a
+          nb = norm b
+          nab = norm (a-b)
+          mx = max na nb
+          mn = min na nb
+          r = if mn < eps
+                then mx
+                else nab/mx
+
+square m = rows m == cols m
+
+unitary m = square m && m <> ctrans m |~| ident (rows m)
+
+eigProp m = complex m <> v |~| v <> diag s
+    where (s, v) = eig m
+
+eigSHProp m = m <> v |~| v <> real (diag s)
+              && unitary v
+              && m |~| v <> real (diag s) <> ctrans v
+    where (s, v) = eigSH m
+
 bigtests = do
     putStrLn "--------- big matrices -----"
     runTestTT $ TestList
@@ -31,8 +57,4 @@ bigtests = do
      ]
     return ()
 
-main = do
-    args <- getArgs
-    if "--big" `elem` args
-        then bigtests
-        else runTests 20
+main = bigtests