def test_iterate(self): index = 0 s = Stream.iterate(index, lambda i: i + 1) for elem in s: self.assertEqual(elem, index) index += 1 if index == 10: break
def test_limit(self): s = Stream.iterate(0, lambda i: i + 1).limit(5).toList() self.assertEqual(s, [0, 1, 2, 3, 4])