ADT codegen working for simple types

[scheme.git] / tests.scm

diff --git a/tests.scm b/tests.scm
index 4e50dc61908cee17c4d1c1ada96c57282abf5830..2b6e09a4c2c63207707f061f722ca4ca3c377a66 100644 (file)
--- a/tests.scm
+++ b/tests.scm
@@ -43,6 +43,24 @@
   (let ((str (read-file "/tmp/test-output.txt")))
     (test str output)))
 
+(test (data-tors '(A . ((foo Int Bool)
+                       (bar Bool))))
+      '((foo (A foo constructor)
+            abs Int (abs Bool A))
+       (foo~0 (A foo 0) abs A Int)
+       (foo~1 (A foo 1) abs A Bool)
+       (bar (A bar constructor) abs Bool A)
+       (bar~0 (A bar 0) abs A Bool)))
+
+(test (data-tors-type-env
+       '(A . ((foo Int Bool)
+             (bar Bool))))
+      '((foo . (abs Int (abs Bool A)))
+       (foo~0 . (abs A Int))
+       (foo~1 . (abs A Bool))
+       (bar . (abs Bool A))
+       (bar~0 . (abs A Bool))))
+
 (test-types (typecheck '((lambda (x) (+ ((lambda (y) (x y 3)) 5) 2))))
            '(abs (abs Int (abs Int Int)) Int))
 
@@ -95,6 +113,9 @@
                           (pow 4 2))))
            'Int)
 
+                                       ; ADTs
+
+
 (test-types
  (typecheck
   '((data A
@@ -115,6 +136,27 @@
       y)))
  'Int)
 
+
+                                       ; pattern matching
+(test (let-bindings '(let ([(foo x) a]) x))
+      '((x (foo~0 a))))
+
+(test (let-bindings '(let ([x (foo 42)] [(foo y) x]) x))
+      '((x (foo 42))
+       (y (foo~0 x))))
+
+                                       ; type annotations
+
+(test (annotate-types
+       '((let ([x 42]
+              [y (+ 1 x)])
+          (- y x))))
+
+      '((let ()
+         ((let ((x (42 : Int))
+                (y (((+ : (abs Int (abs Int Int))) (1 : Int) (x : Int)) : Int)))
+            (((- : (abs Int (abs Int Int))) (y : Int) (x : Int)) : Int)) : Int)) : Int))
+
 (test-expr '(+ 1 2) 3)
 (test-expr '(bool->int (= 2 0)) 0)
 (test-expr '((lambda (x) ((lambda (y) (+ x y)) 42)) 100) 142)
@@ -168,3 +210,23 @@
                           (go p 1)))])
              (pow 4 2))
           16)
+
+                                       ; mutual recursion
+;; (test-prog-stdout '((let ([f (lambda (n)
+;;                           (if (= n 0)
+;;                               0
+;;                               (let ()
+;;                                 (print "a")
+;;                                 (g (- n 1)))))]
+;;                      [g (lambda (m)
+;;                           (let ()
+;;                             (print "b")
+;;                              (f (- m 1))))])
+;;                      (f 10))) "ababababab")
+
+                                       ; adts and pattern matching
+
+(test-prog '((data A [foo Bool Int])
+            (let ([(foo x y) (foo (= 3 3) 42)])
+              y))
+          42)