mirror of
https://github.com/wren-lang/wren.git
synced 2026-01-14 15:47:59 +01:00
49 lines
1.1 KiB
Plaintext
49 lines
1.1 KiB
Plaintext
// Infinite iterator demonstrating that Sequence.map is not eager
|
|
class FibIterator {
|
|
construct new() {
|
|
_current = 0
|
|
_next = 1
|
|
}
|
|
|
|
def iterate {
|
|
var sum = _current + _next
|
|
_current = _next
|
|
_next = sum
|
|
}
|
|
|
|
def value { _current }
|
|
}
|
|
|
|
class Fib is Sequence {
|
|
construct new() {}
|
|
|
|
def iterate(iterator) {
|
|
if (iterator == null) return FibIterator.new()
|
|
iterator.iterate
|
|
return iterator
|
|
}
|
|
|
|
def 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
|