46 lines
1.2 KiB
Haskell
46 lines
1.2 KiB
Haskell
|
module Test.Data.Trie where
|
||
|
|
|
||
|
|
import Test.Tasty
|
||
|
|
import Test.Tasty.Hedgehog (testProperty)
|
||
|
|
import Hedgehog
|
||
|
|
import Data.Monoid (Sum(..))
|
||
|
|
import Data.ByteString (ByteString)
|
||
|
|
import Data.Map (Map)
|
||
|
|
import qualified Hedgehog.Gen as Gen
|
||
|
|
import qualified Hedgehog.Range as Range
|
||
|
|
import qualified Data.Trie as T
|
||
|
|
import qualified Data.Map.Strict as M
|
||
|
|
|
||
|
|
genMap :: Gen (Map ByteString (Sum Int))
|
||
|
|
genMap = Gen.map (Range.linear 0 100) genT
|
||
|
|
where
|
||
|
|
genT = (,) <$> Gen.bytes (Range.linear 0 20) <*> fmap Sum (Gen.integral (Range.linear 0 10))
|
||
|
|
|
||
|
|
propIsomorphic :: Property
|
||
|
|
propIsomorphic = property $ do
|
||
|
|
m <- forAll genMap
|
||
|
|
let wanted = m
|
||
|
|
got = M.fromList . T.toList . T.fromList . M.toList $ wanted
|
||
|
|
wanted === got
|
||
|
|
|
||
|
|
propElems :: Property
|
||
|
|
propElems = property $ do
|
||
|
|
m <- forAll genMap
|
||
|
|
let wanted = M.elems m
|
||
|
|
got = T.elems . T.fromList . M.toList $ m
|
||
|
|
wanted === got
|
||
|
|
|
||
|
|
propKeys :: Property
|
||
|
|
propKeys = property $ do
|
||
|
|
m <- forAll genMap
|
||
|
|
let wanted = M.keys m
|
||
|
|
got = T.keys . T.fromList . M.toList $ m
|
||
|
|
wanted === got
|
||
|
|
|
||
|
|
tests :: TestTree
|
||
|
|
tests = testGroup "Data.Trie"
|
||
|
|
[ testProperty "isomorphic to Map" $ propIsomorphic
|
||
|
|
, testProperty "isomorphic elems" $ propElems
|
||
|
|
, testProperty "isomorphic keys" $ propKeys
|
||
|
|
]
|