svn+ssh://pythondev@svn.python.org/python/trunk ........ r53454 | brett.cannon | 2007-01-15 20:12:08 +0100 (Mon, 15 Jan 2007) | 3 lines Add a note for strptime that just because strftime supports some extra directive that is not documented that strptime will as well. ........ r53458 | vinay.sajip | 2007-01-16 10:50:07 +0100 (Tue, 16 Jan 2007) | 1 line Updated rotating file handlers to use _open(). ........ r53459 | marc-andre.lemburg | 2007-01-16 14:03:06 +0100 (Tue, 16 Jan 2007) | 2 lines Add news items for the recent pybench and platform changes. ........ r53460 | sjoerd.mullender | 2007-01-16 17:42:38 +0100 (Tue, 16 Jan 2007) | 4 lines Fixed ntpath.expandvars to not replace references to non-existing variables with nothing. Also added tests. This fixes bug #494589. ........ r53464 | neal.norwitz | 2007-01-17 07:23:51 +0100 (Wed, 17 Jan 2007) | 1 line Give Calvin Spealman access for python-dev summaries. ........ r53465 | neal.norwitz | 2007-01-17 09:37:26 +0100 (Wed, 17 Jan 2007) | 1 line Remove Calvin since he only has access to the website currently. ........ r53466 | thomas.heller | 2007-01-17 10:40:34 +0100 (Wed, 17 Jan 2007) | 2 lines Replace C++ comments with C comments. ........ r53472 | andrew.kuchling | 2007-01-17 20:55:06 +0100 (Wed, 17 Jan 2007) | 1 line [Part of bug #1599254] Add suggestion to Mailbox docs to use Maildir, and warn user to lock/unlock mailboxes when modifying them ........ r53475 | georg.brandl | 2007-01-17 22:09:04 +0100 (Wed, 17 Jan 2007) | 2 lines Bug #1637967: missing //= operator in list. ........ r53477 | georg.brandl | 2007-01-17 22:19:58 +0100 (Wed, 17 Jan 2007) | 2 lines Bug #1629125: fix wrong data type (int -> Py_ssize_t) in PyDict_Next docs. ........ r53481 | neal.norwitz | 2007-01-18 06:40:58 +0100 (Thu, 18 Jan 2007) | 1 line Try reverting part of r53145 that seems to cause the Windows buildbots to fail in test_uu.UUFileTest.test_encode ........ r53482 | fred.drake | 2007-01-18 06:42:30 +0100 (Thu, 18 Jan 2007) | 1 line add missing version entry ........ r53483 | neal.norwitz | 2007-01-18 07:20:55 +0100 (Thu, 18 Jan 2007) | 7 lines This test doesn't pass on Windows. The cause seems to be that chmod doesn't support the same funcationality as on Unix. I'm not sure if this fix is the best (or if it will even work)--it's a test to see if the buildbots start passing again. It might be better to not even run this test if it's windows (or non-posix). ........ r53488 | neal.norwitz | 2007-01-19 06:53:33 +0100 (Fri, 19 Jan 2007) | 1 line SF #1635217, Fix unbalanced paren ........ r53489 | martin.v.loewis | 2007-01-19 07:42:22 +0100 (Fri, 19 Jan 2007) | 3 lines Prefix AST symbols with _Py_. Fixes #1637022. Will backport. ........ r53497 | martin.v.loewis | 2007-01-19 19:01:38 +0100 (Fri, 19 Jan 2007) | 2 lines Add UUIDs for 2.5.1 and 2.5.2 ........ r53499 | raymond.hettinger | 2007-01-19 19:07:18 +0100 (Fri, 19 Jan 2007) | 1 line SF# 1635892: Fix docs for betavariate's input parameters . ........ r53503 | martin.v.loewis | 2007-01-20 15:05:39 +0100 (Sat, 20 Jan 2007) | 2 lines Merge 53501 and 53502 from 25 branch: Add /GS- for AMD64 and Itanium builds where missing. ........ r53504 | walter.doerwald | 2007-01-20 18:28:31 +0100 (Sat, 20 Jan 2007) | 2 lines Port test_resource.py to unittest. ........ r53505 | walter.doerwald | 2007-01-20 19:19:33 +0100 (Sat, 20 Jan 2007) | 2 lines Add argument tests an calls of resource.getrusage(). ........ r53506 | walter.doerwald | 2007-01-20 20:03:17 +0100 (Sat, 20 Jan 2007) | 2 lines resource.RUSAGE_BOTH might not exist. ........ r53507 | walter.doerwald | 2007-01-21 00:07:28 +0100 (Sun, 21 Jan 2007) | 2 lines Port test_new.py to unittest. ........ r53508 | martin.v.loewis | 2007-01-21 10:33:07 +0100 (Sun, 21 Jan 2007) | 2 lines Patch #1610575: Add support for _Bool to struct. ........ r53509 | georg.brandl | 2007-01-21 11:28:43 +0100 (Sun, 21 Jan 2007) | 3 lines Bug #1486663: don't reject keyword arguments for subclasses of builtin types. ........ r53511 | georg.brandl | 2007-01-21 11:35:10 +0100 (Sun, 21 Jan 2007) | 2 lines Patch #1627441: close sockets properly in urllib2. ........ r53517 | georg.brandl | 2007-01-22 20:40:21 +0100 (Mon, 22 Jan 2007) | 3 lines Use new email module names (#1637162, #1637159, #1637157). ........ r53518 | andrew.kuchling | 2007-01-22 21:26:40 +0100 (Mon, 22 Jan 2007) | 1 line Improve pattern used for mbox 'From' lines; add a simple test ........ r53519 | andrew.kuchling | 2007-01-22 21:27:50 +0100 (Mon, 22 Jan 2007) | 1 line Make comment match the code ........ r53522 | georg.brandl | 2007-01-22 22:10:33 +0100 (Mon, 22 Jan 2007) | 2 lines Bug #1249573: fix rfc822.parsedate not accepting a certain date format ........ r53524 | georg.brandl | 2007-01-22 22:23:41 +0100 (Mon, 22 Jan 2007) | 2 lines Bug #1627316: handle error in condition/ignore pdb commands more gracefully. ........ r53526 | lars.gustaebel | 2007-01-23 12:17:33 +0100 (Tue, 23 Jan 2007) | 4 lines Patch #1507247: tarfile.py: use current umask for intermediate directories. ........ r53527 | thomas.wouters | 2007-01-23 14:42:00 +0100 (Tue, 23 Jan 2007) | 13 lines SF patch #1630975: Fix crash when replacing sys.stdout in sitecustomize When running the interpreter in an environment that would cause it to set stdout/stderr/stdin's encoding, having a sitecustomize that would replace them with something other than PyFile objects would crash the interpreter. Fix it by simply ignoring the encoding-setting for non-files. This could do with a test, but I can think of no maintainable and portable way to test this bug, short of adding a sitecustomize.py to the buildsystem and have it always run with it (hmmm....) ........ r53528 | thomas.wouters | 2007-01-23 14:50:49 +0100 (Tue, 23 Jan 2007) | 4 lines Add news entry about last checkin (oops.) ........ r53531 | martin.v.loewis | 2007-01-23 22:11:47 +0100 (Tue, 23 Jan 2007) | 4 lines Make PyTraceBack_Here use the current thread, not the frame's thread state. Fixes #1579370. Will backport. ........ r53535 | brett.cannon | 2007-01-24 00:21:22 +0100 (Wed, 24 Jan 2007) | 5 lines Fix crasher for when an object's __del__ creates a new weakref to itself. Patch only fixes new-style classes; classic classes still buggy. Closes bug #1377858. Already backported. ........ r53536 | walter.doerwald | 2007-01-24 01:42:19 +0100 (Wed, 24 Jan 2007) | 2 lines Port test_popen.py to unittest. ........
982 lines
31 KiB
Python
Executable File
982 lines
31 KiB
Python
Executable File
#! /usr/bin/env python
|
|
"""Test the arraymodule.
|
|
Roger E. Masse
|
|
"""
|
|
|
|
import unittest
|
|
from test import test_support
|
|
from weakref import proxy
|
|
import array, cStringIO, math
|
|
from pickle import loads, dumps
|
|
|
|
class ArraySubclass(array.array):
|
|
pass
|
|
|
|
class ArraySubclassWithKwargs(array.array):
|
|
def __init__(self, typecode, newarg=None):
|
|
array.array.__init__(typecode)
|
|
|
|
tests = [] # list to accumulate all tests
|
|
typecodes = "cubBhHiIlLfd"
|
|
|
|
class BadConstructorTest(unittest.TestCase):
|
|
|
|
def test_constructor(self):
|
|
self.assertRaises(TypeError, array.array)
|
|
self.assertRaises(TypeError, array.array, spam=42)
|
|
self.assertRaises(TypeError, array.array, 'xx')
|
|
self.assertRaises(ValueError, array.array, 'x')
|
|
|
|
tests.append(BadConstructorTest)
|
|
|
|
class BaseTest(unittest.TestCase):
|
|
# Required class attributes (provided by subclasses
|
|
# typecode: the typecode to test
|
|
# example: an initializer usable in the constructor for this type
|
|
# smallerexample: the same length as example, but smaller
|
|
# biggerexample: the same length as example, but bigger
|
|
# outside: An entry that is not in example
|
|
# minitemsize: the minimum guaranteed itemsize
|
|
|
|
def assertEntryEqual(self, entry1, entry2):
|
|
self.assertEqual(entry1, entry2)
|
|
|
|
def badtypecode(self):
|
|
# Return a typecode that is different from our own
|
|
return typecodes[(typecodes.index(self.typecode)+1) % len(typecodes)]
|
|
|
|
def test_constructor(self):
|
|
a = array.array(self.typecode)
|
|
self.assertEqual(a.typecode, self.typecode)
|
|
self.assert_(a.itemsize>=self.minitemsize)
|
|
self.assertRaises(TypeError, array.array, self.typecode, None)
|
|
|
|
def test_len(self):
|
|
a = array.array(self.typecode)
|
|
a.append(self.example[0])
|
|
self.assertEqual(len(a), 1)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertEqual(len(a), len(self.example))
|
|
|
|
def test_buffer_info(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(TypeError, a.buffer_info, 42)
|
|
bi = a.buffer_info()
|
|
self.assert_(isinstance(bi, tuple))
|
|
self.assertEqual(len(bi), 2)
|
|
self.assert_(isinstance(bi[0], (int, int)))
|
|
self.assert_(isinstance(bi[1], int))
|
|
self.assertEqual(bi[1], len(a))
|
|
|
|
def test_byteswap(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(TypeError, a.byteswap, 42)
|
|
if a.itemsize in (1, 2, 4, 8):
|
|
b = array.array(self.typecode, self.example)
|
|
b.byteswap()
|
|
if a.itemsize==1:
|
|
self.assertEqual(a, b)
|
|
else:
|
|
self.assertNotEqual(a, b)
|
|
b.byteswap()
|
|
self.assertEqual(a, b)
|
|
|
|
def test_copy(self):
|
|
import copy
|
|
a = array.array(self.typecode, self.example)
|
|
b = copy.copy(a)
|
|
self.assertNotEqual(id(a), id(b))
|
|
self.assertEqual(a, b)
|
|
|
|
def test_deepcopy(self):
|
|
import copy
|
|
a = array.array(self.typecode, self.example)
|
|
b = copy.deepcopy(a)
|
|
self.assertNotEqual(id(a), id(b))
|
|
self.assertEqual(a, b)
|
|
|
|
def test_pickle(self):
|
|
for protocol in (0, 1, 2):
|
|
a = array.array(self.typecode, self.example)
|
|
b = loads(dumps(a, protocol))
|
|
self.assertNotEqual(id(a), id(b))
|
|
self.assertEqual(a, b)
|
|
|
|
a = ArraySubclass(self.typecode, self.example)
|
|
a.x = 10
|
|
b = loads(dumps(a, protocol))
|
|
self.assertNotEqual(id(a), id(b))
|
|
self.assertEqual(a, b)
|
|
self.assertEqual(a.x, b.x)
|
|
self.assertEqual(type(a), type(b))
|
|
|
|
def test_insert(self):
|
|
a = array.array(self.typecode, self.example)
|
|
a.insert(0, self.example[0])
|
|
self.assertEqual(len(a), 1+len(self.example))
|
|
self.assertEqual(a[0], a[1])
|
|
self.assertRaises(TypeError, a.insert)
|
|
self.assertRaises(TypeError, a.insert, None)
|
|
self.assertRaises(TypeError, a.insert, 0, None)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a.insert(-1, self.example[0])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(
|
|
self.typecode,
|
|
self.example[:-1] + self.example[:1] + self.example[-1:]
|
|
)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a.insert(-1000, self.example[0])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[:1] + self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a.insert(1000, self.example[0])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example + self.example[:1])
|
|
)
|
|
|
|
def test_tofromfile(self):
|
|
a = array.array(self.typecode, 2*self.example)
|
|
self.assertRaises(TypeError, a.tofile)
|
|
self.assertRaises(TypeError, a.tofile, cStringIO.StringIO())
|
|
f = open(test_support.TESTFN, 'wb')
|
|
try:
|
|
a.tofile(f)
|
|
f.close()
|
|
b = array.array(self.typecode)
|
|
f = open(test_support.TESTFN, 'rb')
|
|
self.assertRaises(TypeError, b.fromfile)
|
|
self.assertRaises(
|
|
TypeError,
|
|
b.fromfile,
|
|
cStringIO.StringIO(), len(self.example)
|
|
)
|
|
b.fromfile(f, len(self.example))
|
|
self.assertEqual(b, array.array(self.typecode, self.example))
|
|
self.assertNotEqual(a, b)
|
|
b.fromfile(f, len(self.example))
|
|
self.assertEqual(a, b)
|
|
self.assertRaises(EOFError, b.fromfile, f, 1)
|
|
f.close()
|
|
finally:
|
|
if not f.closed:
|
|
f.close()
|
|
test_support.unlink(test_support.TESTFN)
|
|
|
|
def test_tofromlist(self):
|
|
a = array.array(self.typecode, 2*self.example)
|
|
b = array.array(self.typecode)
|
|
self.assertRaises(TypeError, a.tolist, 42)
|
|
self.assertRaises(TypeError, b.fromlist)
|
|
self.assertRaises(TypeError, b.fromlist, 42)
|
|
self.assertRaises(TypeError, b.fromlist, [None])
|
|
b.fromlist(a.tolist())
|
|
self.assertEqual(a, b)
|
|
|
|
def test_tofromstring(self):
|
|
a = array.array(self.typecode, 2*self.example)
|
|
b = array.array(self.typecode)
|
|
self.assertRaises(TypeError, a.tostring, 42)
|
|
self.assertRaises(TypeError, b.fromstring)
|
|
self.assertRaises(TypeError, b.fromstring, 42)
|
|
b.fromstring(a.tostring())
|
|
self.assertEqual(a, b)
|
|
if a.itemsize>1:
|
|
self.assertRaises(ValueError, b.fromstring, "x")
|
|
|
|
def test_repr(self):
|
|
a = array.array(self.typecode, 2*self.example)
|
|
self.assertEqual(a, eval(repr(a), {"array": array.array}))
|
|
|
|
a = array.array(self.typecode)
|
|
self.assertEqual(repr(a), "array('%s')" % self.typecode)
|
|
|
|
def test_str(self):
|
|
a = array.array(self.typecode, 2*self.example)
|
|
str(a)
|
|
|
|
def test_cmp(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assert_((a == 42) is False)
|
|
self.assert_((a != 42) is True)
|
|
|
|
self.assert_((a == a) is True)
|
|
self.assert_((a != a) is False)
|
|
self.assert_((a < a) is False)
|
|
self.assert_((a <= a) is True)
|
|
self.assert_((a > a) is False)
|
|
self.assert_((a >= a) is True)
|
|
|
|
al = array.array(self.typecode, self.smallerexample)
|
|
ab = array.array(self.typecode, self.biggerexample)
|
|
|
|
self.assert_((a == 2*a) is False)
|
|
self.assert_((a != 2*a) is True)
|
|
self.assert_((a < 2*a) is True)
|
|
self.assert_((a <= 2*a) is True)
|
|
self.assert_((a > 2*a) is False)
|
|
self.assert_((a >= 2*a) is False)
|
|
|
|
self.assert_((a == al) is False)
|
|
self.assert_((a != al) is True)
|
|
self.assert_((a < al) is False)
|
|
self.assert_((a <= al) is False)
|
|
self.assert_((a > al) is True)
|
|
self.assert_((a >= al) is True)
|
|
|
|
self.assert_((a == ab) is False)
|
|
self.assert_((a != ab) is True)
|
|
self.assert_((a < ab) is True)
|
|
self.assert_((a <= ab) is True)
|
|
self.assert_((a > ab) is False)
|
|
self.assert_((a >= ab) is False)
|
|
|
|
def test_add(self):
|
|
a = array.array(self.typecode, self.example) \
|
|
+ array.array(self.typecode, self.example[::-1])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example + self.example[::-1])
|
|
)
|
|
|
|
b = array.array(self.badtypecode())
|
|
self.assertRaises(TypeError, a.__add__, b)
|
|
|
|
self.assertRaises(TypeError, a.__add__, "bad")
|
|
|
|
def test_iadd(self):
|
|
a = array.array(self.typecode, self.example[::-1])
|
|
b = a
|
|
a += array.array(self.typecode, 2*self.example)
|
|
self.assert_(a is b)
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[::-1]+2*self.example)
|
|
)
|
|
|
|
b = array.array(self.badtypecode())
|
|
self.assertRaises(TypeError, a.__add__, b)
|
|
|
|
self.assertRaises(TypeError, a.__iadd__, "bad")
|
|
|
|
def test_mul(self):
|
|
a = 5*array.array(self.typecode, self.example)
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, 5*self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)*5
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example*5)
|
|
)
|
|
|
|
a = 0*array.array(self.typecode, self.example)
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode)
|
|
)
|
|
|
|
a = (-1)*array.array(self.typecode, self.example)
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode)
|
|
)
|
|
|
|
self.assertRaises(TypeError, a.__mul__, "bad")
|
|
|
|
def test_imul(self):
|
|
a = array.array(self.typecode, self.example)
|
|
b = a
|
|
|
|
a *= 5
|
|
self.assert_(a is b)
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, 5*self.example)
|
|
)
|
|
|
|
a *= 0
|
|
self.assert_(a is b)
|
|
self.assertEqual(a, array.array(self.typecode))
|
|
|
|
a *= 1000
|
|
self.assert_(a is b)
|
|
self.assertEqual(a, array.array(self.typecode))
|
|
|
|
a *= -1
|
|
self.assert_(a is b)
|
|
self.assertEqual(a, array.array(self.typecode))
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a *= -1
|
|
self.assertEqual(a, array.array(self.typecode))
|
|
|
|
self.assertRaises(TypeError, a.__imul__, "bad")
|
|
|
|
def test_getitem(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertEntryEqual(a[0], self.example[0])
|
|
self.assertEntryEqual(a[0], self.example[0])
|
|
self.assertEntryEqual(a[-1], self.example[-1])
|
|
self.assertEntryEqual(a[-1], self.example[-1])
|
|
self.assertEntryEqual(a[len(self.example)-1], self.example[-1])
|
|
self.assertEntryEqual(a[-len(self.example)], self.example[0])
|
|
self.assertRaises(TypeError, a.__getitem__)
|
|
self.assertRaises(IndexError, a.__getitem__, len(self.example))
|
|
self.assertRaises(IndexError, a.__getitem__, -len(self.example)-1)
|
|
|
|
def test_setitem(self):
|
|
a = array.array(self.typecode, self.example)
|
|
a[0] = a[-1]
|
|
self.assertEntryEqual(a[0], a[-1])
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[0] = a[-1]
|
|
self.assertEntryEqual(a[0], a[-1])
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[-1] = a[0]
|
|
self.assertEntryEqual(a[0], a[-1])
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[-1] = a[0]
|
|
self.assertEntryEqual(a[0], a[-1])
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[len(self.example)-1] = a[0]
|
|
self.assertEntryEqual(a[0], a[-1])
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[-len(self.example)] = a[-1]
|
|
self.assertEntryEqual(a[0], a[-1])
|
|
|
|
self.assertRaises(TypeError, a.__setitem__)
|
|
self.assertRaises(TypeError, a.__setitem__, None)
|
|
self.assertRaises(TypeError, a.__setitem__, 0, None)
|
|
self.assertRaises(
|
|
IndexError,
|
|
a.__setitem__,
|
|
len(self.example), self.example[0]
|
|
)
|
|
self.assertRaises(
|
|
IndexError,
|
|
a.__setitem__,
|
|
-len(self.example)-1, self.example[0]
|
|
)
|
|
|
|
def test_delitem(self):
|
|
a = array.array(self.typecode, self.example)
|
|
del a[0]
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[1:])
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
del a[-1]
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[:-1])
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
del a[len(self.example)-1]
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[:-1])
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
del a[-len(self.example)]
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[1:])
|
|
)
|
|
|
|
self.assertRaises(TypeError, a.__delitem__)
|
|
self.assertRaises(TypeError, a.__delitem__, None)
|
|
self.assertRaises(IndexError, a.__delitem__, len(self.example))
|
|
self.assertRaises(IndexError, a.__delitem__, -len(self.example)-1)
|
|
|
|
def test_getslice(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertEqual(a[:], a)
|
|
|
|
self.assertEqual(
|
|
a[1:],
|
|
array.array(self.typecode, self.example[1:])
|
|
)
|
|
|
|
self.assertEqual(
|
|
a[:1],
|
|
array.array(self.typecode, self.example[:1])
|
|
)
|
|
|
|
self.assertEqual(
|
|
a[:-1],
|
|
array.array(self.typecode, self.example[:-1])
|
|
)
|
|
|
|
self.assertEqual(
|
|
a[-1:],
|
|
array.array(self.typecode, self.example[-1:])
|
|
)
|
|
|
|
self.assertEqual(
|
|
a[-1:-1],
|
|
array.array(self.typecode)
|
|
)
|
|
|
|
self.assertEqual(
|
|
a[2:1],
|
|
array.array(self.typecode)
|
|
)
|
|
|
|
self.assertEqual(
|
|
a[1000:],
|
|
array.array(self.typecode)
|
|
)
|
|
self.assertEqual(a[-1000:], a)
|
|
self.assertEqual(a[:1000], a)
|
|
self.assertEqual(
|
|
a[:-1000],
|
|
array.array(self.typecode)
|
|
)
|
|
self.assertEqual(a[-1000:1000], a)
|
|
self.assertEqual(
|
|
a[2000:1000],
|
|
array.array(self.typecode)
|
|
)
|
|
|
|
def test_setslice(self):
|
|
a = array.array(self.typecode, self.example)
|
|
a[:1] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example + self.example[1:])
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[:-1] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example + self.example[-1:])
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[-1:] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[:-1] + self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[1:] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[:1] + self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[1:-1] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(
|
|
self.typecode,
|
|
self.example[:1] + self.example + self.example[-1:]
|
|
)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[1000:] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, 2*self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[-1000:] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[:1000] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[:-1000] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, 2*self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[1:0] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[:1] + self.example + self.example[1:])
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a[2000:1000] = a
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, 2*self.example)
|
|
)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(TypeError, a.__setslice__, 0, 0, None)
|
|
self.assertRaises(TypeError, a.__setitem__, slice(0, 1), None)
|
|
|
|
b = array.array(self.badtypecode())
|
|
self.assertRaises(TypeError, a.__setslice__, 0, 0, b)
|
|
self.assertRaises(TypeError, a.__setitem__, slice(0, 1), b)
|
|
|
|
def test_index(self):
|
|
example = 2*self.example
|
|
a = array.array(self.typecode, example)
|
|
self.assertRaises(TypeError, a.index)
|
|
for x in example:
|
|
self.assertEqual(a.index(x), example.index(x))
|
|
self.assertRaises(ValueError, a.index, None)
|
|
self.assertRaises(ValueError, a.index, self.outside)
|
|
|
|
def test_count(self):
|
|
example = 2*self.example
|
|
a = array.array(self.typecode, example)
|
|
self.assertRaises(TypeError, a.count)
|
|
for x in example:
|
|
self.assertEqual(a.count(x), example.count(x))
|
|
self.assertEqual(a.count(self.outside), 0)
|
|
self.assertEqual(a.count(None), 0)
|
|
|
|
def test_remove(self):
|
|
for x in self.example:
|
|
example = 2*self.example
|
|
a = array.array(self.typecode, example)
|
|
pos = example.index(x)
|
|
example2 = example[:pos] + example[pos+1:]
|
|
a.remove(x)
|
|
self.assertEqual(a, array.array(self.typecode, example2))
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(ValueError, a.remove, self.outside)
|
|
|
|
self.assertRaises(ValueError, a.remove, None)
|
|
|
|
def test_pop(self):
|
|
a = array.array(self.typecode)
|
|
self.assertRaises(IndexError, a.pop)
|
|
|
|
a = array.array(self.typecode, 2*self.example)
|
|
self.assertRaises(TypeError, a.pop, 42, 42)
|
|
self.assertRaises(TypeError, a.pop, None)
|
|
self.assertRaises(IndexError, a.pop, len(a))
|
|
self.assertRaises(IndexError, a.pop, -len(a)-1)
|
|
|
|
self.assertEntryEqual(a.pop(0), self.example[0])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[1:]+self.example)
|
|
)
|
|
self.assertEntryEqual(a.pop(1), self.example[2])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[1:2]+self.example[3:]+self.example)
|
|
)
|
|
self.assertEntryEqual(a.pop(0), self.example[1])
|
|
self.assertEntryEqual(a.pop(), self.example[-1])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[3:]+self.example[:-1])
|
|
)
|
|
|
|
def test_reverse(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(TypeError, a.reverse, 42)
|
|
a.reverse()
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example[::-1])
|
|
)
|
|
|
|
def test_extend(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(TypeError, a.extend)
|
|
a.extend(array.array(self.typecode, self.example[::-1]))
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example+self.example[::-1])
|
|
)
|
|
|
|
b = array.array(self.badtypecode())
|
|
self.assertRaises(TypeError, a.extend, b)
|
|
|
|
a = array.array(self.typecode, self.example)
|
|
a.extend(self.example[::-1])
|
|
self.assertEqual(
|
|
a,
|
|
array.array(self.typecode, self.example+self.example[::-1])
|
|
)
|
|
|
|
def test_constructor_with_iterable_argument(self):
|
|
a = array.array(self.typecode, iter(self.example))
|
|
b = array.array(self.typecode, self.example)
|
|
self.assertEqual(a, b)
|
|
|
|
# non-iterable argument
|
|
self.assertRaises(TypeError, array.array, self.typecode, 10)
|
|
|
|
# pass through errors raised in __iter__
|
|
class A:
|
|
def __iter__(self):
|
|
raise UnicodeError
|
|
self.assertRaises(UnicodeError, array.array, self.typecode, A())
|
|
|
|
# pass through errors raised in next()
|
|
def B():
|
|
raise UnicodeError
|
|
yield None
|
|
self.assertRaises(UnicodeError, array.array, self.typecode, B())
|
|
|
|
def test_coveritertraverse(self):
|
|
try:
|
|
import gc
|
|
except ImportError:
|
|
return
|
|
a = array.array(self.typecode)
|
|
l = [iter(a)]
|
|
l.append(l)
|
|
gc.collect()
|
|
|
|
def test_buffer(self):
|
|
a = array.array(self.typecode, self.example)
|
|
b = buffer(a)
|
|
self.assertEqual(b[0], a.tostring()[0])
|
|
|
|
def test_weakref(self):
|
|
s = array.array(self.typecode, self.example)
|
|
p = proxy(s)
|
|
self.assertEqual(p.tostring(), s.tostring())
|
|
s = None
|
|
self.assertRaises(ReferenceError, len, p)
|
|
|
|
def test_bug_782369(self):
|
|
import sys
|
|
if hasattr(sys, "getrefcount"):
|
|
for i in range(10):
|
|
b = array.array('B', range(64))
|
|
rc = sys.getrefcount(10)
|
|
for i in range(10):
|
|
b = array.array('B', range(64))
|
|
self.assertEqual(rc, sys.getrefcount(10))
|
|
|
|
def test_subclass_with_kwargs(self):
|
|
# SF bug #1486663 -- this used to erroneously raise a TypeError
|
|
ArraySubclassWithKwargs('b', newarg=1)
|
|
|
|
|
|
class StringTest(BaseTest):
|
|
|
|
def test_setitem(self):
|
|
super(StringTest, self).test_setitem()
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(TypeError, a.__setitem__, 0, self.example[:2])
|
|
|
|
class CharacterTest(StringTest):
|
|
typecode = 'c'
|
|
example = '\x01azAZ\x00\xfe'
|
|
smallerexample = '\x01azAY\x00\xfe'
|
|
biggerexample = '\x01azAZ\x00\xff'
|
|
outside = '\x33'
|
|
minitemsize = 1
|
|
|
|
def test_subbclassing(self):
|
|
class EditableString(array.array):
|
|
def __new__(cls, s, *args, **kwargs):
|
|
return array.array.__new__(cls, 'c', s)
|
|
|
|
def __init__(self, s, color='blue'):
|
|
array.array.__init__(self, 'c', s)
|
|
self.color = color
|
|
|
|
def strip(self):
|
|
self[:] = array.array('c', self.tostring().strip())
|
|
|
|
def __repr__(self):
|
|
return 'EditableString(%r)' % self.tostring()
|
|
|
|
s = EditableString("\ttest\r\n")
|
|
s.strip()
|
|
self.assertEqual(s.tostring(), "test")
|
|
|
|
self.assertEqual(s.color, "blue")
|
|
s.color = "red"
|
|
self.assertEqual(s.color, "red")
|
|
self.assertEqual(s.__dict__.keys(), ["color"])
|
|
|
|
def test_nounicode(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(ValueError, a.fromunicode, unicode(''))
|
|
self.assertRaises(ValueError, a.tounicode)
|
|
|
|
tests.append(CharacterTest)
|
|
|
|
if test_support.have_unicode:
|
|
class UnicodeTest(StringTest):
|
|
typecode = 'u'
|
|
example = unicode(r'\x01\u263a\x00\ufeff', 'unicode-escape')
|
|
smallerexample = unicode(r'\x01\u263a\x00\ufefe', 'unicode-escape')
|
|
biggerexample = unicode(r'\x01\u263a\x01\ufeff', 'unicode-escape')
|
|
outside = unicode('\x33')
|
|
minitemsize = 2
|
|
|
|
def test_unicode(self):
|
|
self.assertRaises(TypeError, array.array, 'b', unicode('foo', 'ascii'))
|
|
|
|
a = array.array('u', unicode(r'\xa0\xc2\u1234', 'unicode-escape'))
|
|
a.fromunicode(unicode(' ', 'ascii'))
|
|
a.fromunicode(unicode('', 'ascii'))
|
|
a.fromunicode(unicode('', 'ascii'))
|
|
a.fromunicode(unicode(r'\x11abc\xff\u1234', 'unicode-escape'))
|
|
s = a.tounicode()
|
|
self.assertEqual(
|
|
s,
|
|
unicode(r'\xa0\xc2\u1234 \x11abc\xff\u1234', 'unicode-escape')
|
|
)
|
|
|
|
s = unicode(r'\x00="\'a\\b\x80\xff\u0000\u0001\u1234', 'unicode-escape')
|
|
a = array.array('u', s)
|
|
self.assertEqual(
|
|
repr(a),
|
|
r"""array('u', u'\x00="\'a\\b\x80\xff\x00\x01\u1234')"""
|
|
)
|
|
|
|
self.assertRaises(TypeError, a.fromunicode)
|
|
|
|
tests.append(UnicodeTest)
|
|
|
|
class NumberTest(BaseTest):
|
|
|
|
def test_extslice(self):
|
|
a = array.array(self.typecode, range(5))
|
|
self.assertEqual(a[::], a)
|
|
self.assertEqual(a[::2], array.array(self.typecode, [0,2,4]))
|
|
self.assertEqual(a[1::2], array.array(self.typecode, [1,3]))
|
|
self.assertEqual(a[::-1], array.array(self.typecode, [4,3,2,1,0]))
|
|
self.assertEqual(a[::-2], array.array(self.typecode, [4,2,0]))
|
|
self.assertEqual(a[3::-2], array.array(self.typecode, [3,1]))
|
|
self.assertEqual(a[-100:100:], a)
|
|
self.assertEqual(a[100:-100:-1], a[::-1])
|
|
self.assertEqual(a[-100:100:2], array.array(self.typecode, [0,2,4]))
|
|
self.assertEqual(a[1000:2000:2], array.array(self.typecode, []))
|
|
self.assertEqual(a[-1000:-2000:-2], array.array(self.typecode, []))
|
|
|
|
def test_delslice(self):
|
|
a = array.array(self.typecode, range(5))
|
|
del a[::2]
|
|
self.assertEqual(a, array.array(self.typecode, [1,3]))
|
|
a = array.array(self.typecode, range(5))
|
|
del a[1::2]
|
|
self.assertEqual(a, array.array(self.typecode, [0,2,4]))
|
|
a = array.array(self.typecode, range(5))
|
|
del a[1::-2]
|
|
self.assertEqual(a, array.array(self.typecode, [0,2,3,4]))
|
|
a = array.array(self.typecode, range(10))
|
|
del a[::1000]
|
|
self.assertEqual(a, array.array(self.typecode, [1,2,3,4,5,6,7,8,9]))
|
|
|
|
def test_assignment(self):
|
|
a = array.array(self.typecode, range(10))
|
|
a[::2] = array.array(self.typecode, [42]*5)
|
|
self.assertEqual(a, array.array(self.typecode, [42, 1, 42, 3, 42, 5, 42, 7, 42, 9]))
|
|
a = array.array(self.typecode, range(10))
|
|
a[::-4] = array.array(self.typecode, [10]*3)
|
|
self.assertEqual(a, array.array(self.typecode, [0, 10, 2, 3, 4, 10, 6, 7, 8 ,10]))
|
|
a = array.array(self.typecode, range(4))
|
|
a[::-1] = a
|
|
self.assertEqual(a, array.array(self.typecode, [3, 2, 1, 0]))
|
|
a = array.array(self.typecode, range(10))
|
|
b = a[:]
|
|
c = a[:]
|
|
ins = array.array(self.typecode, range(2))
|
|
a[2:3] = ins
|
|
b[slice(2,3)] = ins
|
|
c[2:3:] = ins
|
|
|
|
def test_iterationcontains(self):
|
|
a = array.array(self.typecode, range(10))
|
|
self.assertEqual(list(a), range(10))
|
|
b = array.array(self.typecode, [20])
|
|
self.assertEqual(a[-1] in a, True)
|
|
self.assertEqual(b[0] not in a, True)
|
|
|
|
def check_overflow(self, lower, upper):
|
|
# method to be used by subclasses
|
|
|
|
# should not overflow assigning lower limit
|
|
a = array.array(self.typecode, [lower])
|
|
a[0] = lower
|
|
# should overflow assigning less than lower limit
|
|
self.assertRaises(OverflowError, array.array, self.typecode, [lower-1])
|
|
self.assertRaises(OverflowError, a.__setitem__, 0, lower-1)
|
|
# should not overflow assigning upper limit
|
|
a = array.array(self.typecode, [upper])
|
|
a[0] = upper
|
|
# should overflow assigning more than upper limit
|
|
self.assertRaises(OverflowError, array.array, self.typecode, [upper+1])
|
|
self.assertRaises(OverflowError, a.__setitem__, 0, upper+1)
|
|
|
|
def test_subclassing(self):
|
|
typecode = self.typecode
|
|
class ExaggeratingArray(array.array):
|
|
__slots__ = ['offset']
|
|
|
|
def __new__(cls, typecode, data, offset):
|
|
return array.array.__new__(cls, typecode, data)
|
|
|
|
def __init__(self, typecode, data, offset):
|
|
self.offset = offset
|
|
|
|
def __getitem__(self, i):
|
|
return array.array.__getitem__(self, i) + self.offset
|
|
|
|
a = ExaggeratingArray(self.typecode, [3, 6, 7, 11], 4)
|
|
self.assertEntryEqual(a[0], 7)
|
|
|
|
self.assertRaises(AttributeError, setattr, a, "color", "blue")
|
|
|
|
class SignedNumberTest(NumberTest):
|
|
example = [-1, 0, 1, 42, 0x7f]
|
|
smallerexample = [-1, 0, 1, 42, 0x7e]
|
|
biggerexample = [-1, 0, 1, 43, 0x7f]
|
|
outside = 23
|
|
|
|
def test_overflow(self):
|
|
a = array.array(self.typecode)
|
|
lower = -1 * int(pow(2, a.itemsize * 8 - 1))
|
|
upper = int(pow(2, a.itemsize * 8 - 1)) - 1
|
|
self.check_overflow(lower, upper)
|
|
|
|
class UnsignedNumberTest(NumberTest):
|
|
example = [0, 1, 17, 23, 42, 0xff]
|
|
smallerexample = [0, 1, 17, 23, 42, 0xfe]
|
|
biggerexample = [0, 1, 17, 23, 43, 0xff]
|
|
outside = 0xaa
|
|
|
|
def test_overflow(self):
|
|
a = array.array(self.typecode)
|
|
lower = 0
|
|
upper = int(pow(2, a.itemsize * 8)) - 1
|
|
self.check_overflow(lower, upper)
|
|
|
|
|
|
class ByteTest(SignedNumberTest):
|
|
typecode = 'b'
|
|
minitemsize = 1
|
|
tests.append(ByteTest)
|
|
|
|
class UnsignedByteTest(UnsignedNumberTest):
|
|
typecode = 'B'
|
|
minitemsize = 1
|
|
tests.append(UnsignedByteTest)
|
|
|
|
class ShortTest(SignedNumberTest):
|
|
typecode = 'h'
|
|
minitemsize = 2
|
|
tests.append(ShortTest)
|
|
|
|
class UnsignedShortTest(UnsignedNumberTest):
|
|
typecode = 'H'
|
|
minitemsize = 2
|
|
tests.append(UnsignedShortTest)
|
|
|
|
class IntTest(SignedNumberTest):
|
|
typecode = 'i'
|
|
minitemsize = 2
|
|
tests.append(IntTest)
|
|
|
|
class UnsignedIntTest(UnsignedNumberTest):
|
|
typecode = 'I'
|
|
minitemsize = 2
|
|
tests.append(UnsignedIntTest)
|
|
|
|
class LongTest(SignedNumberTest):
|
|
typecode = 'l'
|
|
minitemsize = 4
|
|
tests.append(LongTest)
|
|
|
|
class UnsignedLongTest(UnsignedNumberTest):
|
|
typecode = 'L'
|
|
minitemsize = 4
|
|
tests.append(UnsignedLongTest)
|
|
|
|
class FPTest(NumberTest):
|
|
example = [-42.0, 0, 42, 1e5, -1e10]
|
|
smallerexample = [-42.0, 0, 42, 1e5, -2e10]
|
|
biggerexample = [-42.0, 0, 42, 1e5, 1e10]
|
|
outside = 23
|
|
|
|
def assertEntryEqual(self, entry1, entry2):
|
|
self.assertAlmostEqual(entry1, entry2)
|
|
|
|
def test_byteswap(self):
|
|
a = array.array(self.typecode, self.example)
|
|
self.assertRaises(TypeError, a.byteswap, 42)
|
|
if a.itemsize in (1, 2, 4, 8):
|
|
b = array.array(self.typecode, self.example)
|
|
b.byteswap()
|
|
if a.itemsize==1:
|
|
self.assertEqual(a, b)
|
|
else:
|
|
# On alphas treating the byte swapped bit patters as
|
|
# floats/doubles results in floating point exceptions
|
|
# => compare the 8bit string values instead
|
|
self.assertNotEqual(a.tostring(), b.tostring())
|
|
b.byteswap()
|
|
self.assertEqual(a, b)
|
|
|
|
class FloatTest(FPTest):
|
|
typecode = 'f'
|
|
minitemsize = 4
|
|
tests.append(FloatTest)
|
|
|
|
class DoubleTest(FPTest):
|
|
typecode = 'd'
|
|
minitemsize = 8
|
|
tests.append(DoubleTest)
|
|
|
|
def test_main(verbose=None):
|
|
import sys
|
|
|
|
test_support.run_unittest(*tests)
|
|
|
|
# verify reference counting
|
|
if verbose and hasattr(sys, "gettotalrefcount"):
|
|
import gc
|
|
counts = [None] * 5
|
|
for i in xrange(len(counts)):
|
|
test_support.run_unittest(*tests)
|
|
gc.collect()
|
|
counts[i] = sys.gettotalrefcount()
|
|
print counts
|
|
|
|
if __name__ == "__main__":
|
|
test_main(verbose=True)
|