FilesExpand file tree

 master

/

iterator.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

47 lines (34 loc) · 845 Bytes

 master

/

iterator.py

Copy path

Top

File metadata and controls

• Code
• Blame

47 lines (34 loc) · 845 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

"""

http://ginstrom.com/scribbles/2007/10/08/design-patterns-python-style/

Implementation of the iterator pattern with a generator

*TL;DR

Traverses a container and accesses the container's elements.

"""

def count_to(count: int):

"""Counts by word numbers, up to a maximum of five"""

numbers = ["one", "two", "three", "four", "five"]

yield from numbers[:count]

# Test the generator

def count_to_two() -> None:

return count_to(2)

def count_to_five() -> None:

return count_to(5)

def main():

"""

# Counting to two...

>>> for number in count_to_two():

... print(number)

one

two

# Counting to five...

>>> for number in count_to_five():

... print(number)

one

two

three

four

five

"""

if __name__ == "__main__":

import doctest

doctest.testmod()