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

Use mkMapTerm in tests

parent fb8b17c4
{-# LANGUAGE TypeApplications #-} {-# LANGUAGE TypeApplications #-}
module Test.Ngrams.Query (tests) where module Test.Ngrams.Query (tests, mkMapTerm) where
import Control.Monad import Control.Monad
import Data.Coerce import Data.Coerce
......
...@@ -20,6 +20,7 @@ import Gargantext.Database.Schema.Context ...@@ -20,6 +20,7 @@ import Gargantext.Database.Schema.Context
import Test.HUnit import Test.HUnit
import Test.Hspec import Test.Hspec
import Test.Instances () import Test.Instances ()
import Test.Ngrams.Query (mkMapTerm)
import Test.QuickCheck import Test.QuickCheck
import Test.QuickCheck qualified as QC import Test.QuickCheck qualified as QC
...@@ -101,28 +102,12 @@ tests = describe "Ngrams" $ do ...@@ -101,28 +102,12 @@ tests = describe "Ngrams" $ do
testFilterNgramsNodesEmptyQuery :: Assertion testFilterNgramsNodesEmptyQuery :: Assertion
testFilterNgramsNodesEmptyQuery = do testFilterNgramsNodesEmptyQuery = do
let input = Map.fromList [ let input = Map.fromList [
("car", NgramsElement { _ne_ngrams = "car" ("vehicle", mkMapTerm "vehicle" & ne_children .~ mSetFromList ["car"])
, _ne_size = 1 , ("car", mkMapTerm "car" & ne_root .~ Just "vehicle"
, _ne_list = MapTerm & ne_parent .~ Just "vehicle"
, _ne_occurrences = mempty & ne_children .~ mSetFromList ["ford"])
, _ne_root = Just "vehicle" , ("ford", mkMapTerm "ford" & ne_root .~ Just "vehicle"
, _ne_parent = Just "vehicle" & ne_parent .~ Just "car")
, _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"]})
] ]
let actual = filterNgramsNodes (Just MapTerm) Nothing Nothing (const True) input let actual = filterNgramsNodes (Just MapTerm) Nothing Nothing (const True) input
actual @?= (Set.fromList $ Map.elems 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