()
| 2732 | |
| 2733 | |
| 2734 | def test_table_take_vanilla_functionality(): |
| 2735 | table = pa.table( |
| 2736 | [pa.array([1, 2, 3, None, 5]), |
| 2737 | pa.array(['a', 'b', 'c', 'd', 'e'])], |
| 2738 | ['f1', 'f2']) |
| 2739 | |
| 2740 | assert table.take(pa.array([2, 3])).equals(table.slice(2, 2)) |
| 2741 | |
| 2742 | |
| 2743 | def test_table_take_null_index(): |