home *** CD-ROM | disk | FTP | other *** search
/ Komputer for Alle 2004 #2 / K-CD-2-2004.ISO / OpenOffice Sv / f_0397 / python-core-2.2.2 / lib / test / test_coercion.py < prev    next >
Encoding:
Python Source  |  2003-07-18  |  3.0 KB  |  119 lines
  1. import copy
  2. import sys
  3. import warnings
  4.  
  5. # Fake a number that implements numeric methods through __coerce__
  6. class CoerceNumber:
  7.     def __init__(self, arg):
  8.         self.arg = arg
  9.  
  10.     def __repr__(self):
  11.         return '<CoerceNumber %s>' % repr(self.arg)
  12.  
  13.     def __coerce__(self, other):
  14.         if isinstance(other, CoerceNumber):
  15.             return self.arg, other.arg
  16.         else:
  17.             return (self.arg, other)
  18.  
  19.  
  20. # Fake a number that implements numeric ops through methods.
  21. class MethodNumber:
  22.  
  23.     def __init__(self,arg):
  24.         self.arg = arg
  25.  
  26.     def __repr__(self):
  27.         return '<MethodNumber %s>' % repr(self.arg)
  28.  
  29.     def __add__(self,other):
  30.         return self.arg + other
  31.  
  32.     def __radd__(self,other):
  33.         return other + self.arg
  34.  
  35.     def __sub__(self,other):
  36.         return self.arg - other
  37.  
  38.     def __rsub__(self,other):
  39.         return other - self.arg
  40.  
  41.     def __mul__(self,other):
  42.         return self.arg * other
  43.  
  44.     def __rmul__(self,other):
  45.         return other * self.arg
  46.  
  47.     def __div__(self,other):
  48.         return self.arg / other
  49.  
  50.     def __rdiv__(self,other):
  51.         return other / self.arg
  52.  
  53.     def __pow__(self,other):
  54.         return self.arg ** other
  55.  
  56.     def __rpow__(self,other):
  57.         return other ** self.arg
  58.  
  59.     def __mod__(self,other):
  60.         return self.arg % other
  61.  
  62.     def __rmod__(self,other):
  63.         return other % self.arg
  64.  
  65.     def __cmp__(self, other):
  66.         return cmp(self.arg, other)
  67.  
  68.  
  69. candidates = [ 2, 4.0, 2L, 2+0j, [1], (2,), None,
  70.                MethodNumber(1), CoerceNumber(2)]
  71.  
  72. infix_binops = [ '+', '-', '*', '/', '**', '%' ]
  73. prefix_binops = [ 'divmod' ]
  74.  
  75. def do_infix_binops():
  76.     for a in candidates:
  77.         for b in candidates:
  78.             for op in infix_binops:
  79.                 print '%s %s %s' % (a, op, b),
  80.                 try:
  81.                     x = eval('a %s b' % op)
  82.                 except:
  83.                     error = sys.exc_info()[:2]
  84.                     print '... %s' % error[0]
  85.                 else:
  86.                     print '=', x
  87.                 try:
  88.                     z = copy.copy(a)
  89.                 except copy.Error:
  90.                     z = a # assume it has no inplace ops
  91.                 print '%s %s= %s' % (a, op, b),
  92.                 try:
  93.                     exec('z %s= b' % op)
  94.                 except:
  95.                     error = sys.exc_info()[:2]
  96.                     print '... %s' % error[0]
  97.                 else:
  98.                     print '=>', z
  99.  
  100. def do_prefix_binops():
  101.     for a in candidates:
  102.         for b in candidates:
  103.             for op in prefix_binops:
  104.                 print '%s(%s, %s)' % (op, a, b),
  105.                 try:
  106.                     x = eval('%s(a, b)' % op)
  107.                 except:
  108.                     error = sys.exc_info()[:2]
  109.                     print '... %s' % error[0]
  110.                 else:
  111.                     print '=', x
  112.  
  113. warnings.filterwarnings("ignore",
  114.                         r'complex divmod\(\), // and % are deprecated',
  115.                         DeprecationWarning,
  116.                         r'test_coercion$')
  117. do_infix_binops()
  118. do_prefix_binops()
  119.