()
| 2755 | |
| 2756 | |
| 2757 | def test_table_take_non_consecutive(): |
| 2758 | table = pa.table( |
| 2759 | [pa.array([1, 2, 3, None, 5]), |
| 2760 | pa.array(['a', 'b', 'c', 'd', 'e'])], |
| 2761 | ['f1', 'f2']) |
| 2762 | |
| 2763 | result_non_consecutive = pa.table( |
| 2764 | [pa.array([2, None]), |
| 2765 | pa.array(['b', 'd'])], |
| 2766 | ['f1', 'f2']) |
| 2767 | |
| 2768 | assert table.take(pa.array([1, 3])).equals(result_non_consecutive) |
| 2769 | |
| 2770 | |
| 2771 | def test_table_select(): |