grisette-0.2.0.0: test/Grisette/Core/Control/Monad/UnionMTests.hs

{-# LANGUAGE OverloadedStrings #-}

module Grisette.Core.Control.Monad.UnionMTests where

import Grisette.Core.Control.Monad.UnionM
import Grisette.Core.Data.Class.GenSym
import Grisette.Core.Data.Class.SimpleMergeable
import Grisette.Core.Data.Class.Solvable
import Test.Tasty
import Test.Tasty.HUnit

unionMTests :: TestTree
unionMTests =
  testGroup
    "UnionMTests"
    [ testCase "unionSize" $ do
        unionSize (single 1 :: UnionM Integer) @=? 1
        unionSize (mrgIf (ssym "a") (single 1) (single 2) :: UnionM Integer) @=? 2
        unionSize (choose [1, 2, 3, 4, 5, 6, 7] "a" :: UnionM Integer) @=? 7
    ]