[SOLVED] CS {-

{-
Module: PruneBreadthTest.

Used to test Part I.d.
-}
module PruneBreadthTest where

import Test.Hspec
import Test.Hspec.Contrib.HUnit (fromHUnitTest)
import Test.Hspec.QuickCheck
import Test.HUnit
import Test.QuickCheck

import Types
import Players.Minimax
import Util.Assertion
import Util.StateTreeInstances

type TestTree = StateTree Int Int

{-
Unit tests.
-}

Tree of breadth 0.
testTree0 :: TestTree
testTree0 = StateTree 6 []

Tree of breadth 1.
testTree1 :: TestTree
testTree1 = StateTree 8 [(1, testTree0)]

Tree of breadth 2.
testTree2 :: TestTree
testTree2 = StateTree 4 [(2, testTree1), (3, testTree0)]

Tree of breadth 3.
testTree3 :: TestTree
testTree3 = StateTree 1[(5, testTree0), (7, testTree2), (1, testTree1)]

Pruning at breadth 0.
testPruneBreadth0 :: Test
testPruneBreadth0 = TestCase (assertEqual pruneBreadth 0 testTree3 breadth 0)
where
breadth = stateTreeBreadth (pruneBreadth 0 testTree3)

Pruning at breadth 1.
testPruneBreadth1 :: Test
testPruneBreadth1 = TestCase (assertEqual pruneBreadth 1 testTree3 breadth 1)
where
breadth = stateTreeBreadth (pruneBreadth 1 testTree3)

Pruning at breadth 2.
testPruneBreadth2 :: Test
testPruneBreadth2 = TestCase (assertEqual pruneBreadth 2 testTree3 breadth 2)
where
breadth = stateTreeBreadth (pruneBreadth 2 testTree3)

Pruning at breadth 3.
testPruneBreadth3 :: Test
testPruneBreadth3 = TestCase (assertEqual pruneBreadth 3 testTree3 breadth 3)
where
breadth = stateTreeBreadth (pruneBreadth 3 testTree3)

All unit tests together.
pruneBreadthUnitTests :: Spec
pruneBreadthUnitTests = fromHUnitTest $
TestList [
TestLabel testPruneBreadth0 testPruneBreadth0,
TestLabel testPruneBreadth1 testPruneBreadth1,
TestLabel testPruneBreadth2 testPruneBreadth2,
TestLabel testPruneBreadth3 testPruneBreadth3]

{-
QuickCheck tests.
-}

Pruning at breadth 0 should always return trees of breadth 0.
testPruneBreadthQuickCheck0 :: TestTree -> Expectation
testPruneBreadthQuickCheck0 t =
assertEqual stateTreeBreadth (pruneBreadth 0 t) (stateTreeBreadth (pruneBreadth 0 t)) 0

Pruning at breadth 5 should always return trees of breadth less than 5.
testPruneBreadthQuickCheck5 :: TestTree -> Expectation
testPruneBreadthQuickCheck5 t =
assertLessThanEq stateTreeBreadth (pruneBreadth 5 t) (stateTreeBreadth (pruneBreadth 5 t)) 5

Pruning at breadth 10 should always return trees of breadth less than 10.
testPruneBreadthQuickCheck10 :: TestTree -> Expectation
testPruneBreadthQuickCheck10 t =
assertLessThanEq stateTreeBreadth (pruneBreadth 10 t) (stateTreeBreadth (pruneBreadth 10 t)) 10

All QuickCheck tests together.
pruneBreadthQuickCheckTests :: Spec
pruneBreadthQuickCheckTests = do
prop testPruneBreadthQuickCheck0 testPruneBreadthQuickCheck0
prop testPruneBreadthQuickCheck5 testPruneBreadthQuickCheck5
prop testPruneBreadthQuickCheck10 testPruneBreadthQuickCheck10