X-Git-Url: https://git.lukelau.me/?p=scheme.git;a=blobdiff_plain;f=tests.scm;h=6b66bec1ae3b6de291a2636778fc2656a6c246a6;hp=4ff6d760315700a6b862b1ce0818dafb78956640;hb=d0e9f5296b7510fe057be4a2f9e2a31ed856652c;hpb=91145e54f41ee88f1e279a80430b3f5ed4e7a8c6

diff --git a/tests.scm b/tests.scm
index 4ff6d76..6b66bec 100644
--- a/tests.scm
+++ b/tests.scm
@@ -1,12 +1,15 @@
 (load "codegen.scm")
 (load "typecheck.scm")
 
-(define (test actual expected)
-  (when (not (equal? actual expected))
+(define (test-f pred actual expected)
+  (when (not (pred actual expected))
     (error #f
 	   (format "test failed:\nexpected: ~a\nactual:   ~a"
 		   expected actual))))
 
+(define (test . xs) (apply test-f (cons equal? xs)))
+(define (test-types . xs) (apply test-f (cons types-unify? xs)))
+
 (define (read-file file)
   (call-with-input-file file
     (lambda (p)
@@ -30,9 +33,23 @@
   (let ((str (read-file "/tmp/test-output.txt")))
     (test str output)))
 
-(test (typecheck '(lambda (x) (+ ((lambda (y) (x y 3)) 5) 2)))
+(test-types (typecheck '(lambda (x) (+ ((lambda (y) (x y 3)) 5) 2)))
 	    '(abs (abs int (abs int int)) int))
 
+					; recursive types
+
+(test-types (substitute '((t1 (abs t1 t10))) 't1) '(abs t1 t10))
+
+(test-types (typecheck '(let ([bar (lambda (y) y)]
+			      [foo (lambda (x) (foo (bar #t)))])
+			  foo))
+	    '(abs bool t0))
+
+(test-types (typecheck '(let ([bar (lambda (y) y)]
+			[foo (lambda (x) (foo (bar #t)))])
+		    bar))
+      '(abs t0 t0))
+
 (test-prog '(+ 1 2) 3)
 (test-prog '((lambda (x) ((lambda (y) (+ x y)) 42)) 100) 142)
 
@@ -59,3 +76,8 @@
 (test-prog-stdout '(let () ((lambda (f) (f "foo")) print) 0) "foo")
 (test-prog '((lambda (f) (f 3 3)) (lambda (x y) (bool->int (= x y)))) 1)
 (test-prog '(bool->int ((lambda (f) (! (f 2 3))) =)) 1)
+
+					; recursion (hangs at typechecking)
+(test-prog '(let [(fac (lambda (f n x) (if (= n 0) x (f f (- n 1) (* x x)))))]
+	      (fac fac 3 2))
+	   8)