diff --git a/vector.py b/vector.py index a7fb54d..a6d2317 100644 --- a/vector.py +++ b/vector.py @@ -21,8 +21,20 @@ class Vector(object): def __eq__(self, v): return self.coordinates == v.coordinates + + def plus(self,v): + new_coordinates = [x+y for x,y in zip(self.coordinates, v.coordinates)] + return Vector(new_coordinates) # Tests vektor_1 = Vector([1,2,3]) -print vektor_1 \ No newline at end of file +print vektor_1 + +vektor_2 = Vector([1,2,3]) +vektor_3 = Vector([-1,2,3]) + +print vektor_1 == vektor_2 +print vektor_2 == vektor_3 + +print vektor_1.plus(vektor_2) \ No newline at end of file