| 4926 | } |
| 4927 | |
| 4928 | void AssertStreamNext(struct ArrowDeviceArrayStream* c_stream, |
| 4929 | const RecordBatch& expected) { |
| 4930 | struct ArrowDeviceArray c_array; |
| 4931 | ASSERT_EQ(0, c_stream->get_next(c_stream, &c_array)); |
| 4932 | |
| 4933 | DeviceArrayExportGuard guard(&c_array); |
| 4934 | ASSERT_FALSE(ArrowDeviceArrayIsReleased(&c_array)); |
| 4935 | |
| 4936 | ASSERT_OK_AND_ASSIGN(auto batch, |
| 4937 | ImportDeviceRecordBatch(&c_array, expected.schema(), |
| 4938 | TestDeviceArrayRoundtrip::DeviceMapper)); |
| 4939 | AssertBatchesEqual(expected, *batch); |
| 4940 | } |
| 4941 | |
| 4942 | void AssertStreamNext(struct ArrowDeviceArrayStream* c_stream, const Array& expected) { |
| 4943 | struct ArrowDeviceArray c_array; |
nothing calls this directly
no test coverage detected