FilesExpand file tree

 master

/

bingoaddable.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

86 lines (65 loc) · 1.97 KB

 master

/

bingoaddable.py

Copy path

Top

File metadata and controls

• Code
• Blame

86 lines (65 loc) · 1.97 KB

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

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

"""

======================

AddableBingoCage tests

======================

Tests for __add__:

# BEGIN ADDABLE_BINGO_ADD_DEMO

>>> vowels = 'AEIOU'

>>> globe = AddableBingoCage(vowels) # <1>

>>> globe.inspect()

('A', 'E', 'I', 'O', 'U')

>>> globe.pick() in vowels # <2>

True

>>> len(globe.inspect()) # <3>

4

>>> globe2 = AddableBingoCage('XYZ') # <4>

>>> globe3 = globe + globe2

>>> len(globe3.inspect()) # <5>

7

>>> void = globe + [10, 20] # <6>

Traceback (most recent call last):

...

TypeError: unsupported operand type(s) for +: 'AddableBingoCage' and 'list'

# END ADDABLE_BINGO_ADD_DEMO

Tests for __iadd__:

# BEGIN ADDABLE_BINGO_IADD_DEMO

>>> globe_orig = globe # <1>

>>> len(globe.inspect()) # <2>

4

>>> globe += globe2 # <3>

>>> len(globe.inspect())

7

>>> globe += ['M', 'N'] # <4>

>>> len(globe.inspect())

9

>>> globe is globe_orig # <5>

True

>>> globe += 1 # <6>

Traceback (most recent call last):

...

TypeError: right operand in += must be 'AddableBingoCage' or an iterable

# END ADDABLE_BINGO_IADD_DEMO

"""

# BEGIN ADDABLE_BINGO

import itertools # <1>

from tombola import Tombola

from bingo import BingoCage

class AddableBingoCage(BingoCage): # <2>

def __add__(self, other):

if isinstance(other, Tombola): # <3>

return AddableBingoCage(self.inspect() + other.inspect()) # <6>

else:

return NotImplemented

def __iadd__(self, other):

if isinstance(other, Tombola):

other_iterable = other.inspect() # <4>

else:

try:

other_iterable = iter(other) # <5>

except TypeError: # <6>

self_cls = type(self).__name__

msg = "right operand in += must be {!r} or an iterable"

raise TypeError(msg.format(self_cls))

self.load(other_iterable) # <7>

return self # <8>

# END ADDABLE_BINGO