Lib/dos-8x3/test_arr.py

   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         a.tolist()
  88         a.tostring()
  89         if verbose:
  90             print 'array of %s converted to a list: ' % a.typecode, a.tolist()
  91         if verbose:
  92             print 'array of %s converted to a string: ' \
  93                    % a.typecode, `a.tostring()`
  94
  95         if type == 'c':
  96             a = array.array(type, "abcde")
  97             a[:-1] = a
  98             if a != array.array(type, "abcdee"):
  99                 raise TestFailed, "array(%s) self-slice-assign (head)" % `type`
 100             a = array.array(type, "abcde")
 101             a[1:] = a
 102             if a != array.array(type, "aabcde"):
 103                 raise TestFailed, "array(%s) self-slice-assign (tail)" % `type`
 104             a = array.array(type, "abcde")
 105             a[1:-1] = a
 106             if a != array.array(type, "aabcdee"):
 107                 raise TestFailed, "array(%s) self-slice-assign (cntr)" % `type`
 108         else:
 109             a = array.array(type, [1, 2, 3, 4, 5])
 110             a[:-1] = a
 111             if a != array.array(type, [1, 2, 3, 4, 5, 5]):
 112                 raise TestFailed, "array(%s) self-slice-assign (head)" % `type`
 113             a = array.array(type, [1, 2, 3, 4, 5])
 114             a[1:] = a
 115             if a != array.array(type, [1, 1, 2, 3, 4, 5]):
 116                 raise TestFailed, "array(%s) self-slice-assign (tail)" % `type`
 117             a = array.array(type, [1, 2, 3, 4, 5])
 118             a[1:-1] = a
 119             if a != array.array(type, [1, 1, 2, 3, 4, 5, 5]):
 120                 raise TestFailed, "array(%s) self-slice-assign (cntr)" % `type`
 121
 122         # test that overflow exceptions are raised as expected for assignment
 123         # to array of specific integral types
 124         from math import pow
 125         if type in ('b', 'h', 'i', 'l'):
 126             # check signed and unsigned versions
 127             a = array.array(type)
 128             signedLowerLimit = -1 * long(pow(2, a.itemsize * 8 - 1))
 129             signedUpperLimit = long(pow(2, a.itemsize * 8 - 1)) - 1L
 130             unsignedLowerLimit = 0
 131             unsignedUpperLimit = long(pow(2, a.itemsize * 8)) - 1L
 132             testoverflow(type, signedLowerLimit, signedUpperLimit)
 133             testoverflow(type.upper(), unsignedLowerLimit, unsignedUpperLimit)
 134
 135
 136
 137 main()
 138