X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=test%2FDoctests.hs;h=3cfd2a83cfbe679a541b18ac509f0780e175f552;hb=08d08d4b0616286e5344d9aa9cc585a871ebc232;hp=3ce1b4a04c736bc5bfaa9346430a3145e5d59d7c;hpb=57a982c464095201f7635977aac937ad6a08c0b0;p=numerical-analysis.git

diff --git a/test/Doctests.hs b/test/Doctests.hs
index 3ce1b4a..3cfd2a8 100644
--- a/test/Doctests.hs
+++ b/test/Doctests.hs
@@ -2,8 +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/Roots/Simple.hs",
-                "src/Roots/Fast.hs"]
+main = do
+  sources <- find_sources
+  doctest $ ["-isrc"] ++ sources