mirror of
https://github.com/pocketpy/pocketpy
synced 2025-11-09 21:20:17 +00:00
Remove separate zero division case checks
This commit is contained in:
parent
e237a7dfa2
commit
15d017c18d
@ -68,14 +68,10 @@ class complex:
|
||||
def __truediv__(self, other):
|
||||
if type(other) is complex:
|
||||
denominator = other.real ** 2 + other.imag ** 2
|
||||
if denominator == 0:
|
||||
raise ZeroDivisionError("Division by zero in complex number division")
|
||||
real_part = (self.real * other.real + self.imag * other.imag) / denominator
|
||||
imag_part = (self.imag * other.real - self.real * other.imag) / denominator
|
||||
return complex(real_part, imag_part)
|
||||
if type(other) in (int, float):
|
||||
if other == 0:
|
||||
raise ZeroDivisionError("Division by zero in complex number division")
|
||||
return complex(self.real / other, self.imag / other)
|
||||
return NotImplemented
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user