ch11-24: clean up by @eumiro & sync with Atlas

Author: ramalho

11-pythonic-obj/private/.gitignore

@@ -0,0 +1,2 @@
+*.class
+.jython_cache/

13-protocol-abc/README.rst

@@ -0,0 +1,4 @@
+Sample code for Chapter 11 - "Interfaces, protocols and ABCs"
+
+From the book "Fluent Python" by Luciano Ramalho (O'Reilly, 2015)
+http://shop.oreilly.com/product/0636920032519.do

13-protocol-abc/bingo.py

@@ -0,0 +1,28 @@
+# tag::TOMBOLA_BINGO[]
+
+import random
+
+from tombola import Tombola
+
+
+class BingoCage(Tombola):  # <1>
+
+    def __init__(self, items):
+        self._randomizer = random.SystemRandom()  # <2>
+        self._items = []
+        self.load(items)  # <3>
+
+    def load(self, items):
+        self._items.extend(items)
+        self._randomizer.shuffle(self._items)  # <4>
+
+    def pick(self):  # <5>
+        try:
+            return self._items.pop()
+        except IndexError:
+            raise LookupError('pick from empty BingoCage')
+
+    def __call__(self):  # <6>
+        self.pick()
+
+# end::TOMBOLA_BINGO[]

13-protocol-abc/double/double_object.py

@@ -0,0 +1,2 @@
+def double(x: object) -> object:
+    return x * 2

13-protocol-abc/double/double_protocol.py

@@ -0,0 +1,11 @@
+from typing import TypeVar, Protocol
+
+T = TypeVar('T')  # <1>
+
+class Repeatable(Protocol):
+    def __mul__(self: T, repeat_count: int) -> T: ...  # <2>
+
+RT = TypeVar('RT', bound=Repeatable)  # <3>
+
+def double(x: RT) -> RT:  # <4>
+    return x * 2

13-protocol-abc/double/double_sequence.py

@@ -0,0 +1,6 @@
+from collections import abc
+from typing import Any
+
+def double(x: abc.Sequence) -> Any:
+    return x * 2
+

13-protocol-abc/double/double_test.py

@@ -0,0 +1,56 @@
+from typing import TYPE_CHECKING
+import pytest
+from double_protocol import double
+
+def test_double_int() -> None:
+    given = 2
+    result = double(given)
+    assert result == given * 2
+    if TYPE_CHECKING:
+        reveal_type(given)
+        reveal_type(result)
+
+
+def test_double_str() -> None:
+    given = 'A'
+    result = double(given)
+    assert result == given * 2
+    if TYPE_CHECKING:
+        reveal_type(given)
+        reveal_type(result)
+
+
+def test_double_fraction() -> None:
+    from fractions import Fraction
+    given = Fraction(2, 5)
+    result = double(given)
+    assert result == given * 2
+    if TYPE_CHECKING:
+        reveal_type(given)
+        reveal_type(result)
+
+
+def test_double_array() -> None:
+    from array import array
+    given = array('d', [1.0, 2.0, 3.14])
+    result = double(given)
+    if TYPE_CHECKING:
+        reveal_type(given)
+        reveal_type(result)
+
+
+def test_double_nparray() -> None:
+    import numpy as np  # type: ignore
+    given = np.array([[1, 2], [3, 4]])
+    result = double(given)
+    comparison = result == given * 2
+    assert comparison.all()
+    if TYPE_CHECKING:
+        reveal_type(given)
+        reveal_type(result)
+
+
+def test_double_none() -> None:
+    given = None
+    with pytest.raises(TypeError):
+        result = double(given)

13-protocol-abc/drum.py

@@ -0,0 +1,17 @@
+from random import shuffle
+
+from tombola import Tombola
+
+
+class TumblingDrum(Tombola):
+
+    def __init__(self, iterable):
+        self._balls = []
+        self.load(iterable)
+
+    def load(self, iterable):
+        self._balls.extend(iterable)
+        shuffle(self._balls)
+
+    def pick(self):
+        return self._balls.pop()

13-protocol-abc/frenchdeck2.py

@@ -0,0 +1,26 @@
+import collections
+
+Card = collections.namedtuple('Card', ['rank', 'suit'])
+
+class FrenchDeck2(collections.MutableSequence):
+    ranks = [str(n) for n in range(2, 11)] + list('JQKA')
+    suits = 'spades diamonds clubs hearts'.split()
+
+    def __init__(self):
+        self._cards = [Card(rank, suit) for suit in self.suits
+                                        for rank in self.ranks]
+
+    def __len__(self):
+        return len(self._cards)
+
+    def __getitem__(self, position):
+        return self._cards[position]
+
+    def __setitem__(self, position, value):  # <1>
+        self._cards[position] = value
+
+    def __delitem__(self, position):  # <2>
+        del self._cards[position]
+
+    def insert(self, position, value):  # <3>
+        self._cards.insert(position, value)

13-protocol-abc/lotto.py

@@ -0,0 +1,30 @@
+# tag::LOTTERY_BLOWER[]
+
+import random
+
+from tombola import Tombola
+
+
+class LotteryBlower(Tombola):
+
+    def __init__(self, iterable):
+        self._balls = list(iterable)  # <1>
+
+    def load(self, iterable):
+        self._balls.extend(iterable)
+
+    def pick(self):
+        try:
+            position = random.randrange(len(self._balls))  # <2>
+        except ValueError:
+            raise LookupError('pick from empty BingoCage')
+        return self._balls.pop(position)  # <3>
+
+    def loaded(self):  # <4>
+        return bool(self._balls)
+
+    def inspect(self):  # <5>
+        return tuple(sorted(self._balls))
+
+
+# end::LOTTERY_BLOWER[]