()
| 2741 | |
| 2742 | |
| 2743 | def test_table_take_null_index(): |
| 2744 | table = pa.table( |
| 2745 | [pa.array([1, 2, 3, None, 5]), |
| 2746 | pa.array(['a', 'b', 'c', 'd', 'e'])], |
| 2747 | ['f1', 'f2']) |
| 2748 | |
| 2749 | result_with_null_index = pa.table( |
| 2750 | [pa.array([1, None]), |
| 2751 | pa.array(['a', None])], |
| 2752 | ['f1', 'f2']) |
| 2753 | |
| 2754 | assert table.take(pa.array([0, None])).equals(result_with_null_index) |
| 2755 | |
| 2756 | |
| 2757 | def test_table_take_non_consecutive(): |