def test_buffer_resize(self):
data = np.zeros(10)
B = Buffer(data=data)
data = np.zeros(20)
B._need_resize = False
B.set_data(data)
assert B.nbytes == data.nbytes
assert B._need_resize is True
def test_data_copy(self):
data = np.zeros(100)
B = Buffer(data=data)
B.set_data(data=data[:50], offset=0, copy=True)
assert B._pending_data[-1][0].base is not data
def test_data_storage(self):
data = np.zeros(100)
B = Buffer(data=data)
B.set_data(data=data[:50], offset=0, copy=False)
assert B._pending_data[-1][0].base is data
def test_set_whole_data(self):
data = np.zeros(100)
B = Buffer(data=data)
B.set_data(data=data)
assert len(B._pending_data) == 1
