[SOLVED] CS {-

{-
Module: PruneDepthTest.

Used to test Part I.c.
-}
module PruneDepthTest 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 depth 0.
testTree0 :: TestTree
testTree0 = StateTree 6 []

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

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

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

Pruning at depth 0.
testPruneDepth0 :: Test
testPruneDepth0 = TestCase (assertEqual pruneDepth 0 testTree3 depth 0)
where
depth = stateTreeDepth (pruneDepth 0 testTree3)

Pruning at depth 1.
testPruneDepth1 :: Test
testPruneDepth1 = TestCase (assertEqual pruneDepth 1 testTree3 depth 1)
where
depth = stateTreeDepth (pruneDepth 1 testTree3)

Pruning at depth 2.
testPruneDepth2 :: Test
testPruneDepth2 = TestCase (assertEqual pruneDepth 2 testTree3 depth 2)
where
depth = stateTreeDepth (pruneDepth 2 testTree3)

Pruning at depth 3.
testPruneDepth3 :: Test
testPruneDepth3 = TestCase (assertEqual pruneDepth 3 testTree3 depth 3)
where
depth = stateTreeDepth (pruneDepth 3 testTree3)

All unit tests together.
pruneDepthUnitTests :: Spec
pruneDepthUnitTests = fromHUnitTest $
TestList [
TestLabel testPruneDepth0 testPruneDepth0,
TestLabel testPruneDepth1 testPruneDepth1,
TestLabel testPruneDepth2 testPruneDepth2,
TestLabel testPruneDepth3 testPruneDepth3]

{-
QuickCheck tests.
-}

Pruning at depth 0 should always return trees of depth 0.
testPruneDepthQuickCheck0 :: TestTree -> Expectation
testPruneDepthQuickCheck0 t =
assertEqual stateTreeDepth (pruneDepth 0 t) (stateTreeDepth (pruneDepth 0 t)) 0

Pruning at depth 5 should always return trees of depth less than 5.
testPruneDepthQuickCheck5 :: TestTree -> Expectation
testPruneDepthQuickCheck5 t =
assertLessThanEq stateTreeDepth (pruneDepth 5 t) (stateTreeDepth (pruneDepth 5 t)) 5

Pruning at depth 10 should always return trees of depth less than 10.
testPruneDepthQuickCheck10 :: TestTree -> Expectation
testPruneDepthQuickCheck10 t =
assertLessThanEq stateTreeDepth (pruneDepth 10 t) (stateTreeDepth (pruneDepth 10 t)) 10

All QuickCheck tests together.
pruneDepthQuickCheckTests :: Spec
pruneDepthQuickCheckTests = do
prop testPruneDepthQuickCheck0 testPruneDepthQuickCheck0
prop testPruneDepthQuickCheck5 testPruneDepthQuickCheck5
prop testPruneDepthQuickCheck10 testPruneDepthQuickCheck10