| 1802 | |
| 1803 | @pytest.mark.parametrize("typ", ["array", "chunked_array"]) |
| 1804 | def test_compare_array(typ): |
| 1805 | if typ == "array": |
| 1806 | def con(values): |
| 1807 | return pa.array(values) |
| 1808 | else: |
| 1809 | def con(values): |
| 1810 | return pa.chunked_array([values]) |
| 1811 | |
| 1812 | arr1 = con([1, 2, 3, 4, None]) |
| 1813 | arr2 = con([1, 1, 4, None, 4]) |
| 1814 | |
| 1815 | result = pc.equal(arr1, arr2) |
| 1816 | assert result.equals(con([True, False, False, None, None])) |
| 1817 | |
| 1818 | result = pc.not_equal(arr1, arr2) |
| 1819 | assert result.equals(con([False, True, True, None, None])) |
| 1820 | |
| 1821 | result = pc.less(arr1, arr2) |
| 1822 | assert result.equals(con([False, False, True, None, None])) |
| 1823 | |
| 1824 | result = pc.less_equal(arr1, arr2) |
| 1825 | assert result.equals(con([True, False, True, None, None])) |
| 1826 | |
| 1827 | result = pc.greater(arr1, arr2) |
| 1828 | assert result.equals(con([False, True, False, None, None])) |
| 1829 | |
| 1830 | result = pc.greater_equal(arr1, arr2) |
| 1831 | assert result.equals(con([True, True, False, None, None])) |
| 1832 | |
| 1833 | |
| 1834 | @pytest.mark.parametrize("typ", ["array", "chunked_array"]) |