#!/usr/bin/runghc

import Test.QuickCheck
import TranspositionDecrypt
import TranspositionEncrypt

-- |
-- >>> quickCheck test
-- +++ OK, passed 100 tests.
test :: String -> Gen Bool
test message = do
  key <- choose (1, length message) :: Gen Int
  return (messagePad key message == (decryptString (numOfColumns key message) . encryptString key) message)

main :: IO ()
main = quickCheck test