()
| 703 | |
| 704 | |
| 705 | def test_recordbatch_take(): |
| 706 | batch = pa.record_batch( |
| 707 | [pa.array([1, 2, 3, None, 5]), |
| 708 | pa.array(['a', 'b', 'c', 'd', 'e'])], |
| 709 | ['f1', 'f2']) |
| 710 | assert batch.take(pa.array([2, 3])).equals(batch.slice(2, 2)) |
| 711 | assert batch.take(pa.array([2, None])).equals( |
| 712 | pa.record_batch([pa.array([3, None]), pa.array(['c', None])], |
| 713 | ['f1', 'f2'])) |
| 714 | |
| 715 | |
| 716 | def test_recordbatch_column_sets_private_name(): |