Accept test file as command-line argument.

2 files changed, 8 insertions, 2 deletions

diff --git a/test/TestSabadie.hs b/test/TestSabadie.hs
index d8d1949..76d6aee 100644
--- a/test/TestSabadie.hs
+++ b/test/TestSabadie.hs
@@ -1,6 +1,8 @@
 module TestSabadie where
 
 import Codec.Wavefront
+import Data.Maybe
+import System.Environment
 import System.Directory
 import System.IO
 import Text.Show.Pretty (ppShow)
@@ -14,5 +16,6 @@ getDataDir = do
 
 main = do
     ddir <- getDataDir
-    r1 <- fromFile (ddir ++ "/01.obj")
+    fn <- fromMaybe (ddir ++ "/01.obj") . listToMaybe <$>  getArgs
+    r1 <- fromFile fn
     either print (putStrLn . ppShow) r1
diff --git a/test/TestSundqvist.hs b/test/TestSundqvist.hs
index 25d3f8f..d02da10 100644
--- a/test/TestSundqvist.hs
+++ b/test/TestSundqvist.hs
@@ -1,9 +1,11 @@
 module TestSundqvist where
 
 
+import Data.Maybe
 import Graphics.WaveFront
 import Graphics.WaveFront.Load as Load
 import System.Directory
+import System.Environment
 import System.IO
 
 getDataDir :: IO FilePath
@@ -15,5 +17,6 @@ getDataDir = do
 
 main = do
     ddir <- getDataDir
-    r1 <- Load.obj (ddir ++ "/01.obj")
+    fn <- fromMaybe (ddir ++ "/01.obj") . listToMaybe <$>  getArgs
+    r1 <- Load.obj fn
     either print (mapM_ print) r1