Experimenting with magic methods.

jwasham · jwasham · commit 680938bec900 · 2016-08-06T18:17:16.000-07:00
diff --git a/experiments/deck.py b/experiments/deck.py
@@ -4,26 +4,32 @@
 Card = collections.namedtuple('Card', ['rank', 'suit'])
 
 
-class Deck(object):
+class Deck:
     ranks = [str(i) for i in range(2, 11)] + list('JKQA')
     suits = 'spades diamonds clubs hearts'.split()
 
     def __init__(self):
-        self._cards = [Card(rank, suit) for rank in self.ranks
-                                        for suit in self.suits]
+        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 __repr__(self):
+        return 'Deck()'
+
 
 def main():
     deck = Deck()
     # print(len(deck))
-    print(random.choice(deck))
-
+    for __ in range(5):
+        print(random.choice(deck))
+    # print(deck[12::13])
+    # for card in reversed(deck):
+    #     print(card)
 
 if __name__ == "__main__":
     main()
diff --git a/experiments/vector.py b/experiments/vector.py
@@ -0,0 +1,37 @@
+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})".format(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 __mult__(self, scalar):
+        return Vector(self.x * scalar, self.y * scalar)
+
+
+def main():
+    v1 = Vector(2, 4)
+    v2 = Vector(2, 1)
+
+    print(v1)
+    print(v2)
+    # assert (v1 + v2) == Vector(4, 5)
+    # assert abs( * 3)
+
+if __name__ == "__main__":
+    main()
