add some simple tests for the evaluator and fix a bug they found.

swolchok · swolchok · commit eb9af00ceca0 · 2010-01-06T00:44:03.000-05:00
diff --git a/eval.py b/eval.py
@@ -23,7 +23,7 @@ def ac_global_name(s):
                      map)
 
 arc_globals = dict((ac_global_name(Symbol(f.func_name)), f) for f in EXPOSED_FUNCTIONS)
-arc_globals[t] = t
+arc_globals[ac_global_name(t)] = t
 arc_globals[ac_global_name(Symbol('+'))] = lambda *args: sum(args) 
                    
 
diff --git a/test_eval.py b/test_eval.py
@@ -0,0 +1,38 @@
+import unittest
+
+import eval
+from larktypes import String
+from roots import *
+from symbol import *
+
+class TestEval(unittest.TestCase):
+  def teststring(self):
+    self.assertEqual(eval.ac_eval(String('ab'), {}), String('ab'))
+
+  def testint(self):
+    self.assertEqual(eval.ac_eval(1, {}), 1)
+
+  def testfloat(self):
+    self.assertEqual(eval.ac_eval(0.5, {}), 0.5)
+
+  def testvarref(self):
+    self.assertEqual(eval.ac_eval(Symbol('a'), {Symbol('a'): 42}), 42)
+
+  def testquote(self):
+    self.assertEqual(eval.ac_eval([Symbol('quote'), [Symbol('a'), nil]], {}),
+                     Symbol('a'))
+
+  def testif(self):
+    self.assertEqual(eval.ac_eval([Symbol('if'), nil], {}), nil)
+    self.assertEqual(eval.ac_eval(list(Symbol('if'), t, 1, 2), {}), 1)
+
+  def testfncall(self):
+    # (fn (a) (if a 1 2))
+    fun = list(Symbol('fn'), list(Symbol('a')),
+               list(Symbol('if'), Symbol('a'), 1, 2))
+    self.assertEqual(eval.ac_eval(list(fun, nil), {}), 2)
+    self.assertEqual(eval.ac_eval(list(fun, String('x')), {}), 1)
+
+
+if __name__ == '__main__':
+  unittest.main()
