suzanne.soy/lens
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add stream set and nested tests

Browse Source
This commit is contained in:
Jack Firth 2015-08-24 14:08:34 -07:00
parent 14a8fdcbaa
commit c014d0abfa
1 changed files with 37 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
lens/private/stream.rkt
View File

@ -1,16 +1,27 @@
#lang racket/base #lang sweet-exp racket/base
(provide stream-first-lens require racket/contract/base
stream-rest-lens
stream-ref-lens)
(require racket/stream provide
contract-out
stream-first-lens (lens/c stream? any/c)
stream-rest-lens (lens/c stream? stream?)
stream-ref-lens (-> exact-nonnegative-integer? (lens/c stream? any/c))
require racket/stream
fancy-app fancy-app
"base/main.rkt" "base/main.rkt"
"compound/main.rkt") "compound/main.rkt"
module+ test
require rackunit
"test-util/test-lens.rkt"
module+ test
(define-check (check-stream-equal? stream1 stream2)
(equal? (stream->list stream1) (stream->list stream2)))
(module+ test
(require rackunit "test-util/test-lens.rkt"))
(define (stream-ref-lens i) (define (stream-ref-lens i)
(lens-compose stream-first-lens (stream-tail-lens i))) (lens-compose stream-first-lens (stream-tail-lens i)))
@ -43,14 +54,22 @@
(for/fold ([rst rst]) ([v (in-list rev-fst)]) (for/fold ([rst rst]) ([v (in-list rev-fst)])
(stream-cons v rst))) (stream-cons v rst)))
module+ test
(check-equal? (lens-view stream-first-lens (stream 'a 'b 'c)) 'a)
(check-equal? (lens-view (stream-ref-lens 2) (stream 'a 'b 'c)) 'c)
(check-stream-equal? (lens-set stream-first-lens (stream 'a 'b 'c) 1)
(stream 1 'b 'c))
(check-stream-equal? (lens-set (stream-ref-lens 2) (stream 'a 'b 'c) 1)
(stream 'a 'b 1))
(define (stream-ref-nested-lens . is) (define (stream-ref-nested-lens . is)
(apply lens-thrush (map stream-ref-lens is))) (apply lens-thrush (map stream-ref-lens is)))
(module+ test module+ test
(check-view stream-first-lens (stream 'a 'b 'c) 'a) (check-equal? (lens-view (stream-ref-nested-lens 1 2 0)
(check-view (stream-ref-lens 2) (stream 'a 'b 'c) 'c) (stream 'a (stream 1 2 (stream 'foo 'bar 'baz) 3 4) 'b 'c 'd))
(check-set-view stream-first-lens (stream 'a 'b 'c) (gensym)) 'foo)
(check-set-view (stream-ref-lens 2) (stream 'a 'b 'c) (gensym)) (check-stream-equal? (lens-set (stream-ref-nested-lens 1 2 0)
(check-set-set stream-first-lens (stream 'a 'b 'c) (gensym) (gensym)) (stream 'a (stream 1 2 (stream 'foo 'bar 'baz) 3 4) 'b 'c 'd)
(check-set-set (stream-ref-lens 2) (stream 'a 'b 'c) (gensym) (gensym)) 'FOO)
) (stream 'a (stream 1 2 (stream 'FOO 'bar 'baz) 3 4) 'b 'c 'd))