(use debug heap test)

(define (figure-6.2)
  (let ((heap (make-max-heap)))
    (for-each (lambda (i) (heap-insert! heap i i))
      '(16 4 10 14 7 9 3 2 8 1))
    heap))

(let ((heap (figure-6.2)))
  (let iter ((data '()))
    (if (heap-empty? heap)
        (test
         "Heap data" 
         data
         '(1 2 3 4 7 8 9 10 14 16))
        (iter
         (cons (heap-extract-extremum! heap) data)))))

(let ((heap (figure-6.2)))
  (test "Heap-extremum"
        16
        (heap-extremum heap))
  (test "Heap-extract-extremum!"
        16
        (heap-extract-extremum! heap))
  (test "Next heap-extremum"
        14
        (heap-extremum heap)))

(define (figure-6.5)
  (let ((heap (make-max-heap)))
    (for-each (lambda (i) (heap-insert! heap i i))
      '(16 14 10 8 7 9 3 2 4 1))
    heap))

(let ((heap (figure-6.5)))
  (heap-change-key! heap 4 15)
  (let iter ((data '()))
    (if (heap-empty? heap)
        (test
         "Heap-change-key!"
         data
         '(1 2 3 7 8 9 10 14 4 16))
        (iter
         (cons (heap-extract-extremum! heap) data)))))

(let ((heap (figure-6.5)))
  (heap-delete! heap 8)
  (let iter ((data '()))
    (if (heap-empty? heap)
        (test
         "Heap-delete!"
         data
         '(1 2 3 4 7 9 10 14 16))
        (iter
         (cons (heap-extract-extremum! heap) data)))))

(let ((heap (make-max-heap)))
  (heap-insert! heap 1 'a)
  (heap-insert! heap 2 'b)
  (heap-insert! heap 3 'c)
  (test-assert "heap-member?" (heap-member? heap 'a))
  (test "heap-key -- exists" 1 (heap-key heap 'a))
  (test "heap-key -- doesn't exist" #f (heap-key heap 'd))
  (heap-delete! heap 'c)
  (test-assert "not-heap-member?" (not (heap-member? heap 'c))))