()
| 2802 | |
| 2803 | |
| 2804 | def test_table_take_non_consecutive(): |
| 2805 | table = pa.table( |
| 2806 | [pa.array([1, 2, 3, None, 5]), |
| 2807 | pa.array(['a', 'b', 'c', 'd', 'e'])], |
| 2808 | ['f1', 'f2']) |
| 2809 | |
| 2810 | result_non_consecutive = pa.table( |
| 2811 | [pa.array([2, None]), |
| 2812 | pa.array(['b', 'd'])], |
| 2813 | ['f1', 'f2']) |
| 2814 | |
| 2815 | assert table.take(pa.array([1, 3])).equals(result_non_consecutive) |
| 2816 | |
| 2817 | |
| 2818 | def test_table_select(): |