Commit ded26a8c authored by Robert Bradshaw's avatar Robert Bradshaw

More Py3 test fixes.

parent c7a7c577
...@@ -117,13 +117,12 @@ def test_view(): ...@@ -117,13 +117,12 @@ def test_view():
def test_extend(): def test_extend():
""" """
>>> a = array.array('f', [1.0, 2.0, 3.0])
>>> test_extend() >>> test_extend()
""" """
cdef array.array ca = array.array('i', [1, 2, 3]) cdef array.array ca = array.array('i', [1, 2, 3])
cdef array.array cb = array.array('i', range(4, 6)) cdef array.array cb = array.array('i', [4, 5])
array.extend(ca, cb) array.extend(ca, cb)
assert list(ca) == range(1, 6), list(ca) assert list(ca) == [1, 2, 3, 4, 5], list(ca)
def test_likes(a): def test_likes(a):
""" """
...@@ -138,14 +137,15 @@ def test_likes(a): ...@@ -138,14 +137,15 @@ def test_likes(a):
def test_extend_buffer(): def test_extend_buffer():
""" """
>>> a = array.array('f', [1.0, 2.0, 3.0])
>>> test_extend_buffer() >>> test_extend_buffer()
array('c', 'abcdefghij') array('l', [15, 37, 389, 5077])
""" """
cdef array.array ca = array.array('c', "abcdef") cdef array.array ca = array.array('l', [15, 37])
cdef char* s = "ghij" cdef long[2] s
array.extend_buffer(ca, s, len(s)) # or use stdlib.strlen s[0] = 389
s[1] = 5077
array.extend_buffer(ca, <char*> &s, 2)
assert ca._c[9] == 'j' assert ca._L[3] == 5077
assert len(ca) == 10 assert len(ca) == 4
return ca return ca
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment