87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
#! /usr/bin/env python
|
|
"""Test the arraymodule.
|
|
Roger E. Masse
|
|
"""
|
|
import array
|
|
from test_support import verbose, TESTFN, unlink, TestFailed
|
|
|
|
def main():
|
|
|
|
testtype('c', 'c')
|
|
|
|
for type in (['b', 'h', 'i', 'l', 'f', 'd']):
|
|
testtype(type, 1)
|
|
|
|
unlink(TESTFN)
|
|
|
|
|
|
def testtype(type, example):
|
|
|
|
a = array.array(type)
|
|
a.append(example)
|
|
if verbose:
|
|
print 40*'*'
|
|
print 'array after append: ', a
|
|
a.typecode
|
|
a.itemsize
|
|
if a.typecode in ('i', 'b', 'h', 'l'):
|
|
a.byteswap()
|
|
|
|
if a.typecode == 'c':
|
|
f = open(TESTFN, "w")
|
|
f.write("The quick brown fox jumps over the lazy dog.\n")
|
|
f.close()
|
|
f = open(TESTFN, 'r')
|
|
a.fromfile(f, 10)
|
|
f.close()
|
|
if verbose:
|
|
print 'char array with 10 bytes of TESTFN appended: ', a
|
|
a.fromlist(['a', 'b', 'c'])
|
|
if verbose:
|
|
print 'char array with list appended: ', a
|
|
|
|
a.insert(0, example)
|
|
if verbose:
|
|
print 'array of %s after inserting another:' % a.typecode, a
|
|
f = open(TESTFN, 'w')
|
|
a.tofile(f)
|
|
f.close()
|
|
a.tolist()
|
|
a.tostring()
|
|
if verbose:
|
|
print 'array of %s converted to a list: ' % a.typecode, a.tolist()
|
|
if verbose:
|
|
print 'array of %s converted to a string: ' \
|
|
% a.typecode, `a.tostring()`
|
|
|
|
if type == 'c':
|
|
a = array.array(type, "abcde")
|
|
a[:-1] = a
|
|
if a != array.array(type, "abcdee"):
|
|
raise TestFailed, "array(%s) self-slice-assign (head)" % `type`
|
|
a = array.array(type, "abcde")
|
|
a[1:] = a
|
|
if a != array.array(type, "aabcde"):
|
|
raise TestFailed, "array(%s) self-slice-assign (tail)" % `type`
|
|
a = array.array(type, "abcde")
|
|
a[1:-1] = a
|
|
if a != array.array(type, "aabcdee"):
|
|
raise TestFailed, "array(%s) self-slice-assign (cntr)" % `type`
|
|
else:
|
|
a = array.array(type, [1, 2, 3, 4, 5])
|
|
a[:-1] = a
|
|
if a != array.array(type, [1, 2, 3, 4, 5, 5]):
|
|
raise TestFailed, "array(%s) self-slice-assign (head)" % `type`
|
|
a = array.array(type, [1, 2, 3, 4, 5])
|
|
a[1:] = a
|
|
if a != array.array(type, [1, 1, 2, 3, 4, 5]):
|
|
raise TestFailed, "array(%s) self-slice-assign (tail)" % `type`
|
|
a = array.array(type, [1, 2, 3, 4, 5])
|
|
a[1:-1] = a
|
|
if a != array.array(type, [1, 1, 2, 3, 4, 5, 5]):
|
|
raise TestFailed, "array(%s) self-slice-assign (cntr)" % `type`
|
|
|
|
|
|
main()
|
|
|