mirror of
https://github.com/github/semantic.git
synced 2024-12-30 18:36:27 +03:00
29 lines
1.2 KiB
Haskell
29 lines
1.2 KiB
Haskell
{-# LANGUAGE DataKinds #-}
|
|
module Data.RandomWalkSimilarity.Spec where
|
|
|
|
import Data.DList as DList hiding (toList)
|
|
import Data.RandomWalkSimilarity
|
|
import Data.RandomWalkSimilarity.Arbitrary ()
|
|
import Data.Record
|
|
import Data.Record.Arbitrary ()
|
|
import Data.String
|
|
import Prologue
|
|
import Term
|
|
import Term.Arbitrary
|
|
import Test.Hspec
|
|
import Test.Hspec.QuickCheck
|
|
import Test.QuickCheck hiding (Fixed)
|
|
|
|
spec :: Spec
|
|
spec = parallel $ do
|
|
describe "pqGrams" $ do
|
|
prop "produces grams with stems of the specified length" $ forAll (arbitrary `suchThat` (\ (_, p, q) -> p > 0 && q > 0)) $
|
|
\ (term, p, q) -> (pqGrams p q (toTerm term :: Term String (Record '[String])) :: Bag (Gram String)) `shouldSatisfy` all ((== p) . length . stem)
|
|
|
|
prop "produces grams with bases of the specified length" $ forAll (arbitrary `suchThat` (\ (_, p, q) -> p > 0 && q > 0)) $
|
|
\ (term, p, q) -> (pqGrams p q (toTerm term :: Term String (Record '[String])) :: Bag (Gram String)) `shouldSatisfy` all ((== q) . length . base)
|
|
|
|
describe "featureVector" $ do
|
|
prop "produces a vector of the specified dimension" $ forAll (arbitrary `suchThat` ((> 0) . snd)) $
|
|
\ (grams, d) -> length (featureVector (fromList (grams :: [Gram String])) d) `shouldBe` d
|