diff --git a/test/Spec.hs b/test/Spec.hs index bff5723..b22f420 100644 --- a/test/Spec.hs +++ b/test/Spec.hs @@ -135,6 +135,8 @@ main = hspec $ do runGEvalTest "accuracy-probs" `shouldReturnAlmost` 0.4 it "sorted" $ runGEvalTest "accuracy-on-sorted" `shouldReturnAlmost` 0.75 + it "with filtering" $ + runGEvalTest "accuracy-filtering" `shouldReturnAlmost` 1.0 describe "F-measure" $ do it "simple example" $ runGEvalTest "f-measure-simple" `shouldReturnAlmost` 0.57142857 diff --git a/test/accuracy-filtering/accuracy-filtering-solution/test-A/out.tsv b/test/accuracy-filtering/accuracy-filtering-solution/test-A/out.tsv new file mode 100644 index 0000000..1295207 --- /dev/null +++ b/test/accuracy-filtering/accuracy-filtering-solution/test-A/out.tsv @@ -0,0 +1,5 @@ +foo +foo +bar +bar +foo diff --git a/test/accuracy-filtering/accuracy-filtering/config.txt b/test/accuracy-filtering/accuracy-filtering/config.txt new file mode 100644 index 0000000..f61b446 --- /dev/null +++ b/test/accuracy-filtering/accuracy-filtering/config.txt @@ -0,0 +1 @@ +--metric Accuracy:f diff --git a/test/accuracy-filtering/accuracy-filtering/test-A/expected.tsv b/test/accuracy-filtering/accuracy-filtering/test-A/expected.tsv new file mode 100644 index 0000000..a792a9d --- /dev/null +++ b/test/accuracy-filtering/accuracy-filtering/test-A/expected.tsv @@ -0,0 +1,5 @@ +foo +bar +bar +bar +bar diff --git a/test/accuracy-filtering/accuracy-filtering/test-A/in.tsv b/test/accuracy-filtering/accuracy-filtering/test-A/in.tsv new file mode 100644 index 0000000..d4cfa74 --- /dev/null +++ b/test/accuracy-filtering/accuracy-filtering/test-A/in.tsv @@ -0,0 +1,5 @@ +1 yep 111 +1 nope 2323 +3 yep 32323 +2 yep 222 +11 nope sdsds