(batch, expected_schema)
| 3803 | |
| 3804 | |
| 3805 | def verify_cuda_recordbatch(batch, expected_schema): |
| 3806 | batch.validate() |
| 3807 | assert batch.device_type == pa.DeviceAllocationType.CUDA |
| 3808 | assert batch.is_cpu is False |
| 3809 | assert batch.num_columns == len(expected_schema.names) |
| 3810 | assert batch.column_names == expected_schema.names |
| 3811 | assert str(batch) in repr(batch) |
| 3812 | for c in batch.columns: |
| 3813 | assert c.device_type == pa.DeviceAllocationType.CUDA |
| 3814 | assert batch.schema == expected_schema |
| 3815 | |
| 3816 | |
| 3817 | def test_recordbatch_non_cpu(cuda_context, cpu_recordbatch, cuda_recordbatch, |
no test coverage detected