()
| 658 | |
| 659 | |
| 660 | def test_BufferReader(): |
| 661 | size = 1000 |
| 662 | arr, cbuf = make_random_buffer(size=size, target='device') |
| 663 | |
| 664 | reader = cuda.BufferReader(cbuf) |
| 665 | reader.seek(950) |
| 666 | assert reader.tell() == 950 |
| 667 | |
| 668 | data = reader.read(100) |
| 669 | assert len(data) == 50 |
| 670 | assert reader.tell() == 1000 |
| 671 | |
| 672 | reader.seek(925) |
| 673 | arr2 = np.zeros(100, dtype=np.uint8) |
| 674 | n = reader.readinto(arr2) |
| 675 | assert n == 75 |
| 676 | assert reader.tell() == 1000 |
| 677 | np.testing.assert_equal(arr[925:], arr2[:75]) |
| 678 | |
| 679 | reader.seek(0) |
| 680 | assert reader.tell() == 0 |
| 681 | buf2 = reader.read_buffer() |
| 682 | arr2 = np.frombuffer(buf2.copy_to_host(), dtype=np.uint8) |
| 683 | np.testing.assert_equal(arr, arr2) |
| 684 | |
| 685 | |
| 686 | def test_BufferReader_zero_size(): |
nothing calls this directly
no test coverage detected