Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
56 changes: 54 additions & 2 deletions app/main.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,55 @@
from __future__ import annotations


class Distance:
# Write your code here
pass
def __init__(self, km: int | float) -> None:
self.km = km

def __str__(self) -> str:
return f"Distance: {self.km} kilometers."

def __repr__(self) -> str:
return f"Distance(km={self.km})"

def __add__(self, other: int | Distance) -> Distance:
if isinstance(other, Distance):
return Distance(self.km + other.km)
return Distance(self.km + other)

def __iadd__(self, other: int | Distance) -> Distance:
if isinstance(other, Distance):
self.km += other.km
return self
self.km += other
return self

def __mul__(self, other: int) -> Distance:

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

__mul__ must return a new Distance instance with km set to the product, not mutate self. The description example uses distance2 = distance1 * 5 and then checks isinstance(distance2, Distance) and distance2.km == 100, which implies distance1 should stay unchanged.

return Distance(self.km * other)

def __truediv__(self, other: float | int) -> Distance:
return Distance(round((self.km / other), 2))

def __lt__(self, other: int | Distance) -> bool:
if isinstance(other, Distance):
return self.km < other.km
return self.km < other

def __gt__(self, other: int | Distance) -> bool:
if isinstance(other, Distance):
return self.km > other.km
return self.km > other

def __eq__(self, other: int | Distance) -> bool:
if isinstance(other, Distance):
return self.km == other.km
return self.km == other

def __le__(self, other: int | Distance) -> bool:
if isinstance(other, Distance):
return self.km <= other.km
return self.km <= other

def __ge__(self, other: int | Distance) -> bool:
if isinstance(other, Distance):
return self.km >= other.km
return self.km >= other
Loading