home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2011 June / maximum-cd-2011-06.iso / DiscContents / LibO_3.3.1_Win_x86_install_multi.exe / libreoffice1.cab / test_augassign.py < prev    next >
Encoding:
Python Source  |  2011-02-15  |  7.8 KB  |  328 lines
  1. # Augmented assignment test.
  2.  
  3. from test.test_support import run_unittest
  4. import unittest
  5.  
  6.  
  7. class AugAssignTest(unittest.TestCase):
  8.     def testBasic(self):
  9.         x = 2
  10.         x += 1
  11.         x *= 2
  12.         x **= 2
  13.         x -= 8
  14.         x //= 5
  15.         x %= 3
  16.         x &= 2
  17.         x |= 5
  18.         x ^= 1
  19.         x /= 2
  20.         if 1/2 == 0:
  21.             # classic division
  22.             self.assertEquals(x, 3)
  23.         else:
  24.             # new-style division (with -Qnew)
  25.             self.assertEquals(x, 3.0)
  26.  
  27.     def testInList(self):
  28.         x = [2]
  29.         x[0] += 1
  30.         x[0] *= 2
  31.         x[0] **= 2
  32.         x[0] -= 8
  33.         x[0] //= 5
  34.         x[0] %= 3
  35.         x[0] &= 2
  36.         x[0] |= 5
  37.         x[0] ^= 1
  38.         x[0] /= 2
  39.         if 1/2 == 0:
  40.             self.assertEquals(x[0], 3)
  41.         else:
  42.             self.assertEquals(x[0], 3.0)
  43.  
  44.     def testInDict(self):
  45.         x = {0: 2}
  46.         x[0] += 1
  47.         x[0] *= 2
  48.         x[0] **= 2
  49.         x[0] -= 8
  50.         x[0] //= 5
  51.         x[0] %= 3
  52.         x[0] &= 2
  53.         x[0] |= 5
  54.         x[0] ^= 1
  55.         x[0] /= 2
  56.         if 1/2 == 0:
  57.             self.assertEquals(x[0], 3)
  58.         else:
  59.             self.assertEquals(x[0], 3.0)
  60.  
  61.     def testSequences(self):
  62.         x = [1,2]
  63.         x += [3,4]
  64.         x *= 2
  65.  
  66.         self.assertEquals(x, [1, 2, 3, 4, 1, 2, 3, 4])
  67.  
  68.         x = [1, 2, 3]
  69.         y = x
  70.         x[1:2] *= 2
  71.         y[1:2] += [1]
  72.  
  73.         self.assertEquals(x, [1, 2, 1, 2, 3])
  74.         self.assert_(x is y)
  75.  
  76.     def testCustomMethods1(self):
  77.  
  78.         class aug_test:
  79.             def __init__(self, value):
  80.                 self.val = value
  81.             def __radd__(self, val):
  82.                 return self.val + val
  83.             def __add__(self, val):
  84.                 return aug_test(self.val + val)
  85.  
  86.         class aug_test2(aug_test):
  87.             def __iadd__(self, val):
  88.                 self.val = self.val + val
  89.                 return self
  90.  
  91.         class aug_test3(aug_test):
  92.             def __iadd__(self, val):
  93.                 return aug_test3(self.val + val)
  94.  
  95.         x = aug_test(1)
  96.         y = x
  97.         x += 10
  98.  
  99.         self.assert_(isinstance(x, aug_test))
  100.         self.assert_(y is not x)
  101.         self.assertEquals(x.val, 11)
  102.  
  103.         x = aug_test2(2)
  104.         y = x
  105.         x += 10
  106.  
  107.         self.assert_(y is x)
  108.         self.assertEquals(x.val, 12)
  109.  
  110.         x = aug_test3(3)
  111.         y = x
  112.         x += 10
  113.  
  114.         self.assert_(isinstance(x, aug_test3))
  115.         self.assert_(y is not x)
  116.         self.assertEquals(x.val, 13)
  117.  
  118.  
  119.     def testCustomMethods2(test_self):
  120.         output = []
  121.  
  122.         class testall:
  123.             def __add__(self, val):
  124.                 output.append("__add__ called")
  125.             def __radd__(self, val):
  126.                 output.append("__radd__ called")
  127.             def __iadd__(self, val):
  128.                 output.append("__iadd__ called")
  129.                 return self
  130.  
  131.             def __sub__(self, val):
  132.                 output.append("__sub__ called")
  133.             def __rsub__(self, val):
  134.                 output.append("__rsub__ called")
  135.             def __isub__(self, val):
  136.                 output.append("__isub__ called")
  137.                 return self
  138.  
  139.             def __mul__(self, val):
  140.                 output.append("__mul__ called")
  141.             def __rmul__(self, val):
  142.                 output.append("__rmul__ called")
  143.             def __imul__(self, val):
  144.                 output.append("__imul__ called")
  145.                 return self
  146.  
  147.             def __div__(self, val):
  148.                 output.append("__div__ called")
  149.             def __rdiv__(self, val):
  150.                 output.append("__rdiv__ called")
  151.             def __idiv__(self, val):
  152.                 output.append("__idiv__ called")
  153.                 return self
  154.  
  155.             def __floordiv__(self, val):
  156.                 output.append("__floordiv__ called")
  157.                 return self
  158.             def __ifloordiv__(self, val):
  159.                 output.append("__ifloordiv__ called")
  160.                 return self
  161.             def __rfloordiv__(self, val):
  162.                 output.append("__rfloordiv__ called")
  163.                 return self
  164.  
  165.             def __truediv__(self, val):
  166.                 output.append("__truediv__ called")
  167.                 return self
  168.             def __itruediv__(self, val):
  169.                 output.append("__itruediv__ called")
  170.                 return self
  171.  
  172.             def __mod__(self, val):
  173.                 output.append("__mod__ called")
  174.             def __rmod__(self, val):
  175.                 output.append("__rmod__ called")
  176.             def __imod__(self, val):
  177.                 output.append("__imod__ called")
  178.                 return self
  179.  
  180.             def __pow__(self, val):
  181.                 output.append("__pow__ called")
  182.             def __rpow__(self, val):
  183.                 output.append("__rpow__ called")
  184.             def __ipow__(self, val):
  185.                 output.append("__ipow__ called")
  186.                 return self
  187.  
  188.             def __or__(self, val):
  189.                 output.append("__or__ called")
  190.             def __ror__(self, val):
  191.                 output.append("__ror__ called")
  192.             def __ior__(self, val):
  193.                 output.append("__ior__ called")
  194.                 return self
  195.  
  196.             def __and__(self, val):
  197.                 output.append("__and__ called")
  198.             def __rand__(self, val):
  199.                 output.append("__rand__ called")
  200.             def __iand__(self, val):
  201.                 output.append("__iand__ called")
  202.                 return self
  203.  
  204.             def __xor__(self, val):
  205.                 output.append("__xor__ called")
  206.             def __rxor__(self, val):
  207.                 output.append("__rxor__ called")
  208.             def __ixor__(self, val):
  209.                 output.append("__ixor__ called")
  210.                 return self
  211.  
  212.             def __rshift__(self, val):
  213.                 output.append("__rshift__ called")
  214.             def __rrshift__(self, val):
  215.                 output.append("__rrshift__ called")
  216.             def __irshift__(self, val):
  217.                 output.append("__irshift__ called")
  218.                 return self
  219.  
  220.             def __lshift__(self, val):
  221.                 output.append("__lshift__ called")
  222.             def __rlshift__(self, val):
  223.                 output.append("__rlshift__ called")
  224.             def __ilshift__(self, val):
  225.                 output.append("__ilshift__ called")
  226.                 return self
  227.  
  228.         x = testall()
  229.         x + 1
  230.         1 + x
  231.         x += 1
  232.  
  233.         x - 1
  234.         1 - x
  235.         x -= 1
  236.  
  237.         x * 1
  238.         1 * x
  239.         x *= 1
  240.  
  241.         if 1/2 == 0:
  242.             x / 1
  243.             1 / x
  244.             x /= 1
  245.         else:
  246.             # True division is in effect, so "/" doesn't map to __div__ etc;
  247.             # but the canned expected-output file requires that those get called.
  248.             x.__div__(1)
  249.             x.__rdiv__(1)
  250.             x.__idiv__(1)
  251.  
  252.         x // 1
  253.         1 // x
  254.         x //= 1
  255.  
  256.         x % 1
  257.         1 % x
  258.         x %= 1
  259.  
  260.         x ** 1
  261.         1 ** x
  262.         x **= 1
  263.  
  264.         x | 1
  265.         1 | x
  266.         x |= 1
  267.  
  268.         x & 1
  269.         1 & x
  270.         x &= 1
  271.  
  272.         x ^ 1
  273.         1 ^ x
  274.         x ^= 1
  275.  
  276.         x >> 1
  277.         1 >> x
  278.         x >>= 1
  279.  
  280.         x << 1
  281.         1 << x
  282.         x <<= 1
  283.  
  284.         test_self.assertEquals(output, '''\
  285. __add__ called
  286. __radd__ called
  287. __iadd__ called
  288. __sub__ called
  289. __rsub__ called
  290. __isub__ called
  291. __mul__ called
  292. __rmul__ called
  293. __imul__ called
  294. __div__ called
  295. __rdiv__ called
  296. __idiv__ called
  297. __floordiv__ called
  298. __rfloordiv__ called
  299. __ifloordiv__ called
  300. __mod__ called
  301. __rmod__ called
  302. __imod__ called
  303. __pow__ called
  304. __rpow__ called
  305. __ipow__ called
  306. __or__ called
  307. __ror__ called
  308. __ior__ called
  309. __and__ called
  310. __rand__ called
  311. __iand__ called
  312. __xor__ called
  313. __rxor__ called
  314. __ixor__ called
  315. __rshift__ called
  316. __rrshift__ called
  317. __irshift__ called
  318. __lshift__ called
  319. __rlshift__ called
  320. __ilshift__ called
  321. '''.splitlines())
  322.  
  323. def test_main():
  324.     run_unittest(AugAssignTest)
  325.  
  326. if __name__ == '__main__':
  327.     test_main()
  328.