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