From 723fa29e5ceb2f7ee4a13632ccfe3acf5e8808c2 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jans.tuomi@gmail.com>
Date: Thu, 8 Dec 2022 13:16:16 +0200
Subject: Add :tag AST, refactor core to use it

---
 test/Spec.hs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

(limited to 'test/Spec.hs')

diff --git a/test/Spec.hs b/test/Spec.hs
index e27159c..6cb9fac 100644
--- a/test/Spec.hs
+++ b/test/Spec.hs
@@ -89,7 +89,7 @@ e2eTests = testGroup "e2e" [
 
      do let env = makeEnv [builtinAdd2, builtinSubtract2, ("_", astHole)]
         let script1 = "(let memo fibo (\\[n]\
-                      \(match n\
+                      \  (match n\
                       \    0  0\
                       \    1  1\
                       \    _  (+ (fibo (- n 1)) (fibo (- n 2))))))\
@@ -99,6 +99,16 @@ e2eTests = testGroup "e2e" [
             runInlineScript "<test>" script1
 
         let expectedLastAST = astInteger 12586269025
+        assertEqual "" expectedLastAST (last gotASTs),
+
+     do let env = makeEnv [builtinAdd2, builtinEq2]
+        let script1 = "(let a :thing)\
+                      \(let b :thing)\
+                      \(eq? a b)"
+        (gotASTs, _) <- expectSuccessL env $
+            runInlineScript "<test>" script1
+
+        let expectedLastAST = astBoolean True
         assertEqual "" expectedLastAST (last gotASTs)
     ]
 
-- 
cgit v1.3