add node->generator

1 files changed, 20 insertions, 8 deletions

diff --git a/tests/run.scm b/tests/run.scm
index 25cbc4c..8cfdabc 100644
--- a/tests/run.scm
+++ b/tests/run.scm
@@ -422,13 +422,11 @@
 ;;; Update
 ;;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 
+(define %insert insert)
 (define (insert elem set)
-  (update number-comparator
-          set
-          elem
-          (lambda (value) value)
-          (lambda ()
-            (wb-tree-node elem '() '()))))
+  (%insert number-comparator
+           set
+           elem))
 
 (test-group "update of unordered list is balanced"
   (define (insert-unordered lst)
@@ -471,7 +469,7 @@
                             (when (verbose)
                               (dynamic-property-set! 'elem elem)
                               (dynamic-property-set! 'set set))
-                            (delete number-comparator elem set))
+                            (delete number-comparator set elem))
                           set1
                           lst2)))
       (procedure lst1 lst2 diff deleted))))
@@ -480,7 +478,7 @@
   (define (delete-balanced? lst1 lst2)
     (let* ((set (fold insert '() lst1))
            (deleted (fold (lambda (elem set)
-                            (delete number-comparator elem set))
+                            (delete number-comparator set elem))
                           set
                           lst2)))
       (balanced? deleted)))
@@ -534,3 +532,17 @@
   (test-property xor-subset-of-lset
                  (list (list-generator-of (exact-integer-generator))
                        (list-generator-of (exact-integer-generator)))))
+
+;;; ;;;;;;;;;;
+;;; Generators
+;;; ;;;;;;;;;;
+
+(test-group "node->generator"
+  (define (node->generator-works lst)
+    (let* ((node (fold insert '() lst))
+           (result-lst
+            (generator->list (node->generator node))))
+      (lset= = lst result-lst)))
+  (test-property node->generator-works
+                 (list (list-generator-of (exact-integer-generator)))))
+