| 273 | |
| 274 | |
| 275 | def test_bytes_reader(): |
| 276 | # Like a BytesIO, but zero-copy underneath for C++ consumers |
| 277 | data = b'some sample data' |
| 278 | f = pa.BufferReader(data) |
| 279 | assert f.tell() == 0 |
| 280 | |
| 281 | assert f.size() == len(data) |
| 282 | |
| 283 | assert f.read(4) == b'some' |
| 284 | assert f.tell() == 4 |
| 285 | |
| 286 | f.seek(0) |
| 287 | assert f.tell() == 0 |
| 288 | |
| 289 | f.seek(0, 2) |
| 290 | assert f.tell() == len(data) |
| 291 | |
| 292 | f.seek(5) |
| 293 | assert f.tell() == 5 |
| 294 | |
| 295 | assert f.read(50) == b'sample data' |
| 296 | |
| 297 | assert not f.closed |
| 298 | f.close() |
| 299 | assert f.closed |
| 300 | |
| 301 | |
| 302 | def test_bytes_reader_non_bytes(): |