example-code/01-data-model/vector2d.py at master · fluentpython/example-code

This repository was archived by the owner on Dec 2, 2021. It is now read-only.

24 lines (17 loc) · 509 Bytes

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)' % (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 __mul__(self, scalar):

return Vector(self.x * scalar, self.y * scalar)

Provide feedback