]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
Make Fraction to complex comparisons with <=, <, >= or > raise TypeError.
authorMark Dickinson <dickinsm@gmail.com>
Sat, 27 Mar 2010 11:09:29 +0000 (11:09 +0000)
committerMark Dickinson <dickinsm@gmail.com>
Sat, 27 Mar 2010 11:09:29 +0000 (11:09 +0000)
Lib/fractions.py
Lib/test/test_fractions.py

index a934389a37dfd1c4f39385146507f39dd92c6906..7a3e5553c4bfd612d0b3f829b0f7980b56fd7680 100644 (file)
@@ -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)
index 63ac7fd57ca99bf448fdbe39b0bea200599d8565..a24fcd3c3ab0501a251b69fe4c4d299e1db73f03 100644 (file)
@@ -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))