Instances.hs

module Tests.Instances where

    import Wordify.Rules.Tile
    import Test.QuickCheck (Arbitrary, arbitrary, listOf, (==>), sized, oneof, choose, Gen, elements)
    import Wordify.Rules.LetterBag
    import Wordify.Rules.Pos
    import Data.Char
    import Wordify.Rules.Pos.Internal
    import Wordify.Rules.Square
    import Data.Map
    import Wordify.Rules.Board
    import Wordify.Rules.Board.Internal
    import System.Random
    import Wordify.Rules.LetterBag.Internal

    instance Arbitrary Tile where
        arbitrary = do
            chr <- arbitrary :: Gen Char
            value <- arbitrary :: Gen Int
            tile <- elements [Letter chr value, Blank Nothing]
            return tile

    instance Arbitrary LetterBag where
        arbitrary = do
           tiles <- listOf (arbitrary :: Gen Tile)
           seed <- arbitrary :: Gen Int
           let generator = mkStdGen seed
           return $ makeBagUsingGenerator tiles generator

    instance Arbitrary Pos where
        arbitrary = do
           x <- choose (1,15)
           y <- choose (1,15)
           let gridCo = [chr (x + 64)] ++ (show y)
           return $ Pos x y gridCo

    instance Arbitrary Square where
        arbitrary = do
            tile <- arbitrary :: Gen Tile
            square <- elements [Normal (Just tile), Normal Nothing, DoubleLetter (Just tile), DoubleLetter Nothing, DoubleWord (Just tile),
             DoubleWord Nothing, TripleLetter (Just tile), TripleLetter Nothing, TripleWord (Just tile), TripleWord Nothing]
            return square

    instance Arbitrary Board where
        arbitrary = do
            let Board squares = emptyBoard
            let originalSquares = toList squares
            positions <- listOf (arbitrary :: Gen Pos)
            squares <- listOf (arbitrary :: Gen Square)
            return $ Board $ fromList $ originalSquares ++ (zip positions squares)