#! /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')
    testtype('u', u'\u263a')
    for type in (['b', 'h', 'i', 'l', 'f', 'd']):
        testtype(type, 1)
    testunicode()
    testsubclassing()
    unlink(TESTFN)

def testunicode():
    try:
        array.array('b', u'foo')
    except TypeError:
        pass
    else:
        raise TestFailed("creating a non-unicode array from "
                         "a Unicode string should fail")

    x = array.array('u', u'\xa0\xc2\u1234')
    x.fromunicode(u' ')
    x.fromunicode(u'')
    x.fromunicode(u'')
    x.fromunicode(u'\x11abc\xff\u1234')
    s = x.tounicode()
    if s != u'\xa0\xc2\u1234 \x11abc\xff\u1234':
        raise TestFailed("fromunicode()/tounicode()")

    s = u'\x00="\'a\\b\x80\xff\u0000\u0001\u1234'
    a = array.array('u', s)
    if verbose:
        print "repr of type 'u' array:", repr(a)
        print "              expected: array('u', %r)" % s

def testsubclassing():
    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()
    if s.tostring() != 'test':
        raise TestFailed, "subclassing array.array failed somewhere"
    if s.color != 'blue':
        raise TestFailed, "assigning attributes to instance of array subclass"
    s.color = 'red'
    if s.color != 'red':
        raise TestFailed, "assigning attributes to instance of array subclass"
    if s.__dict__.keys() != ['color']:
        raise TestFailed, "array subclass __dict__"

    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('i', [3, 6, 7, 11], 4)
    if a[0] != 7:
        raise TestFailed, "array subclass overriding __getitem__"
    try:
        a.color = 'blue'
    except AttributeError:
        pass
    else:
        raise TestFailed, "array subclass __slots__ was ignored"


