Commit 0314bf55 authored by Alfredo Di Napoli's avatar Alfredo Di Napoli

Use mkMapTerm in tests

parent fb8b17c4
{-# LANGUAGE TypeApplications #-}
module Test.Ngrams.Query (tests) where
module Test.Ngrams.Query (tests, mkMapTerm) where
import Control.Monad
import Data.Coerce
......
......@@ -20,6 +20,7 @@ import Gargantext.Database.Schema.Context
import Test.HUnit
import Test.Hspec
import Test.Instances ()
import Test.Ngrams.Query (mkMapTerm)
import Test.QuickCheck
import Test.QuickCheck qualified as QC
......@@ -101,28 +102,12 @@ tests = describe "Ngrams" $ do
testFilterNgramsNodesEmptyQuery :: Assertion
testFilterNgramsNodesEmptyQuery = do
let input = Map.fromList [
("car", NgramsElement { _ne_ngrams = "car"
, _ne_size = 1
, _ne_list = MapTerm
, _ne_occurrences = mempty
, _ne_root = Just "vehicle"
, _ne_parent = Just "vehicle"
, _ne_children = mSetFromList ["ford"]
})
, ("ford", NgramsElement { _ne_ngrams = "ford"
, _ne_size = 1
, _ne_list = MapTerm
, _ne_occurrences = mempty
, _ne_root = Just "vehicle"
, _ne_parent = Just "car"
, _ne_children = mempty})
, ("vehicle", NgramsElement { _ne_ngrams = "vehicle"
, _ne_size = 1
, _ne_list = MapTerm
, _ne_occurrences = mempty
, _ne_root = Nothing
, _ne_parent = Nothing
, _ne_children = mSetFromList ["car"]})
("vehicle", mkMapTerm "vehicle" & ne_children .~ mSetFromList ["car"])
, ("car", mkMapTerm "car" & ne_root .~ Just "vehicle"
& ne_parent .~ Just "vehicle"
& ne_children .~ mSetFromList ["ford"])
, ("ford", mkMapTerm "ford" & ne_root .~ Just "vehicle"
& ne_parent .~ Just "car")
]
let actual = filterNgramsNodes (Just MapTerm) Nothing Nothing (const True) input
actual @?= (Set.fromList $ Map.elems input)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment