Added type hints to iterator pattern

2ykwang · 2ykwang · commit d752949b0efa · 2021-12-06T10:35:04.000+09:00
diff --git a/patterns/behavioral/iterator_alt.py b/patterns/behavioral/iterator_alt.py
@@ -4,6 +4,7 @@
 *TL;DR
 Traverses a container and accesses the container's elements.
 """
+from __future__ import annotations
 
 
 class NumberWords:
@@ -17,14 +18,14 @@ class NumberWords:
 "five",
 )
 
- def __init__(self, start, stop):
+ def __init__(self, start: int, stop: int) -> None:
 self.start = start
 self.stop = stop
 
- def __iter__(self): # this makes the class an Iterable
+ def __iter__(self) -> NumberWords: # this makes the class an Iterable
 return self
 
- def __next__(self): # this makes the class an Iterator
+ def __next__(self) -> str: # this makes the class an Iterator
 if self.start > self.stop or self.start > len(self._WORD_MAP):
 raise StopIteration
 current = self.start

-Original file line number
+Diff line change
 *TL;DR
 Traverses a container and accesses the container's elements.
 """
 +from __future__ importannotations
 classNumberWords:
 "five",
+ )
 -def__init__(self, start, stop):
 +def__init__(self, start: int, stop: int) ->None:
 self.start=start
 self.stop=stop
 -def__iter__(self): # this makes the class an Iterable
 +def__iter__(self)->NumberWords: # this makes the class an Iterable
 returnself
 -def__next__(self): # this makes the class an Iterator
 +def__next__(self)->str: # this makes the class an Iterator
 ifself.start>self.stoporself.start>len(self._WORD_MAP):
 raiseStopIteration
 current=self.start