Added a PassTest file intended for tests of the occam passes, and put in it a test for the functions-to-procs pass
This commit is contained in:
parent
6d0cf6464a
commit
4d16320d26
153
PassTest.hs
Normal file
153
PassTest.hs
Normal file
|
@ -0,0 +1,153 @@
|
|||
{-
|
||||
Tock: a compiler for parallel languages
|
||||
Copyright (C) 2007 University of Kent
|
||||
|
||||
This program is free software; you can redistribute it and/or modify it
|
||||
under the terms of the GNU General Public License as published by the
|
||||
Free Software Foundation, either version 2 of the License, or (at your
|
||||
option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful, but
|
||||
WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License along
|
||||
with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
-}
|
||||
|
||||
module PassTest (tests) where
|
||||
|
||||
import Test.HUnit hiding (State)
|
||||
import Control.Monad.State as CSM
|
||||
import qualified Data.Map as Map
|
||||
import qualified AST as A
|
||||
import TestUtil
|
||||
import TreeUtil
|
||||
import CompState
|
||||
import Control.Monad.Error (runErrorT)
|
||||
import Control.Monad.Identity
|
||||
import Types
|
||||
import Pass
|
||||
import Data.Generics
|
||||
import Utils
|
||||
import SimplifyExprs
|
||||
|
||||
valof0 :: A.Structured
|
||||
valof0 = A.OnlyEL m $ A.ExpressionList m [intLiteral 0]
|
||||
|
||||
valofTwo :: String -> String -> A.Structured
|
||||
valofTwo a b = A.OnlyEL m $ A.ExpressionList m [exprVariable a,exprVariable b]
|
||||
|
||||
assertGetItemCast :: Typeable t => String -> Items -> IO t
|
||||
assertGetItemCast k kv
|
||||
= case (Map.lookup k kv) of
|
||||
Nothing -> (assertFailure "Internal error; expected item not present") >> return (undefined)
|
||||
Just (ADI v) -> case (cast v) of
|
||||
Just v' -> return v'
|
||||
Nothing -> (assertFailure $ "Wrong type when casting in assertGetItemCast for key: " ++ k) >> return (undefined)
|
||||
|
||||
-- Given a body, returns a function spec:
|
||||
singleParamFunc :: A.Structured-> A.Specification
|
||||
singleParamFunc body = A.Specification m (simpleName "foo") (A.Function m A.PlainSpec [A.Int] [A.Formal A.ValAbbrev A.Byte (simpleName "param0")] body)
|
||||
|
||||
-- Returns the expected body of the single parameter process (when the function had valof0 as a body)
|
||||
singleParamBodyExp :: Pattern --to match: A.Process
|
||||
singleParamBodyExp = tag2 A.Seq DontCare $
|
||||
tag2 A.OnlyP DontCare $
|
||||
tag3 A.Assign DontCare [tag2 A.Variable DontCare (Named "ret0" DontCare)] $ tag2 A.ExpressionList DontCare [intLiteral 0]
|
||||
|
||||
-- Returns the expected specification type of the single parameter process
|
||||
singleParamSpecExp :: Pattern -> Pattern --to match: A.SpecType
|
||||
singleParamSpecExp body = tag4 A.Proc DontCare A.PlainSpec [tag3 A.Formal A.ValAbbrev A.Byte (simpleName "param0"), tag3 A.Formal A.Abbrev A.Int (Named "ret0" DontCare)] body
|
||||
|
||||
-- | Tests a function with a single return, and a single parameter.
|
||||
testFunctionsToProcs0 :: Test
|
||||
testFunctionsToProcs0 = testPassWithItemsStateCheck "testFunctionsToProcs0" exp (functionsToProcs orig) (return ()) check
|
||||
where
|
||||
orig = singleParamFunc valof0
|
||||
exp = tag3 A.Specification DontCare (simpleName "foo") procSpec
|
||||
procSpec = singleParamSpecExp singleParamBodyExp
|
||||
--check return parameters were defined:
|
||||
check (items,state) = do ret0 <- ((assertGetItemCast "ret0" items) :: IO A.Name)
|
||||
assertVarDef "testFunctionsToProcs0" state (A.nameName ret0) $
|
||||
tag7 A.NameDef DontCare (A.nameName ret0) (A.nameName ret0) A.VariableName (A.Declaration m A.Int) A.Abbrev A.Unplaced
|
||||
--check proc was defined:
|
||||
assertVarDef "testFunctionsToProcs0" state "foo" $
|
||||
tag7 A.NameDef DontCare ("foo") ("foo") A.ProcName procSpec A.Original A.Unplaced
|
||||
--check csFunctionReturns was changed:
|
||||
assertEqual "testFunctionsToProcs0" (Just [A.Int]) (Map.lookup "foo" (csFunctionReturns state))
|
||||
|
||||
-- | Tests a function with multiple returns, and multiple parameters.
|
||||
testFunctionsToProcs1 :: Test
|
||||
testFunctionsToProcs1 = testPassWithItemsStateCheck "testFunctionsToProcs1 A" exp (functionsToProcs orig) (return ()) check
|
||||
where
|
||||
orig = A.Specification m (simpleName "foo") (A.Function m A.PlainSpec [A.Int,A.Real32]
|
||||
[A.Formal A.ValAbbrev A.Byte (simpleName "param0"),A.Formal A.Abbrev A.Real32 (simpleName "param1")] (valofTwo "param0" "param1"))
|
||||
exp = tag3 A.Specification DontCare (simpleName "foo") procBody
|
||||
procBody = tag4 A.Proc DontCare A.PlainSpec [tag3 A.Formal A.ValAbbrev A.Byte (simpleName "param0"),
|
||||
tag3 A.Formal A.Abbrev A.Real32 (simpleName "param1"),
|
||||
tag3 A.Formal A.Abbrev A.Int (Named "ret0" DontCare),
|
||||
tag3 A.Formal A.Abbrev A.Real32 (Named "ret1" DontCare)] $
|
||||
tag2 A.Seq DontCare $
|
||||
tag2 A.OnlyP DontCare $
|
||||
tag3 A.Assign DontCare [tag2 A.Variable DontCare (Named "ret0" DontCare),tag2 A.Variable DontCare (Named "ret1" DontCare)] $
|
||||
tag2 A.ExpressionList DontCare [exprVariable "param0",exprVariable "param1"]
|
||||
--check return parameters were defined:
|
||||
check (items,state) = do ret0 <- ((assertGetItemCast "ret0" items) :: IO A.Name)
|
||||
ret1 <- ((assertGetItemCast "ret1" items) :: IO A.Name)
|
||||
assertVarDef "testFunctionsToProcs1 B" state (A.nameName ret0) $
|
||||
tag7 A.NameDef DontCare (A.nameName ret0) (A.nameName ret0) A.VariableName (A.Declaration m A.Int) A.Abbrev A.Unplaced
|
||||
assertVarDef "testFunctionsToProcs1 C" state (A.nameName ret1) $
|
||||
tag7 A.NameDef DontCare (A.nameName ret1) (A.nameName ret1) A.VariableName (A.Declaration m A.Real32) A.Abbrev A.Unplaced
|
||||
--check proc was defined:
|
||||
assertVarDef "testFunctionsToProcs1 D" state "foo" $
|
||||
tag7 A.NameDef DontCare ("foo") ("foo") A.ProcName procBody A.Original A.Unplaced
|
||||
--check csFunctionReturns was changed:
|
||||
assertEqual "testFunctionsToProcs1 E" (Just [A.Int,A.Real32]) (Map.lookup "foo" (csFunctionReturns state))
|
||||
|
||||
-- | Tests a function that contains a function.
|
||||
-- Currently I have chosen to put DontCare for the body of the function as stored in the NameDef.
|
||||
-- This behaviour is not too important, and may change at a later date.
|
||||
testFunctionsToProcs2 :: Test
|
||||
testFunctionsToProcs2 = testPassWithItemsStateCheck "testFunctionsToProcs2 A" exp (functionsToProcs orig) (return ()) check
|
||||
where
|
||||
orig = A.Specification m (simpleName "fooOuter") (A.Function m A.PlainSpec [A.Int] [A.Formal A.ValAbbrev A.Byte (simpleName "paramOuter0")] $
|
||||
A.Spec m (singleParamFunc valof0) valof0)
|
||||
exp = tag3 A.Specification DontCare (simpleName "fooOuter") procBodyOuter
|
||||
procHeader body = tag4 A.Proc DontCare A.PlainSpec [tag3 A.Formal A.ValAbbrev A.Byte (simpleName "paramOuter0"), tag3 A.Formal A.Abbrev A.Int (Named "retOuter0" DontCare)] body
|
||||
procBodyOuter = procHeader $
|
||||
tag2 A.Seq DontCare $
|
||||
tag3 A.Spec DontCare (tag3 A.Specification DontCare (simpleName "foo") (singleParamSpecExp singleParamBodyExp)) $
|
||||
tag2 A.OnlyP DontCare $
|
||||
tag3 A.Assign DontCare [tag2 A.Variable DontCare (Named "retOuter0" DontCare)] $ tag2 A.ExpressionList DontCare [intLiteral 0]
|
||||
|
||||
|
||||
--check return parameters were defined:
|
||||
check (items,state) = do retOuter0 <- ((assertGetItemCast "retOuter0" items) :: IO A.Name)
|
||||
ret0 <- ((assertGetItemCast "ret0" items) :: IO A.Name)
|
||||
assertVarDef "testFunctionsToProcs2 B" state (A.nameName ret0) $
|
||||
tag7 A.NameDef DontCare (A.nameName ret0) (A.nameName ret0) A.VariableName (A.Declaration m A.Int) A.Abbrev A.Unplaced
|
||||
assertVarDef "testFunctionsToProcs2 C" state (A.nameName retOuter0) $
|
||||
tag7 A.NameDef DontCare (A.nameName retOuter0) (A.nameName retOuter0) A.VariableName (A.Declaration m A.Int) A.Abbrev A.Unplaced
|
||||
--check proc was defined:
|
||||
assertVarDef "testFunctionsToProcs2 D" state "foo" $
|
||||
tag7 A.NameDef DontCare ("foo") ("foo") A.ProcName (singleParamSpecExp DontCare) A.Original A.Unplaced
|
||||
assertVarDef "testFunctionsToProcs2 E" state "fooOuter" $
|
||||
tag7 A.NameDef DontCare ("fooOuter") ("fooOuter") A.ProcName (procHeader DontCare) A.Original A.Unplaced
|
||||
--check csFunctionReturns was changed:
|
||||
assertEqual "testFunctionsToProcs2 F" (Just [A.Int]) (Map.lookup "foo" (csFunctionReturns state))
|
||||
assertEqual "testFunctionsToProcs2 G" (Just [A.Int]) (Map.lookup "fooOuter" (csFunctionReturns state))
|
||||
|
||||
|
||||
|
||||
--Returns the list of tests:
|
||||
tests :: Test
|
||||
tests = TestList
|
||||
[
|
||||
testFunctionsToProcs0
|
||||
,testFunctionsToProcs1
|
||||
,testFunctionsToProcs2
|
||||
]
|
||||
|
||||
|
|
@ -17,7 +17,7 @@ with this program. If not, see <http://www.gnu.org/licenses/>.
|
|||
-}
|
||||
|
||||
-- | Simplify expressions in the AST.
|
||||
module SimplifyExprs (simplifyExprs) where
|
||||
module SimplifyExprs where
|
||||
|
||||
import Control.Monad.State
|
||||
import Data.Generics
|
||||
|
|
|
@ -21,8 +21,15 @@ module TestMain () where
|
|||
import qualified RainParseTest (tests)
|
||||
import qualified RainPassTest (tests)
|
||||
import qualified UsageCheckTest (tests)
|
||||
import qualified PassTest (tests)
|
||||
import Test.HUnit
|
||||
|
||||
main :: IO ()
|
||||
main = do runTestTT $ TestList [RainParseTest.tests,RainPassTest.tests,UsageCheckTest.tests]
|
||||
main = do runTestTT $ TestList
|
||||
[
|
||||
PassTest.tests
|
||||
,RainParseTest.tests
|
||||
,RainPassTest.tests
|
||||
,UsageCheckTest.tests
|
||||
]
|
||||
return ()
|
||||
|
|
Loading…
Reference in New Issue
Block a user