X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=test%2FDoctests.hs;h=3cfd2a83cfbe679a541b18ac509f0780e175f552;hb=HEAD;hp=f9bacc5b238a57740d046f53b3fa34c0528e175e;hpb=898622d362b9dad42296af6298e20bc9f8410b63;p=numerical-analysis.git

diff --git a/test/Doctests.hs b/test/Doctests.hs
index f9bacc5..3cfd2a8 100644
--- a/test/Doctests.hs
+++ b/test/Doctests.hs
@@ -2,6 +2,12 @@ module Main
 where
 
 import Test.DocTest
+import System.FilePath.Find ((==?), always, extension, find)
+
+find_sources :: IO [FilePath]
+find_sources = find always (extension ==? ".hs") "src/"
 
 main :: IO ()
-main = doctest ["--optghc=-isrc", "src/Root.hs"]
+main = do
+  sources <- find_sources
+  doctest $ ["-isrc"] ++ sources