1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
| import pytest
|
| import pandas as pd
| from pandas.tests.arrays.masked_shared import (
| ComparisonOps,
| NumericOps,
| )
|
|
| class TestComparisonOps(NumericOps, ComparisonOps):
| @pytest.mark.parametrize("other", [True, False, pd.NA, -1, 0, 1])
| def test_scalar(self, other, comparison_op, dtype):
| ComparisonOps.test_scalar(self, other, comparison_op, dtype)
|
| def test_compare_to_int(self, dtype, comparison_op):
| # GH 28930
| op_name = f"__{comparison_op.__name__}__"
| s1 = pd.Series([1, None, 3], dtype=dtype)
| s2 = pd.Series([1, None, 3], dtype="float")
|
| method = getattr(s1, op_name)
| result = method(2)
|
| method = getattr(s2, op_name)
| expected = method(2).astype("boolean")
| expected[s2.isna()] = pd.NA
|
| self.assert_series_equal(result, expected)
|
|
| def test_equals():
| # GH-30652
| # equals is generally tested in /tests/extension/base/methods, but this
| # specifically tests that two arrays of the same class but different dtype
| # do not evaluate equal
| a1 = pd.array([1, 2, None], dtype="Int64")
| a2 = pd.array([1, 2, None], dtype="Int32")
| assert a1.equals(a2) is False
|
|
