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_array.py < prev    next >
Encoding:
Python Source  |  2003-07-18  |  6.5 KB  |  193 lines
  1. #! /usr/bin/env python
  2. """Test the arraymodule.
  3.    Roger E. Masse
  4. """
  5. import array
  6. from test_support import verbose, TESTFN, unlink, TestFailed
  7.  
  8. def main():
  9.  
  10.     testtype('c', 'c')
  11.  
  12.     for type in (['b', 'h', 'i', 'l', 'f', 'd']):
  13.         testtype(type, 1)
  14.  
  15.     unlink(TESTFN)
  16.  
  17.  
  18. def testoverflow(type, lowerLimit, upperLimit):
  19.         # should not overflow assigning lower limit
  20.     if verbose:
  21.         print "overflow test: array(%s, [%s])" % (`type`, `lowerLimit`)
  22.     try:
  23.         a = array.array(type, [lowerLimit])
  24.     except:
  25.         raise TestFailed, "array(%s) overflowed assigning %s" %\
  26.                 (`type`, `lowerLimit`)
  27.     # should overflow assigning less than lower limit
  28.     if verbose:
  29.         print "overflow test: array(%s, [%s])" % (`type`, `lowerLimit-1`)
  30.     try:
  31.         a = array.array(type, [lowerLimit-1])
  32.         raise TestFailed, "array(%s) did not overflow assigning %s" %\
  33.                 (`type`, `lowerLimit-1`)
  34.     except OverflowError:
  35.         pass
  36.     # should not overflow assigning upper limit
  37.     if verbose:
  38.         print "overflow test: array(%s, [%s])" % (`type`, `upperLimit`)
  39.     try:
  40.         a = array.array(type, [upperLimit])
  41.     except:
  42.         raise TestFailed, "array(%s) overflowed assigning %s" %\
  43.                 (`type`, `upperLimit`)
  44.     # should overflow assigning more than upper limit
  45.     if verbose:
  46.         print "overflow test: array(%s, [%s])" % (`type`, `upperLimit+1`)
  47.     try:
  48.         a = array.array(type, [upperLimit+1])
  49.         raise TestFailed, "array(%s) did not overflow assigning %s" %\
  50.                 (`type`, `upperLimit+1`)
  51.     except OverflowError:
  52.         pass
  53.  
  54.  
  55.  
  56. def testtype(type, example):
  57.  
  58.     a = array.array(type)
  59.     a.append(example)
  60.     if verbose:
  61.         print 40*'*'
  62.         print 'array after append: ', a
  63.     a.typecode
  64.     a.itemsize
  65.     if a.typecode in ('i', 'b', 'h', 'l'):
  66.         a.byteswap()
  67.  
  68.     if a.typecode == 'c':
  69.         f = open(TESTFN, "w")
  70.         f.write("The quick brown fox jumps over the lazy dog.\n")
  71.         f.close()
  72.         f = open(TESTFN, 'r')
  73.         a.fromfile(f, 10)
  74.         f.close()
  75.         if verbose:
  76.             print 'char array with 10 bytes of TESTFN appended: ', a
  77.         a.fromlist(['a', 'b', 'c'])
  78.         if verbose:
  79.             print 'char array with list appended: ', a
  80.  
  81.     a.insert(0, example)
  82.     if verbose:
  83.         print 'array of %s after inserting another:' % a.typecode, a
  84.     f = open(TESTFN, 'w')
  85.     a.tofile(f)
  86.     f.close()
  87.  
  88.     # This block is just to verify that the operations don't blow up.
  89.     a.tolist()
  90.     a.tostring()
  91.     repr(a)
  92.     str(a)
  93.  
  94.     if verbose:
  95.         print 'array of %s converted to a list: ' % a.typecode, a.tolist()
  96.     if verbose:
  97.         print 'array of %s converted to a string: ' \
  98.                % a.typecode, `a.tostring()`
  99.  
  100.     if type == 'c':
  101.         a = array.array(type, "abcde")
  102.         a[:-1] = a
  103.         if a != array.array(type, "abcdee"):
  104.             raise TestFailed, "array(%s) self-slice-assign (head)" % `type`
  105.         a = array.array(type, "abcde")
  106.         a[1:] = a
  107.         if a != array.array(type, "aabcde"):
  108.             raise TestFailed, "array(%s) self-slice-assign (tail)" % `type`
  109.         a = array.array(type, "abcde")
  110.         a[1:-1] = a
  111.         if a != array.array(type, "aabcdee"):
  112.             raise TestFailed, "array(%s) self-slice-assign (cntr)" % `type`
  113.         if a.index("e") != 5:
  114.             raise TestFailed, "array(%s) index-test" % `type`
  115.         if a.count("a") != 2:
  116.             raise TestFailed, "array(%s) count-test" % `type`
  117.         a.remove("e")
  118.         if a != array.array(type, "aabcde"):
  119.             raise TestFailed, "array(%s) remove-test" % `type`
  120.         if a.pop(0) != "a":
  121.             raise TestFailed, "array(%s) pop-test" % `type`
  122.         if a.pop(1) != "b":
  123.             raise TestFailed, "array(%s) pop-test" % `type`
  124.         a.extend(array.array(type, "xyz"))
  125.         if a != array.array(type, "acdexyz"):
  126.             raise TestFailed, "array(%s) extend-test" % `type`
  127.         a.pop()
  128.         a.pop()
  129.         a.pop()
  130.         x = a.pop()
  131.         if x != 'e':
  132.             raise TestFailed, "array(%s) pop-test" % `type`
  133.         if a != array.array(type, "acd"):
  134.             raise TestFailed, "array(%s) pop-test" % `type`
  135.         a.reverse()
  136.         if a != array.array(type, "dca"):
  137.             raise TestFailed, "array(%s) reverse-test" % `type`
  138.     else:
  139.         a = array.array(type, [1, 2, 3, 4, 5])
  140.         a[:-1] = a
  141.         if a != array.array(type, [1, 2, 3, 4, 5, 5]):
  142.             raise TestFailed, "array(%s) self-slice-assign (head)" % `type`
  143.         a = array.array(type, [1, 2, 3, 4, 5])
  144.         a[1:] = a
  145.         if a != array.array(type, [1, 1, 2, 3, 4, 5]):
  146.             raise TestFailed, "array(%s) self-slice-assign (tail)" % `type`
  147.         a = array.array(type, [1, 2, 3, 4, 5])
  148.         a[1:-1] = a
  149.         if a != array.array(type, [1, 1, 2, 3, 4, 5, 5]):
  150.             raise TestFailed, "array(%s) self-slice-assign (cntr)" % `type`
  151.         if a.index(5) != 5:
  152.             raise TestFailed, "array(%s) index-test" % `type`
  153.         if a.count(1) != 2:
  154.             raise TestFailed, "array(%s) count-test" % `type`
  155.         a.remove(5)
  156.         if a != array.array(type, [1, 1, 2, 3, 4, 5]):
  157.             raise TestFailed, "array(%s) remove-test" % `type`
  158.         if a.pop(0) != 1:
  159.             raise TestFailed, "array(%s) pop-test" % `type`
  160.         if a.pop(1) != 2:
  161.             raise TestFailed, "array(%s) pop-test" % `type`
  162.         a.extend(array.array(type, [7, 8, 9]))
  163.         if a != array.array(type, [1, 3, 4, 5, 7, 8, 9]):
  164.             raise TestFailed, "array(%s) extend-test" % `type`
  165.         a.pop()
  166.         a.pop()
  167.         a.pop()
  168.         x = a.pop()
  169.         if x != 5:
  170.             raise TestFailed, "array(%s) pop-test" % `type`
  171.         if a != array.array(type, [1, 3, 4]):
  172.             raise TestFailed, "array(%s) pop-test" % `type`
  173.         a.reverse()
  174.         if a != array.array(type, [4, 3, 1]):
  175.             raise TestFailed, "array(%s) reverse-test" % `type`
  176.  
  177.     # test that overflow exceptions are raised as expected for assignment
  178.     # to array of specific integral types
  179.     from math import pow
  180.     if type in ('b', 'h', 'i', 'l'):
  181.         # check signed and unsigned versions
  182.         a = array.array(type)
  183.         signedLowerLimit = -1 * long(pow(2, a.itemsize * 8 - 1))
  184.         signedUpperLimit = long(pow(2, a.itemsize * 8 - 1)) - 1L
  185.         unsignedLowerLimit = 0
  186.         unsignedUpperLimit = long(pow(2, a.itemsize * 8)) - 1L
  187.         testoverflow(type, signedLowerLimit, signedUpperLimit)
  188.         testoverflow(type.upper(), unsignedLowerLimit, unsignedUpperLimit)
  189.  
  190.  
  191.  
  192. main()
  193.