| 1870 | |
| 1871 | @pytest.mark.parametrize("typ", ["array", "chunked_array"]) |
| 1872 | def test_compare_scalar(typ): |
| 1873 | if typ == "array": |
| 1874 | def con(values): |
| 1875 | return pa.array(values) |
| 1876 | else: |
| 1877 | def con(values): |
| 1878 | return pa.chunked_array([values]) |
| 1879 | |
| 1880 | arr = con([1, 2, 3, None]) |
| 1881 | scalar = pa.scalar(2) |
| 1882 | |
| 1883 | result = pc.equal(arr, scalar) |
| 1884 | assert result.equals(con([False, True, False, None])) |
| 1885 | |
| 1886 | if typ == "array": |
| 1887 | nascalar = pa.scalar(None, type="int64") |
| 1888 | result = pc.equal(arr, nascalar) |
| 1889 | assert result.to_pylist() == [None, None, None, None] |
| 1890 | |
| 1891 | result = pc.not_equal(arr, scalar) |
| 1892 | assert result.equals(con([True, False, True, None])) |
| 1893 | |
| 1894 | result = pc.less(arr, scalar) |
| 1895 | assert result.equals(con([True, False, False, None])) |
| 1896 | |
| 1897 | result = pc.less_equal(arr, scalar) |
| 1898 | assert result.equals(con([True, True, False, None])) |
| 1899 | |
| 1900 | result = pc.greater(arr, scalar) |
| 1901 | assert result.equals(con([False, False, True, None])) |
| 1902 | |
| 1903 | result = pc.greater_equal(arr, scalar) |
| 1904 | assert result.equals(con([False, True, True, None])) |
| 1905 | |
| 1906 | |
| 1907 | def test_compare_chunked_array_mixed(): |