| 1724 | |
| 1725 | @pytest.mark.parametrize("typ", ["array", "chunked_array"]) |
| 1726 | def test_compare_scalar(typ): |
| 1727 | if typ == "array": |
| 1728 | def con(values): |
| 1729 | return pa.array(values) |
| 1730 | else: |
| 1731 | def con(values): |
| 1732 | return pa.chunked_array([values]) |
| 1733 | |
| 1734 | arr = con([1, 2, 3, None]) |
| 1735 | scalar = pa.scalar(2) |
| 1736 | |
| 1737 | result = pc.equal(arr, scalar) |
| 1738 | assert result.equals(con([False, True, False, None])) |
| 1739 | |
| 1740 | if typ == "array": |
| 1741 | nascalar = pa.scalar(None, type="int64") |
| 1742 | result = pc.equal(arr, nascalar) |
| 1743 | assert result.to_pylist() == [None, None, None, None] |
| 1744 | |
| 1745 | result = pc.not_equal(arr, scalar) |
| 1746 | assert result.equals(con([True, False, True, None])) |
| 1747 | |
| 1748 | result = pc.less(arr, scalar) |
| 1749 | assert result.equals(con([True, False, False, None])) |
| 1750 | |
| 1751 | result = pc.less_equal(arr, scalar) |
| 1752 | assert result.equals(con([True, True, False, None])) |
| 1753 | |
| 1754 | result = pc.greater(arr, scalar) |
| 1755 | assert result.equals(con([False, False, True, None])) |
| 1756 | |
| 1757 | result = pc.greater_equal(arr, scalar) |
| 1758 | assert result.equals(con([False, True, True, None])) |
| 1759 | |
| 1760 | |
| 1761 | def test_compare_chunked_array_mixed(): |