update from atlas
This commit is contained in:
54
01-data-model/frenchdeck.doctest
Normal file
54
01-data-model/frenchdeck.doctest
Normal file
@@ -0,0 +1,54 @@
|
||||
>>> from frenchdeck import FrenchDeck, Card
|
||||
>>> beer_card = Card('7', 'diamonds')
|
||||
>>> beer_card
|
||||
Card(rank='7', suit='diamonds')
|
||||
>>> deck = FrenchDeck()
|
||||
>>> len(deck)
|
||||
52
|
||||
>>> deck[:3]
|
||||
[Card(rank='2', suit='spades'), Card(rank='3', suit='spades'), Card(rank='4', suit='spades')]
|
||||
>>> deck[12::13]
|
||||
[Card(rank='A', suit='spades'), Card(rank='A', suit='diamonds'), Card(rank='A', suit='clubs'), Card(rank='A', suit='hearts')]
|
||||
>>> Card('Q', 'hearts') in deck
|
||||
True
|
||||
>>> Card('Z', 'clubs') in deck
|
||||
False
|
||||
>>> for card in deck: # doctest: +ELLIPSIS
|
||||
... print(card)
|
||||
Card(rank='2', suit='spades')
|
||||
Card(rank='3', suit='spades')
|
||||
Card(rank='4', suit='spades')
|
||||
...
|
||||
>>> for card in reversed(deck): # doctest: +ELLIPSIS
|
||||
... print(card)
|
||||
Card(rank='A', suit='hearts')
|
||||
Card(rank='K', suit='hearts')
|
||||
Card(rank='Q', suit='hearts')
|
||||
...
|
||||
>>> for n, card in enumerate(deck, 1): # doctest: +ELLIPSIS
|
||||
... print(n, card)
|
||||
1 Card(rank='2', suit='spades')
|
||||
2 Card(rank='3', suit='spades')
|
||||
3 Card(rank='4', suit='spades')
|
||||
...
|
||||
>>> suit_values = dict(spades=3, hearts=2, diamonds=1, clubs=0)
|
||||
>>> def spades_high(card):
|
||||
... rank_value = FrenchDeck.ranks.index(card.rank)
|
||||
... return rank_value * len(suit_values) + suit_values[card.suit]
|
||||
|
||||
Rank test:
|
||||
|
||||
>>> spades_high(Card('2', 'clubs'))
|
||||
0
|
||||
>>> spades_high(Card('A', 'spades'))
|
||||
51
|
||||
|
||||
>>> for card in sorted(deck, key=spades_high): # doctest: +ELLIPSIS
|
||||
... print(card)
|
||||
Card(rank='2', suit='clubs')
|
||||
Card(rank='2', suit='diamonds')
|
||||
Card(rank='2', suit='hearts')
|
||||
...
|
||||
Card(rank='A', suit='diamonds')
|
||||
Card(rank='A', suit='hearts')
|
||||
Card(rank='A', suit='spades')
|
||||
17
01-data-model/frenchdeck.py
Normal file
17
01-data-model/frenchdeck.py
Normal file
@@ -0,0 +1,17 @@
|
||||
import collections
|
||||
|
||||
Card = collections.namedtuple('Card', ['rank', 'suit'])
|
||||
|
||||
class FrenchDeck:
|
||||
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]
|
||||
24
01-data-model/vector2d.py
Normal file
24
01-data-model/vector2d.py
Normal file
@@ -0,0 +1,24 @@
|
||||
from math import hypot
|
||||
|
||||
class Vector:
|
||||
|
||||
def __init__(self, x=0, y=0):
|
||||
self.x = x
|
||||
self.y = y
|
||||
|
||||
def __repr__(self):
|
||||
return 'Vector(%r, %r)' % (self.x, self.y)
|
||||
|
||||
def __abs__(self):
|
||||
return hypot(self.x, self.y)
|
||||
|
||||
def __bool__(self):
|
||||
return bool(abs(self))
|
||||
|
||||
def __add__(self, other):
|
||||
x = self.x + other.x
|
||||
y = self.y + other.y
|
||||
return Vector(x, y)
|
||||
|
||||
def __mul__(self, scalar):
|
||||
return Vector(self.x * scalar, self.y * scalar)
|
||||
Reference in New Issue
Block a user