From a2814f2d12879d0d2c99493204983cd17a07d650 Mon Sep 17 00:00:00 2001
From: Filip Gralinski <filipg@amu.edu.pl>
Date: Fri, 5 Jan 2018 21:56:25 +0100
Subject: [PATCH] add function for evaluating single lines

---
 src/GEval/Core.hs | 14 ++++++++++++++
 test/Spec.hs      |  6 ++++++
 2 files changed, 20 insertions(+)

diff --git a/src/GEval/Core.hs b/src/GEval/Core.hs
index 53527c6..3b85042 100644
--- a/src/GEval/Core.hs
+++ b/src/GEval/Core.hs
@@ -8,6 +8,8 @@
 module GEval.Core
     ( geval,
       gevalCore,
+      gevalCoreOnSingleLines,
+      LineInFile(..),
       Metric(..),
       MetricOrdering(..),
       getMetricOrdering,
@@ -231,6 +233,16 @@ fileAsLineSource :: FilePath -> LineSource (ResourceT IO)
 fileAsLineSource filePath =
   LineSource (CB.sourceFile filePath $= CT.decodeUtf8Lenient =$= CT.lines) filePath 1
 
+gevalCoreOnSingleLines :: Metric -> LineInFile -> LineInFile -> LineInFile -> IO (MetricValue)
+gevalCoreOnSingleLines metric inpLine expLine outLine =
+  gevalCoreOnSources metric (singleLineAsLineSource inpLine)
+                            (singleLineAsLineSource expLine)
+                            (singleLineAsLineSource outLine)
+
+singleLineAsLineSource :: LineInFile -> LineSource (ResourceT IO)
+singleLineAsLineSource (LineInFile filePath lineNo line) =
+  LineSource (CL.sourceList [line]) filePath lineNo
+
 gevalCore :: Metric -> String -> String -> String -> IO (MetricValue)
 gevalCore metric inputFilePath expectedFilePath outFilePath = do
   unlessM (D.doesFileExist expectedFilePath) $ throwM $ NoExpectedFile expectedFilePath
@@ -253,6 +265,8 @@ gevalCoreOnSources RMSE inputLineSource expectedLineSource outLineSource = do
 gevalCoreOnSources metric inputLineSource expectedLineSource outLineSource = do
   gevalCore' metric inputLineSource expectedLineSource outLineSource
 
+data LineInFile = LineInFile FilePath Int Text
+
 gevalCore' :: (MonadIO m, MonadThrow m, MonadBaseControl IO m) => Metric -> LineSource (ResourceT m) -> LineSource (ResourceT m) -> LineSource (ResourceT m) -> m (MetricValue)
 gevalCore' MSE _ = gevalCoreWithoutInput outParser outParser itemError averageC id
   where outParser = getValue . TR.double
diff --git a/test/Spec.hs b/test/Spec.hs
index 2df78aa..8243af1 100644
--- a/test/Spec.hs
+++ b/test/Spec.hs
@@ -175,6 +175,12 @@ main = hspec $ do
                                 ["one", "one"]) `shouldBeAlmost` 0.5
     it "simple test" $ do
       runGEvalTest "map-simple" `shouldReturnAlmost` 0.444444444
+  describe "evaluating single lines" $ do
+    it "RMSE" $ do
+      gevalCoreOnSingleLines RMSE (LineInFile "stub1" 1 "blabla")
+                                  (LineInFile "stub2" 1 "3.4")
+                                  (LineInFile "stub3" 1 "2.6") `shouldReturnAlmost` 0.8
+
 
 neverMatch :: Char -> Int -> Bool
 neverMatch _ _ = False