Make Fraction to complex comparisons with <=, <, >= or > raise TypeError.
This commit is contained in:
parent
355adc5a45
commit
71b7fac07b
|
@ -511,8 +511,10 @@ class Fraction(Rational):
|
|||
if isinstance(other, Rational):
|
||||
return op(self._numerator * other.denominator,
|
||||
self._denominator * other.numerator)
|
||||
if isinstance(other, numbers.Complex) and other.imag == 0:
|
||||
other = other.real
|
||||
# comparisons with complex should raise a TypeError, for consistency
|
||||
# with int<->complex, float<->complex, and complex<->complex comparisons.
|
||||
if isinstance(other, complex):
|
||||
raise TypeError("no ordering relation is defined for complex numbers")
|
||||
if isinstance(other, float):
|
||||
if math.isnan(other) or math.isinf(other):
|
||||
return op(0.0, other)
|
||||
|
|
|
@ -473,8 +473,21 @@ class FractionTest(unittest.TestCase):
|
|||
|
||||
def testBigComplexComparisons(self):
|
||||
self.assertFalse(F(10**23) == complex(10**23))
|
||||
self.assertTrue(F(10**23) > complex(10**23))
|
||||
self.assertFalse(F(10**23) <= complex(10**23))
|
||||
self.assertRaises(TypeError, operator.gt, F(10**23), complex(10**23))
|
||||
self.assertRaises(TypeError, operator.le, F(10**23), complex(10**23))
|
||||
|
||||
x = F(3, 8)
|
||||
z = complex(0.375, 0.0)
|
||||
w = complex(0.375, 0.2)
|
||||
self.assertTrue(x == z)
|
||||
self.assertFalse(x != z)
|
||||
self.assertFalse(x == w)
|
||||
self.assertTrue(x != w)
|
||||
for op in operator.lt, operator.le, operator.gt, operator.ge:
|
||||
self.assertRaises(TypeError, op, x, z)
|
||||
self.assertRaises(TypeError, op, z, x)
|
||||
self.assertRaises(TypeError, op, x, w)
|
||||
self.assertRaises(TypeError, op, w, x)
|
||||
|
||||
def testMixedEqual(self):
|
||||
self.assertTrue(0.5 == F(1, 2))
|
||||
|
|
Loading…
Reference in New Issue