wren/test/core/sequence/where.wren

// Infinite iterator demonstrating that Sequence.where is not eager
class FibIterator {
  this new() {
    _current = 0
    _next = 1
  }

  iterate {
    var sum = _current + _next
    _current = _next
    _next = sum
  }

  value { _current }
}

class Fib is Sequence {
  iterate(iterator) {
    if (iterator == null) return FibIterator.new()
    iterator.iterate
    return iterator
  }

  iteratorValue(iterator) { iterator.value }
}

var largeFibs = Fib.new().where {|fib| fib > 100 }
var iterator = null

IO.print(largeFibs is Sequence) // expect: true
IO.print(largeFibs) // expect: instance of WhereSequence

iterator = largeFibs.iterate(iterator)
IO.print(largeFibs.iteratorValue(iterator)) // expect: 144

iterator = largeFibs.iterate(iterator)
IO.print(largeFibs.iteratorValue(iterator)) // expect: 233

iterator = largeFibs.iterate(iterator)
IO.print(largeFibs.iteratorValue(iterator)) // expect: 377