Skip to content
Open

Solve #2855

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
72 changes: 70 additions & 2 deletions app/main.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,71 @@
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: Distance | int | float) -> Distance:
if isinstance(other, Distance):
return Distance(self.km + other.km)
if isinstance(other, int | float):
return Distance(self.km + other)
raise TypeError
Comment on lines +14 to +19

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

This method still uses an elif and a trailing raise TypeError after the if, which goes against checklist item "Avoid unnecessary else statements" and the previous guidance to remove trailing branches returning/raising after if/elif. You can structure it as if isinstance(other, Distance): ...; if isinstance(other, (int, float)): ...; raise TypeError without elif.


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

def __mul__(self, other: int | float) -> Distance:
if isinstance(other, int | float):
return Distance(self.km * other)
raise TypeError

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

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

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

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

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

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