TR updates, first round

Author: lpozo

python-iterators-iterables/fib_gen.py

@@ -1,23 +1,18 @@
-# def fibonacci_generator(stop):
-#     previous, next = 0, 1
+# def fibonacci_generator(stop=10):
+#     current_fib, next_fib = 0, 1
 #     for _ in range(0, stop):
-#         fib_number = previous
-#         previous, next = next, previous + next
+#         fib_number = current_fib
+#         current_fib, next_fib = next_fib, current_fib + next_fib
 #         yield fib_number
 
 
-def fibonacci_generator(stop=100):
-    previous, next = 0, 1
+def fibonacci_generator(stop=10):
+    current_fib, next_fib = 0, 1
     index = 0
     while True:
         if index == stop:
             return
         index += 1
-        fib_number = previous
-        previous, next = next, previous + next
+        fib_number = current_fib
+        current_fib, next_fib = next_fib, current_fib + next_fib
         yield fib_number
-
-
-print("Fibonacci Generator")
-for fib in fibonacci_generator(10):
-    print(fib)

python-iterators-iterables/fib_iter.py

@@ -1,17 +1,21 @@
-class Fibonaccier:
+class FibonacciIterator:
     def __init__(self, stop=10):
         self._stop = stop
         self._index = 0
-        self._previous = 0
+        self._current = 0
         self._next = 1
 
     def __iter__(self):
         return self
 
     def __next__(self):
-        if self._index >= self._stop:
+        if self._index < self._stop:
+            self._index += 1
+            fib_number = self._current
+            self._current, self._next = (
+                self._next,
+                self._current + self._next,
+            )
+            return fib_number
+        else:
             raise StopIteration
-        self._index += 1
-        fib_number = self._previous
-        self._previous, self._next = self._next, self._previous + self._next
-        return fib_number

python-iterators-iterables/inf_fib.py

@@ -0,0 +1,14 @@
+class FibonacciInfIterator:
+    def __init__(self):
+        self._index = 0
+        self._current = 0
+        self._next = 1
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        self._index += 1
+        fib_number = self._current
+        self._current, self._next = self._next, self._current + self._next
+        return fib_number

python-iterators-iterables/pipeline.py

@@ -0,0 +1,22 @@
+def to_square(numbers):
+    return (number**2 for number in numbers)
+
+
+def to_cube(numbers):
+    return (number**3 for number in numbers)
+
+
+def to_even(numbers):
+    return (number for number in numbers if number % 2 == 0)
+
+
+def to_odd(numbers):
+    return (number for number in numbers if number % 2 != 0)
+
+
+def to_string(numbers):
+    return (str(number) for number in numbers)
+
+
+print(list(to_string(to_square(to_even(range(20))))))
+print(list(to_string(to_cube(to_odd(range(20))))))

python-iterators-iterables/reusable_range.py

@@ -0,0 +1,16 @@
+class ReusableRange:
+    def __init__(self, start=0, stop=None, step=1):
+        if stop is None:
+            stop, start = start, 0
+        self._range = range(start, stop, step)
+        self._iter = iter(self._range)
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        try:
+            return next(self._iter)
+        except StopIteration:
+            self._iter = iter(self._range)
+            raise