add tests

anoma · Feb 1, 2023 · 005885a · 005885a
1 parent fa3bf3c
commit 005885a
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 4 deletions.
diff --git a/src/Juvix/Compiler/Core/Transformation/LambdaLetRecLifting.hs b/src/Juvix/Compiler/Core/Transformation/LambdaLetRecLifting.hs
@@ -185,3 +185,6 @@ nodeIsLetRecLifted = not . hasLetRecs
 
 isLifted :: InfoTable -> Bool
 isLifted = all nodeIsLifted . (^. identContext)
+
+isLetRecLifted :: InfoTable -> Bool
+isLetRecLifted = all nodeIsLetRecLifted . (^. identContext)
diff --git a/test/Core/Transformation/Lifting.hs b/test/Core/Transformation/Lifting.hs
@@ -6,10 +6,12 @@ import Core.Transformation.Base
 import Juvix.Compiler.Core.Transformation
 
 allTests :: TestTree
-allTests = testGroup "Lambda and LetRec lifting" (map liftTest Eval.tests)
-
-pipe :: [TransformationId]
-pipe = [LambdaLetRecLifting]
+allTests =
+  testGroup
+    "Lifting"
+    [ testGroup "Lambda and LetRec lifting" (map liftTest Eval.tests),
+      testGroup "Only LetRec lifting" (map letRecLiftTest Eval.tests)
+    ]
 
 liftTest :: Eval.PosTest -> TestTree
 liftTest _testEval =
@@ -19,3 +21,18 @@ liftTest _testEval =
         _testAssertion = \i -> unless (isLifted i) (error "not lambda lifted"),
         _testEval
       }
+  where
+    pipe :: [TransformationId]
+    pipe = [LambdaLetRecLifting]
+
+letRecLiftTest :: Eval.PosTest -> TestTree
+letRecLiftTest _testEval =
+  fromTest
+    Test
+      { _testTransformations = pipe,
+        _testAssertion = \i -> unless (isLetRecLifted i) (error "not letrec lifted"),
+        _testEval
+      }
+  where
+    pipe :: [TransformationId]
+    pipe = [LetRecLifting]