1 import unittest

     2 from test import test_support

     3 

     4 class Empty:

     5     def __repr__(self):

     6         return '<Empty>'

     7 

     8 class Coerce:

     9     def __init__(self, arg):

    10         self.arg = arg

    11 

    12     def __repr__(self):

    13         return '<Coerce %s>' % self.arg

    14 

    15     def __coerce__(self, other):

    16         if isinstance(other, Coerce):

    17             return self.arg, other.arg

    18         else:

    19             return self.arg, other

    20 

    21 class Cmp:

    22     def __init__(self,arg):

    23         self.arg = arg

    24 

    25     def __repr__(self):

    26         return '<Cmp %s>' % self.arg

    27 

    28     def __cmp__(self, other):

    29         return cmp(self.arg, other)

    30 

    31 class ComparisonTest(unittest.TestCase):

    32     set1 = [2, 2.0, 2L, 2+0j, Coerce(2), Cmp(2.0)]

    33     set2 = [[1], (3,), None, Empty()]

    34     candidates = set1 + set2

    35 

    36     def test_comparisons(self):

    37         for a in self.candidates:

    38             for b in self.candidates:

    39                 if ((a in self.set1) and (b in self.set1)) or a is b:

    40                     self.assertEqual(a, b)

    41                 else:

    42                     self.assertNotEqual(a, b)

    43 

    44     def test_id_comparisons(self):

    45         # Ensure default comparison compares id() of args

    46         L = []

    47         for i in range(10):

    48             L.insert(len(L)//2, Empty())

    49         for a in L:

    50             for b in L:

    51                 self.assertEqual(cmp(a, b), cmp(id(a), id(b)),

    52                                  'a=%r, b=%r' % (a, b))

    53 

    54 def test_main():

    55     test_support.run_unittest(ComparisonTest)

    56 

    57 if __name__ == '__main__':

    58     test_main()