import Data.Multimap (Multimap)
import qualified Data.Multimap as Mmap
import Data.Multimap.ListMultimap (ListMultimap)
import qualified Data.Multimap.ListMultimap as Mmap
import Data.Multiset (Multiset)
import qualified Data.Multiset as Mset
import Test.Hspec
import Test.Hspec.QuickCheck

main :: IO ()
main = hspec $ do
  describe "Multiset" $ do
    it "Empty should have size 0" $ do
      Mset.size Mset.empty `shouldBe` 0
    it "Singleton should have size 1" $ do
      Mset.size (Mset.singleton 'a') `shouldBe` 1
    it "From list" $ do
      Mset.fromList ['a', 'b', 'a'] `shouldBe` Mset.fromCountsList [('a', 2), ('b', 1)]
    it "Incr" $ do
      let m = Mset.fromList ['a']
      Mset.insert 'a' m `shouldBe` Mset.mapCounts (*2) m
  describe "ListMultimap" $ do
    let mm = Mmap.fromCollectionsList [('a', [2])]
        mm' = Mmap.fromCollectionsList [('a', [1,2])]
    it "cons" $ do
      Mmap.cons 'a' 1 mm `shouldBe` mm'
    it "uncon" $ do
      Mmap.uncons 'a' mm `shouldBe` Just (2, Mmap.empty)
      Mmap.uncons 'a' mm' `shouldBe` Just (1, mm)
      Mmap.uncons 'b' mm `shouldBe` Nothing