Files
wren/test/core/sequence/map.wren
T
Bob Nystrom aa72900a9e feat: replace IO class with System class and remove separate IO module
- Delete wren_io.c, wren_io.h, and io.wren source files
- Remove multi-arity print overloads and IO.read/IO.time methods
- Add System class with print, printAll, write, writeAll methods to core.wren
- Update all documentation examples and README to use System.print instead of IO.print
2015-09-15 14:46:09 +00:00

49 lines
1.1 KiB
Plaintext

// Infinite iterator demonstrating that Sequence.map is not eager
class FibIterator {
construct new() {
_current = 0
_next = 1
}
iterate {
var sum = _current + _next
_current = _next
_next = sum
}
value { _current }
}
class Fib is Sequence {
construct new() {}
iterate(iterator) {
if (iterator == null) return FibIterator.new()
iterator.iterate
return iterator
}
iteratorValue(iterator) { iterator.value }
}
var squareFib = Fib.new().map {|fib| fib * fib }
var iterator = null
System.print(squareFib is Sequence) // expect: true
System.print(squareFib) // expect: instance of MapSequence
iterator = squareFib.iterate(iterator)
System.print(squareFib.iteratorValue(iterator)) // expect: 0
iterator = squareFib.iterate(iterator)
System.print(squareFib.iteratorValue(iterator)) // expect: 1
iterator = squareFib.iterate(iterator)
System.print(squareFib.iteratorValue(iterator)) // expect: 1
iterator = squareFib.iterate(iterator)
System.print(squareFib.iteratorValue(iterator)) // expect: 4
iterator = squareFib.iterate(iterator)
System.print(squareFib.iteratorValue(iterator)) // expect: 9