(batch, expected_schema)
| 3692 | |
| 3693 | |
| 3694 | def verify_cuda_recordbatch(batch, expected_schema): |
| 3695 | batch.validate() |
| 3696 | assert batch.device_type == pa.DeviceAllocationType.CUDA |
| 3697 | assert batch.is_cpu is False |
| 3698 | assert batch.num_columns == len(expected_schema.names) |
| 3699 | assert batch.column_names == expected_schema.names |
| 3700 | assert str(batch) in repr(batch) |
| 3701 | for c in batch.columns: |
| 3702 | assert c.device_type == pa.DeviceAllocationType.CUDA |
| 3703 | assert batch.schema == expected_schema |
| 3704 | |
| 3705 | |
| 3706 | def test_recordbatch_non_cpu(cuda_context, cpu_recordbatch, cuda_recordbatch, |
no test coverage detected