exercicio vector repr

ramalho · ramalho · commit 6d366bd1b0dc · 2016-07-30T14:47:08.000-03:00
diff --git a/exercises/vector/vector_repr.py b/exercises/vector/vector_repr.py
@@ -0,0 +1,34 @@
+"""
+A multi-dimensional ``Vector`` class with ``repr``
+
+    >>> v1 = Vector([1, 2, 3])
+    >>> print(v1)
+    (1.0 2.0 3.0)
+    >>> v1
+    Vector([1.0, 2.0, 3.0])
+
+"""
+
+from array import array
+import math
+import reprlib
+
+
+class Vector:
+    typecode = 'd'
+
+    def __init__(self, components):
+        self._components = array(self.typecode, components)
+
+    def __len__(self):
+        return len(self._components)
+
+    def __iter__(self):
+        return iter(self._components)
+
+    def __abs__(self):
+        return math.sqrt(sum(x * x for x in self))
+
+    def __eq__(self, other):
+        return (len(self) == len(other) and
+                all(a == b for a, b in zip(self, other)))
