"""
======================
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