Test.hs

import Typechecker
import Types
import Control.Monad
import Data.Either
import qualified Data.Map as M
import qualified Data.Set as S

testExprs :: [MoleculeExpr]
testExprs = [
    EApp (EAbs "x" (EVar "x" :+: EVar "x")) (EInt 10)
  , EAbs "x" (EVar "x" :+: EVar "x")
  , EApp (EAbs "x" (EVar "x")) (EInt 8)
  , EApp (EApp (EAbs "x" (EAbs "y" (EVar "x" :+: EVar "y"))) (EInt 10)) (EInt 10)
  , EInt 10
  , ETrue
  , EFalse
  , EApp (EAbs "x" ETrue) (EInt 10)
  , EAbs "x" ((EVar "x") :+: EInt 80)
  , EAbs "x" ((EVar "x") :|: ETrue)
  , EApp (EAbs "x" ETrue) (EVar "x")
  , EApp (EAbs "x" $ EApp (EVar "x") (EInt 10)) (EAbs "y" (EVar "y"))
  ]

failExprs :: [MoleculeExpr]
failExprs = [
    ETrue :+: EInt 10
  , EAbs "a" (EVar "a" :|: EVar "b")
  , EVar "y"
  , EApp (EApp (EAbs "x" (EAbs "y" (EVar "x" :+: EVar "y"))) (EInt 10)) (ETrue)
  , EAbs "x" (EVar "x")
  , EAbs "x" ETrue
  ]

report :: MoleculeExpr -> Either MoleculeError MoleculeType -> String
report e (Right t) = concat [
    "\nSuccess:\n"
  , show e
  , "\nhas type: "
  , show t
  ]
report e (Left err) = concat [
    "\nFailure:\n"
  , show e
  , "\nfailed with error:\n"
  , show err
  ]

test = do
  let rights = map typecheck testExprs
      lefts  = map typecheck failExprs
  putStrLn "rights:"
  mapM_ putStrLn $ zipWith report testExprs rights
  putStrLn "lefts:"
  mapM_ putStrLn $ zipWith report failExprs lefts
  guard (all isRight rights)
  guard (all isLeft lefts)
  putStrLn "\ntests passing."