(module generator-test mzscheme
  
  (require (lib "cut.ss" "srfi" "26"))
  
  (require (file "generator.ss")
           (file "test-base.ss"))
  
  (provide generator-tests)

  (define generator-tests
    (test-suite
     "generator.ss"

      (test-case
       "list->generator works when no default specified"
       (let ([gen (list->generator (list 1 2 3))])
         (check-equal? (gen) 1  "check 1")
         (check-equal? (gen) 2  "check 2")
         (check-equal? (gen) 3  "check 3")
         (check-pred generator-end? (gen) "check 4")
         (check-pred generator-end? (gen) "check 5")))
      
      (test-case
       "generator->list works as expected"
       (check-equal? (generator->list (list->generator (list 1 2 3)))
                     (list 1 2 3))
       (check-equal? (generator->list (list->generator (list 1 2 3 generator-end 4 5)))
                     (list 1 2 3)))
      
      
      (test-case
       "generator-accumulate works as expected"
       (check-equal? (generator-accumulate 
                      cons
                      null
                      (list->generator (list 1 2 3)))
                     (list 3 2 1))
       (check-equal? (generator-accumulate
                      (lambda (a b accum)
                        (cons (+ a b) accum))
                      null
                      (list->generator (list 1 2 3 4 5))
                      (list->generator (list 2 4 6 8 10)))
                     (list 15 12 9 6 3))
       (check-equal? (generator-accumulate
                      (lambda (a b accum)
                        (cons (+ a b) accum))
                      null
                      (list->generator (list 1 2 3 4 5))
                      (list->generator (list 2 4 6 generator-end 10)))
                     (list 9 6 3)))
     
      (test-case
       "generator-map works as expected"
       (check-equal? (generator->list 
                      (generator-map 
                       even?
                       (list->generator (list 1 2 3 4 5))))
                     (list #f #t #f #t #f))
       (check-equal? (generator->list
                      (generator-map 
                       +
                       (list->generator (list 1 2 3 4 5))
                       (list->generator (list 1 3 5 7 9))))
                     (list 2 5 8 11 14)))
      
      (test-case
       "generator-fold works as expected"
       (check-equal? (generator->list 
                      (generator-fold
                       +
                       0
                       (list->generator (list 1 2 3 4 5))))
                     (list 1 3 6 10 15))
       (check-equal? (generator->list
                      (generator-fold 
                       +
                       0
                       (list->generator (list 1 2 3 4 5))
                       (list->generator (list 1 3 5 7 9))))
                     (list 2 7 15 26 40)))
      
      (test-case
       "generator-filter works as expected"
       (check-equal? (generator->list 
                      (generator-filter
                       even?
                       (list->generator (list 1 2 3 4 5))))
                     (list 2 4)))
      
      (test-case
       "generator-filter-map works as expected"
       (check-equal? (generator->list 
                      (generator-filter-map
                       (lambda (a)
                         (if (even? a)
                             (* 2 a)
                             #f))
                       (list->generator (list 1 2 3 4 5))))
                     (list 4 8)))
      
      (test-case
       "generator-remove-duplicates works as expected"
       (check-equal? (generator->list 
                      (generator-remove-duplicates
                       (list->generator (list 1 2 2 3 3 3 4 4 5))))
                     (list 1 2 3 4 5)))
      
      (test-case
       "generator-for-each works as expected"
       (check-equal? (let ([accum null])
                       (generator-for-each
                        (lambda (a)
                          (set! accum (cons a accum)))
                        (list->generator (list 1 2 3 4 5)))
                       accum)
                     (list 5 4 3 2 1))
       (check-equal? (let ([accum null])
                       (generator-for-each
                        (lambda (a b)
                          (set! accum (cons (+ a b) accum)))
                        (list->generator (list 1 2 3 4 5))
                        (list->generator (list 2 4 6 8 10)))
                       accum)
                     (list 15 12 9 6 3)))
      
      (test-case
       "generator-project works as expected"
       (let ([generate (generator-project (list #t #t #f #f)
                                          (list->generator (list (list 0 0 0 0)
                                                                 (list 0 0 0 1)
                                                                 (list 0 0 1 0)
                                                                 (list 0 0 1 1)
                                                                 (list 0 1 0 0)
                                                                 (list 0 1 0 1)
                                                                 (list 0 1 1 0)
                                                                 (list 0 1 1 1))))])
         (check-equal? (generate) 
                       (list 0 0 (list (list 0 0)
                                       (list 0 1)
                                       (list 1 0)
                                       (list 1 1)))
                       "check 1")
         (check-equal? (generate)
                       (list 0 1 (list (list 0 0)
                                       (list 0 1)
                                       (list 1 0)
                                       (list 1 1)))
                       "check 2")
         (check-pred generator-end?
                     (generate)
                     "check 3")
         (check-pred generator-end?
                     (generate)
                     "check 4")))
     
     (test-case
      "generator-project with mask completely #t"
      (let ([generate (generator-project (list #t #t #t #t)
                                         (list->generator (list (list 0 0 0 0)
                                                                (list 0 0 0 1)
                                                                (list 0 0 1 0)
                                                                (list 0 0 1 1)
                                                                (list 0 1 0 0)
                                                                (list 0 1 0 1)
                                                                (list 0 1 1 0)
                                                                (list 0 1 1 1))))])
        (check-equal? (generate) (list 0 0 0 0 (list)))
        (check-equal? (generate) (list 0 0 0 1 (list)))
        (check-equal? (generate) (list 0 0 1 0 (list)))
        (check-equal? (generate) (list 0 0 1 1 (list)))
        (check-equal? (generate) (list 0 1 0 0 (list)))
        (check-equal? (generate) (list 0 1 0 1 (list)))
        (check-equal? (generate) (list 0 1 1 0 (list)))
        (check-equal? (generate) (list 0 1 1 1 (list)))
        (check-pred generator-end? (generate))))
     
     (test-case
      "generator-project with mask completely #f"
      (let ([generate (generator-project (list #f #f #f #f)
                                         (list->generator (list (list 0 0 0 0)
                                                                (list 0 0 0 1)
                                                                (list 0 0 1 0)
                                                                (list 0 0 1 1)
                                                                (list 0 1 0 0)
                                                                (list 0 1 0 1)
                                                                (list 0 1 1 0)
                                                                (list 0 1 1 1))))])
        (check-equal? (generate) (list (list (list 0 0 0 0)
                                             (list 0 0 0 1)
                                             (list 0 0 1 0)
                                             (list 0 0 1 1)
                                             (list 0 1 0 0)
                                             (list 0 1 0 1)
                                             (list 0 1 1 0)
                                             (list 0 1 1 1))))
        (check-pred generator-end? (generate))))
     
     (test-case
      "generator-project passes non-list items straight through"
      (let ([generate (generator-project (list #t #t #f #f)
                                         (list->generator (list (list 0 0 0 0)
                                                                (list 0 0 0 1)
                                                                123
                                                                (list 0 1 0 0)
                                                                (list 0 1 0 1)
                                                                456)))])
        (check-equal? (generate) (list 0 0 (list (list 0 0) (list 0 1))) "check 1")
        (check-equal? (generate) 123)
        (check-equal? (generate) (list 0 1 (list (list 0 0) (list 0 1))) "check 2")
        (check-equal? (generate) 456)
        (check-pred generator-end? (generate))))
     
     (test-case
      "generator-project passes non-list items straight through"
      (let ([generate (generator-project (list #t #t #f #f)
                                         (list->generator (list (list 0 0 0 0)
                                                                (list 0 0 0 1)
                                                                123
                                                                (list 0 1 0 0)
                                                                (list 0 1 0 1)
                                                                456)))])
        (check-equal? (generate) (list 0 0 (list (list 0 0) (list 0 1))) "check 1")
        (check-equal? (generate) 123)
        (check-equal? (generate) (list 0 1 (list (list 0 0) (list 0 1))) "check 2")
        (check-equal? (generate) 456)
        (check-pred generator-end? (generate))))
     
     (test-case
      "generator-project uses eq? by default"
      (let ([generate (generator-project (list #t #t #f #f)
                                         (list->generator (list (list "0" "0" "0" "0")
                                                                (list "0" "0" "0" "1")
                                                                123
                                                                (list "0" "1" "0" "0")
                                                                (list "0" "1" "0" "1")
                                                                456)))])
        (check-equal? (generate) (list "0" "0" (list (list "0" "0"))) "check 1")
        (check-equal? (generate) (list "0" "0" (list (list "0" "1"))) "check 2")
        (check-equal? (generate) 123 "check 3")
        (check-equal? (generate) (list "0" "1" (list (list "0" "0"))) "check 4")
        (check-equal? (generate) (list "0" "1" (list (list "0" "1"))) "check 5")
        (check-equal? (generate) 456 "check 6")
        (check-pred generator-end? (generate) "check 7")))
     
     (test-case
      "generator-project can be configured to use equal? instead of eq?"
      (let ([generate (generator-project (list #t #t #f #f)
                                         (list->generator (list (list "0" "0" "0" "0")
                                                                (list "0" "0" "0" "1")
                                                                123
                                                                (list "0" "1" "0" "0")
                                                                (list "0" "1" "0" "1")
                                                                456))
                                         equal?)])
        (check-equal? (generate) (list "0" "0" (list (list "0" "0") (list "0" "1"))) "check 1")
        (check-equal? (generate) 123 "check 2")
        (check-equal? (generate) (list "0" "1" (list (list "0" "0") (list "0" "1"))) "check 3")
        (check-equal? (generate) 456 "check 4")
        (check-pred generator-end? (generate) "check 5")))
      
    ))

  )