Strip annotations in the tests.

Author: robrix

semantic-core/src/Data/Core.hs

@@ -28,6 +28,7 @@ module Data.Core
 , ann
 , annWith
 , instantiate
+, stripAnnotations
 ) where
 
 import Control.Applicative (Alternative (..))
@@ -189,6 +190,12 @@ annWith :: (Carrier sig m, Member Core sig) => CallStack -> m a -> m a
 annWith callStack = maybe id (fmap send . Ann) (stackLoc callStack)
 
 
+stripAnnotations :: (Member Core sig, Syntax sig) => Term sig a -> Term sig a
+stripAnnotations = iter id alg Var Var
+  where alg t | Just c <- prj t, Ann _ b <- c = b
+              | otherwise                     = Term t
+
+
 instance Syntax Core where
   foldSyntax go k h = \case
     Let a -> Let a

semantic-core/test/Spec.hs

@@ -100,7 +100,7 @@ parserSpecs = testGroup "Parsing: simple specs"
   ]
 
 assert_roundtrips :: File (Term Core User) -> Assertion
-assert_roundtrips (File _ core) = parseEither Parse.core (showCore core) @?= Right core
+assert_roundtrips (File _ core) = parseEither Parse.core (showCore core) @?= Right (stripAnnotations core)
 
 parserExamples :: TestTree
 parserExamples = testGroup "Parsing: Eval.hs examples"