improved vector add solution

ramalho · ramalho · commit 45757dac799f · 2016-07-30T16:15:49.000-03:00
diff --git a/exercises/vector/solution/vector.py b/exercises/vector/solution/vector.py
@@ -61,15 +61,12 @@ def __neg__(self):
 
     def __add__(self, other):
         try:
-            if len(self) != len(other):
-                raise ValueError('Only vectors of same length may be added.')
-            return Vector(a + b for a, b in zip(self, other))
+            other = Vector(other)
         except TypeError:
             return NotImplemented
+        if len(self) != len(other):
+            raise ValueError('vectors must have same length for addition')
+        return Vector(a + b for a, b in zip(self, other))
 
     def __radd__(self, other):
         return self + other
-
-
-
-
