()
| 2788 | |
| 2789 | |
| 2790 | def test_table_take_null_index(): |
| 2791 | table = pa.table( |
| 2792 | [pa.array([1, 2, 3, None, 5]), |
| 2793 | pa.array(['a', 'b', 'c', 'd', 'e'])], |
| 2794 | ['f1', 'f2']) |
| 2795 | |
| 2796 | result_with_null_index = pa.table( |
| 2797 | [pa.array([1, None]), |
| 2798 | pa.array(['a', None])], |
| 2799 | ['f1', 'f2']) |
| 2800 | |
| 2801 | assert table.take(pa.array([0, None])).equals(result_with_null_index) |
| 2802 | |
| 2803 | |
| 2804 | def test_table_take_non_consecutive(): |