| 1656 | |
| 1657 | @pytest.mark.parametrize("typ", ["array", "chunked_array"]) |
| 1658 | def test_compare_array(typ): |
| 1659 | if typ == "array": |
| 1660 | def con(values): |
| 1661 | return pa.array(values) |
| 1662 | else: |
| 1663 | def con(values): |
| 1664 | return pa.chunked_array([values]) |
| 1665 | |
| 1666 | arr1 = con([1, 2, 3, 4, None]) |
| 1667 | arr2 = con([1, 1, 4, None, 4]) |
| 1668 | |
| 1669 | result = pc.equal(arr1, arr2) |
| 1670 | assert result.equals(con([True, False, False, None, None])) |
| 1671 | |
| 1672 | result = pc.not_equal(arr1, arr2) |
| 1673 | assert result.equals(con([False, True, True, None, None])) |
| 1674 | |
| 1675 | result = pc.less(arr1, arr2) |
| 1676 | assert result.equals(con([False, False, True, None, None])) |
| 1677 | |
| 1678 | result = pc.less_equal(arr1, arr2) |
| 1679 | assert result.equals(con([True, False, True, None, None])) |
| 1680 | |
| 1681 | result = pc.greater(arr1, arr2) |
| 1682 | assert result.equals(con([False, True, False, None, None])) |
| 1683 | |
| 1684 | result = pc.greater_equal(arr1, arr2) |
| 1685 | assert result.equals(con([True, True, False, None, None])) |
| 1686 | |
| 1687 | |
| 1688 | @pytest.mark.parametrize("typ", ["array", "chunked_array"]) |