Add compute tests

X-R-G-B · Dec 8, 2023 · 45b9331 · 45b9331
1 parent e0aac41
commit 45b9331
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/test/Spec.hs b/test/Spec.hs
@@ -8,7 +8,7 @@ main :: IO ()
 main = defaultMainWithIngredients (htmlRunner : defaultIngredients) tests
 
 tests :: TestTree
-tests = testGroup "Tests" [unitTests]
+tests = testGroup "Tests" [unitTests, computeTests]
 
 unitTests :: TestTree
 unitTests = testGroup "Unit tests"
@@ -19,3 +19,13 @@ unitTests = testGroup "Unit tests"
   , testCase "Basic AST creation 2" $
       assertEqual "42" (Leaf (Number 42)) (Leaf (Number 42))
   ]
+
+computeTests :: TestTree
+computeTests = testGroup "Compute tests"
+  [ testCase "Basic compute 0" $
+      assertEqual "define x 42" (Atom 42) (computeAst (Node "define" [Leaf (Symbol "x"), Leaf (Number 42)]))
+  , testCase "Basic compute 1" $
+      assertEqual "define x 42 and do x + x" (Leaf (Number 42)) (compute (Node "define" [Leaf (Symbol "x"), Leaf (Number 42)]))
+  , testCase "Basic compute 2" $
+      assertEqual "42 + 42" (Atom 84) (computeAst (Node "+" [Leaf (Number 42), Leaf (Number 42)]))
+  ]