Improve tests

1 files changed, 9 insertions, 3 deletions

diff --git a/test/Spec.hs b/test/Spec.hs
index cbe73d7..0869b55 100644
--- a/test/Spec.hs
+++ b/test/Spec.hs
@@ -2,13 +2,19 @@
 import Test.HUnit
 import Control.Monad.Except
 import Tokenizer ( tokenize )
+import Parser ( parse )
 import TestUtils
+import Utils
 
-test1 = (runLContext $ tokenize "+ 1 2")
-    >>= assertEqual "tokenize \"+ 1 2\"" ["+", "1", "2"]
+test1 = (runLContext $ tokenize "(+ 1 (- 10 5))")
+    >>= assertEqual "tokenize" ["(", "+", "1", "(", "-", "10", "5", ")", ")"]
+
+test2 = (runLContext $ parse ["(", "+", "1", "2", ")"])
+    >>= assertEqual "parse" [ASTFunctionCall [ASTSymbol "+", ASTInteger 1, ASTInteger 2]]
 
 tests = TestList $ map TestCase [
-    test1
+    test1,
+    test2
     ]
 
 main :: IO ()