| 4867 | } |
| 4868 | |
| 4869 | void AssertStreamNext(struct ArrowDeviceArrayStream* c_stream, |
| 4870 | const RecordBatch& expected) { |
| 4871 | struct ArrowDeviceArray c_array; |
| 4872 | ASSERT_EQ(0, c_stream->get_next(c_stream, &c_array)); |
| 4873 | |
| 4874 | DeviceArrayExportGuard guard(&c_array); |
| 4875 | ASSERT_FALSE(ArrowDeviceArrayIsReleased(&c_array)); |
| 4876 | |
| 4877 | ASSERT_OK_AND_ASSIGN(auto batch, |
| 4878 | ImportDeviceRecordBatch(&c_array, expected.schema(), |
| 4879 | TestDeviceArrayRoundtrip::DeviceMapper)); |
| 4880 | AssertBatchesEqual(expected, *batch); |
| 4881 | } |
| 4882 | |
| 4883 | void AssertStreamNext(struct ArrowDeviceArrayStream* c_stream, const Array& expected) { |
| 4884 | struct ArrowDeviceArray c_array; |
nothing calls this directly
no test coverage detected