curriculum/challenges/english/blocks/learn-special-methods-by-building-a-vector-space/6601ab2809898f57591f2f7f.md
Now, modify the assignments of v1 and v2. Turn them into R3Vector instances and pass z=1 and z=2 to v1 and v2, respectively.
You should modify the assignment of v1 to be an R3Vector instance.
({ test: () => assert(runPython(`_Node(_code).find_variable("v1").is_equivalent("v1 = R3Vector(x=2, y=3, z=1)")`)) })
You should modify the assignment of v2 to be an R3Vector instance.
({ test: () => assert(runPython(`_Node(_code).find_variable("v2").is_equivalent("v2 = R3Vector(x=0.5, y=1.25, z=2)")`)) })
class R2Vector:
def __init__(self, *, x, y):
self.x = x
self.y = y
def norm(self):
return sum(val**2 for val in vars(self).values())**0.5
def __str__(self):
return str(tuple(getattr(self, i) for i in vars(self)))
def __repr__(self):
arg_list = [f'{key}={val}' for key, val in vars(self).items()]
args = ', '.join(arg_list)
return f'{self.__class__.__name__}({args})'
def __add__(self, other):
if type(self) != type(other):
return NotImplemented
kwargs = {i: getattr(self, i) + getattr(other, i) for i in vars(self)}
return self.__class__(**kwargs)
def __sub__(self, other):
if type(self) != type(other):
return NotImplemented
kwargs = {i: getattr(self, i) - getattr(other, i) for i in vars(self)}
return self.__class__(**kwargs)
def __mul__(self, other):
if type(other) in (int, float):
kwargs = {i: getattr(self, i) * other for i in vars(self)}
return self.__class__(**kwargs)
elif type(self) == type(other):
args = [getattr(self, i) * getattr(other, i) for i in vars(self)]
return sum(args)
return NotImplemented
def __eq__(self, other):
if type(self) != type(other):
return NotImplemented
return all(getattr(self, i) == getattr(other, i) for i in vars(self))
def __ne__(self, other):
return not self == other
def __lt__(self, other):
if type(self) != type(other):
return NotImplemented
return self.norm() < other.norm()
def __gt__(self, other):
if type(self) != type(other):
return NotImplemented
return self.norm() > other.norm()
def __le__(self, other):
return not self > other
def __ge__(self, other):
return not self < other
class R3Vector(R2Vector):
def __init__(self, *, x, y, z):
super().__init__(x=x, y=y)
self.z = z
def cross(self, other):
if type(self) != type(other):
return NotImplemented
kwargs = {
'x': self.y * other.z - self.z * other.y,
'y': self.z * other.x - self.x * other.z,
'z': self.x * other.y - self.y * other.x
}
return self.__class__(**kwargs)
--fcc-editable-region--
v1 = R2Vector(x=2, y=3)
v2 = R2Vector(x=0.5, y=1.25)
--fcc-editable-region--
print(f'v1 = {v1}')
print(f'v2 = {v2}')
v3 = v1 + v2
print(f'v1 + v2 = {v3}')
v4 = v1 - v2
print(f'v1 - v2 = {v4}')
v5 = v1 * v2
print(f'v1 * v2 = {v5}')