2 files changed, 10 insertions, 1 deletions

diff --git a/README b/README
index 3cbcab1..7b2c65a 100644
--- a/README
+++ b/README
@@ -107,3 +107,7 @@ in the Haskell mailing lists for their help.
 - Simon Beaumont reported the need of QuickCheck<2 and the invalid
   asm("finit") on ppc. He also contributed the configuration options
   for the accelerate framework on OS X.
+
+- Daniel Schüssler added compatibility with QuickCheck 2 as well
+  as QuickCheck 1 using the C preprocessor. He also added some
+  implementations for the new "shrink" method of class Arbitrary.
diff --git a/lib/Numeric/LinearAlgebra/Tests.hs b/lib/Numeric/LinearAlgebra/Tests.hs
index 8b92287..7758268 100644
--- a/lib/Numeric/LinearAlgebra/Tests.hs
+++ b/lib/Numeric/LinearAlgebra/Tests.hs
@@ -182,9 +182,10 @@ runTests n = do
     test (\v -> ifft (fft v) |~| v)
     putStrLn "------ vector operations"
     test (\u -> sin u ^ 2 + cos u ^ 2 |~| (1::RM))
+    test $ (\u -> sin u ^ 2 + cos u ^ 2 |~| (1::CM)) . liftMatrix makeUnitary
     test (\u -> sin u ** 2 + cos u ** 2 |~| (1::RM))
     test (\u -> cos u * tan u |~| sin (u::RM))
-    test (\u -> (cos u * tan u) |~| sin (u::CM))
+    test $ (\u -> cos u * tan u |~| sin (u::CM)) . liftMatrix makeUnitary
     putStrLn "------ read . show"
     test (\m -> (m::RM) == read (show m))
     test (\m -> (m::CM) == read (show m))
@@ -208,6 +209,10 @@ runTests n = do
         ]
     return ()
 
+makeUnitary v | realPart n > 1    = v */ n
+              | otherwise = v
+    where n = sqrt (conj v <.> v)
+
 -- -- | Some additional tests on big matrices. They take a few minutes.
 -- runBigTests :: IO ()
 -- runBigTests = undefined