def testoverflow(type, lowerLimit, upperLimit):
        # should not overflow assigning lower limit
    if verbose:
        print "overflow test: array(%s, [%s])" % (`type`, `lowerLimit`)
    try:
        a = array.array(type, [lowerLimit])
    except:
        raise TestFailed, "array(%s) overflowed assigning %s" %\
                (`type`, `lowerLimit`)
    # should overflow assigning less than lower limit
    if verbose:
        print "overflow test: array(%s, [%s])" % (`type`, `lowerLimit-1`)
    try:
        a = array.array(type, [lowerLimit-1])
        raise TestFailed, "array(%s) did not overflow assigning %s" %\
                (`type`, `lowerLimit-1`)
    except OverflowError:
        pass
    # should not overflow assigning upper limit
    if verbose:
        print "overflow test: array(%s, [%s])" % (`type`, `upperLimit`)
    try:
        a = array.array(type, [upperLimit])
    except:
        raise TestFailed, "array(%s) overflowed assigning %s" %\
                (`type`, `upperLimit`)
    # should overflow assigning more than upper limit
    if verbose:
        print "overflow test: array(%s, [%s])" % (`type`, `upperLimit+1`)
    try:
        a = array.array(type, [upperLimit+1])
        raise TestFailed, "array(%s) did not overflow assigning %s" %\
                (`type`, `upperLimit+1`)
    except OverflowError:
        pass


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()

    # This block is just to verify that the operations don't blow up.
    a.tolist()
    a.tostring()
    repr(a)
    str(a)

    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()`

    # Try out inplace addition and multiplication
    a = array.array(type, [example])
    b = a
    a += array.array(type, [example]*2)
    if a is not b:
        raise TestFailed, "array(%s) inplace addition" % `type`
    if a != array.array(type, [example] * 3):
        raise TestFailed, "array(%s) inplace addition" % `type`

    a *= 5
    if a is not b:
        raise TestFailed, "array(%s) inplace multiplication" % `type`
    if a != array.array(type, [example] * 15):
        raise TestFailed, "array(%s) inplace multiplication" % `type`

    a *= 0
    if a is not b:
        raise TestFailed, "array(%s) inplace multiplication by 0" % `type`
    if a != array.array(type, []):
        raise TestFailed, "array(%s) inplace multiplication by 0" % `type`

    a *= 1000
    if a is not b:
        raise TestFailed, "empty array(%s) inplace multiplication" % `type`
    if a != array.array(type, []):
        raise TestFailed, "empty array(%s) inplace multiplication" % `type`

    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`
        if a.index("e") != 5:
            raise TestFailed, "array(%s) index-test" % `type`
        if a.count("a") != 2:
            raise TestFailed, "array(%s) count-test" % `type`
        a.remove("e")
        if a != array.array(type, "aabcde"):
            raise TestFailed, "array(%s) remove-test" % `type`
        if a.pop(0) != "a":
            raise TestFailed, "array(%s) pop-test" % `type`
        if a.pop(1) != "b":
            raise TestFailed, "array(%s) pop-test" % `type`
        a.extend(array.array(type, "xyz"))
        if a != array.array(type, "acdexyz"):
            raise TestFailed, "array(%s) extend-test" % `type`
        a.pop()
        a.pop()
        a.pop()
        x = a.pop()
        if x != 'e':
            raise TestFailed, "array(%s) pop-test" % `type`
        if a != array.array(type, "acd"):
            raise TestFailed, "array(%s) pop-test" % `type`
        a.reverse()
        if a != array.array(type, "dca"):
            raise TestFailed, "array(%s) reverse-test" % `type`
    elif type == 'u':
        a = array.array(type, u"abcde")
        a[:-1] = a
        if a != array.array(type, u"abcdee"):
            raise TestFailed, "array(%s) self-slice-assign (head)" % `type`
        a = array.array(type, u"abcde")
        a[1:] = a
        if a != array.array(type, u"aabcde"):
            raise TestFailed, "array(%s) self-slice-assign (tail)" % `type`
        a = array.array(type, u"abcde")
        a[1:-1] = a
        if a != array.array(type, u"aabcdee"):
            raise TestFailed, "array(%s) self-slice-assign (cntr)" % `type`
        if a.index(u"e") != 5:
            raise TestFailed, "array(%s) index-test" % `type`
        if a.count(u"a") != 2:
            raise TestFailed, "array(%s) count-test" % `type`
        a.remove(u"e")
        if a != array.array(type, u"aabcde"):
            raise TestFailed, "array(%s) remove-test" % `type`
        if a.pop(0) != u"a":
            raise TestFailed, "array(%s) pop-test" % `type`
        if a.pop(1) != u"b":
            raise TestFailed, "array(%s) pop-test" % `type`
        a.extend(array.array(type, u"xyz"))
        if a != array.array(type, u"acdexyz"):
            raise TestFailed, "array(%s) extend-test" % `type`
        a.pop()
        a.pop()
        a.pop()
        x = a.pop()
        if x != u'e':
            raise TestFailed, "array(%s) pop-test" % `type`
        if a != array.array(type, u"acd"):
            raise TestFailed, "array(%s) pop-test" % `type`
        a.reverse()
        if a != array.array(type, u"dca"):
            raise TestFailed, "array(%s) reverse-test" % `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`
        if a.index(5) != 5:
            raise TestFailed, "array(%s) index-test" % `type`
        if a.count(1) != 2:
            raise TestFailed, "array(%s) count-test" % `type`
        a.remove(5)
        if a != array.array(type, [1, 1, 2, 3, 4, 5]):
            raise TestFailed, "array(%s) remove-test" % `type`
        if a.pop(0) != 1:
            raise TestFailed, "array(%s) pop-test" % `type`
        if a.pop(1) != 2:
            raise TestFailed, "array(%s) pop-test" % `type`
        a.extend(array.array(type, [7, 8, 9]))
        if a != array.array(type, [1, 3, 4, 5, 7, 8, 9]):
            raise TestFailed, "array(%s) extend-test" % `type`
        a.pop()
        a.pop()
        a.pop()
        x = a.pop()
        if x != 5:
            raise TestFailed, "array(%s) pop-test" % `type`
        if a != array.array(type, [1, 3, 4]):
            raise TestFailed, "array(%s) pop-test" % `type`
        a.reverse()
        if a != array.array(type, [4, 3, 1]):
            raise TestFailed, "array(%s) reverse-test" % `type`

    # test that overflow exceptions are raised as expected for assignment
    # to array of specific integral types
    from math import pow
    if type in ('b', 'h', 'i', 'l'):
        # check signed and unsigned versions
        a = array.array(type)
        signedLowerLimit = -1 * long(pow(2, a.itemsize * 8 - 1))
        signedUpperLimit = long(pow(2, a.itemsize * 8 - 1)) - 1L
        unsignedLowerLimit = 0
        unsignedUpperLimit = long(pow(2, a.itemsize * 8)) - 1L
        testoverflow(type, signedLowerLimit, signedUpperLimit)
        testoverflow(type.upper(), unsignedLowerLimit, unsignedUpperLimit)


main()