def test_buffered_output_stream_detach():
np_buf = np.zeros(100, dtype=np.int8) # zero-initialized buffer
buf = pa.py_buffer(np_buf)
f = pa.BufferedOutputStream(pa.FixedSizeBufferWriter(buf), buffer_size=4)
f.write(b"12")
assert np_buf[:4].tobytes() == b'\0\0\0\0'
raw = f.detach()
assert f.closed
assert not raw.closed
assert np_buf[:4].tobytes() == b'12\0\0'
def test_buffered_output_stream():
np_buf = np.zeros(100, dtype=np.int8) # zero-initialized buffer
buf = pa.py_buffer(np_buf)
raw = pa.FixedSizeBufferWriter(buf)
f = pa.BufferedOutputStream(raw, buffer_size=4)
f.write(b"12")
assert np_buf[:4].tobytes() == b'\0\0\0\0'
f.flush()
assert np_buf[:4].tobytes() == b'12\0\0'
f.write(b"3456789")
f.close()
assert f.closed
assert raw.closed
assert np_buf[:10].tobytes() == b'123456789\0'
