check the number of lines

geval.cabal

@@ -1,5 +1,5 @@
 name:                geval
-version:             0.1.2.1
+version:             0.2.0.0
 synopsis:            Machine learning evaluation tools
 description:         Please see README.md
 homepage:            http://github.com/name/project

src/GEval/Core.hs

@@ -77,6 +77,9 @@ data GEvalException = NoExpectedFile FilePath
                       | NoExpectedTestDirectory FilePath
                       | NoOutTestDirectory FilePath
                       | FileAlreadyThere FilePath
+                      | TooFewLines
+                      | TooManyLines
+                      deriving (Eq)
 
 instance Exception GEvalException
 
@@ -88,6 +91,8 @@ instance Show GEvalException where
   show (NoExpectedTestDirectory filePath) = somethingWrongWithFilesMessage "No test subdirectory with the expected results" filePath
   show (NoOutTestDirectory filePath) = somethingWrongWithFilesMessage "No test subdirectory with the results obtained" filePath
   show (FileAlreadyThere filePath) = somethingWrongWithFilesMessage "File already there" filePath
+  show TooFewLines = "Too few lines in the output file"
+  show TooManyLines = "Too many lines in the output file"
 
 somethingWrongWithFilesMessage :: String -> FilePath -> String
 somethingWrongWithFilesMessage msg filePath = Prelude.concat
@@ -148,28 +153,38 @@ gevalCore' Accuracy = gevalCore'' strip strip hitOrMiss averageC id
 x /. 0 = 1.0
 x /. y = (fromIntegral x) / (fromIntegral y)
 
+data SourceItem a = Got a | Done
+
 gevalCore'' :: (Text -> a) -> (Text -> b) -> ((a, b) -> c) -> (Sink c (ResourceT IO) d) -> (d -> Double ) -> String -> String -> IO (MetricValue)
 gevalCore'' expParser outParser itemStep aggregator finalStep expectedFilePath outFilePath = do
   v <- runResourceT $
     (getZipSource $ (,)
        <$> ZipSource (items expectedFilePath expParser)
        <*> ZipSource (items outFilePath outParser))
-     $$ (CL.map itemStep
+     $$ (CL.map (checkStep itemStep)
+         =$= CL.catMaybes
          =$ aggregator)
   return $ finalStep v
 
+checkStep :: ((a, b) -> c) -> (SourceItem a, SourceItem b) -> Maybe c
+checkStep step (Got expectedItem, Got outItem) = Just $ step (expectedItem, outItem)
+checkStep _ (Got _, Done) = throw TooFewLines
+checkStep _ (Done, Got _) = throw TooManyLines
+checkStep _ (Done, Done) = Nothing
+
+
 averageC :: MonadResource m => Sink Double m Double
 averageC = getZipSink
     $ (\total count -> total / fromIntegral count)
   <$> ZipSink CC.sum
   <*> ZipSink CC.length
 
-items :: MonadResource m => String -> (Text -> a) -> Source m a
+items :: MonadResource m => String -> (Text -> a) -> Source m (SourceItem a)
 items filePath parser =
-  CB.sourceFile filePath
+  (CB.sourceFile filePath
    $= (CT.decode CT.utf8
        =$= CT.lines
-      =$= CL.map parser)
+       =$= CL.map ((\x -> Got x) . parser))) >> yield Done
 
 itemError :: (Double, Double) -> Double
 itemError (exp, out) = (exp-out)**2

test/Spec.hs

@@ -48,10 +48,22 @@ main = hspec $ do
       precisionCount [["bar", "bar", "bar", "bar", "foo", "xyz", "foo"]] ["foo", "bar", "foo", "baz", "bar", "foo"] `shouldBe` 4
     it "multiple refs" $ do
       precisionCount [["foo", "baz"], ["bar"], ["baz", "xyz"]]  ["foo", "bar", "foo"] `shouldBe` 2
+  describe "error handling" $ do
+    it "too few lines are handled" $ do
+      runGEvalTest "error-too-few-lines" `shouldThrow` (== TooFewLines)
+    it "too many lines are handled" $ do
+      runGEvalTest "error-too-many-lines" `shouldThrow` (== TooManyLines)
+
 
 extractVal :: (Either (ParserResult GEvalOptions) (Maybe MetricValue)) -> IO MetricValue
 extractVal (Right (Just val)) = return val
 
+runGEvalTest testName = (runGEval [
+  "--expected-directory",
+  "test/" ++ testName ++ "/" ++ testName,
+  "--out-directory",
+  "test/" ++ testName ++ "/" ++ testName ++ "-solution"]) >>= extractVal
+
 class AEq a where
     (=~) :: a -> a -> Bool
 

test/error-too-few-lines/error-too-few-lines-solution/test-A/out.tsv

@@ -0,0 +1 @@
+1.0

test/error-too-few-lines/error-too-few-lines/config.txt

@@ -0,0 +1 @@
+--metric MSE

test/error-too-few-lines/error-too-few-lines/test-A/expected.tsv

@@ -0,0 +1,3 @@
+3.0
+10.0
+2.8

test/error-too-few-lines/error-too-few-lines/test-A/in.tsv

@@ -0,0 +1,3 @@
+A	b
+B	c
+A	c

test/error-too-many-lines/error-too-many-lines-solution/test-A/out.tsv

@@ -0,0 +1,4 @@
+1.0
+4.0
+3.2
+-1.0

test/error-too-many-lines/error-too-many-lines/config.txt

@@ -0,0 +1 @@
+--metric MSE

test/error-too-many-lines/error-too-many-lines/test-A/expected.tsv

@@ -0,0 +1,3 @@
+3.0
+10.0
+2.8

test/error-too-many-lines/error-too-many-lines/test-A/in.tsv

@@ -0,0 +1,3 @@
+A	b
+B	c
+A	c