1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
|
from rpython.rlib.buffer import RawByteBuffer
from pypy.interpreter.buffer import RawBufferView, SimpleView
def test_RawBufferView_basic():
buf = RawByteBuffer(10)
view = RawBufferView(buf, 'ignored', 2)
assert view.getlength() == 10
assert view.getformat() == 'ignored'
assert view.getitemsize() == 2
assert view.getndim() == 1
assert view.getshape() == [5]
assert view.getstrides() == [2]
assert view.as_readbuf() is view.as_writebuf() is buf
def test_SimpleView_basic():
buf = RawByteBuffer(10)
view = SimpleView(buf)
assert view.getlength() == 10
assert view.getformat() == 'B'
assert view.getitemsize() == 1
assert view.getndim() == 1
assert view.getshape() == [10]
assert view.getstrides() == [1]
assert view.as_readbuf() is view.as_writebuf() is buf
def test_SimpleView_basic_w_obj():
buf = RawByteBuffer(10)
view = SimpleView(buf, w_obj="fake obj")
assert view.w_obj == "fake obj"
|