|
40 | 40 | ensure_index,
|
41 | 41 | ensure_index_from_sequences,
|
42 | 42 | )
|
| 43 | +from pandas.testing import assert_series_equal |
43 | 44 |
|
44 | 45 |
|
45 | 46 | class TestIndex:
|
@@ -1718,33 +1719,37 @@ def test_is_monotonic_pyarrow_list_type():
|
1718 | 1719 | assert not idx.is_monotonic_increasing
|
1719 | 1720 | assert not idx.is_monotonic_decreasing
|
1720 | 1721 |
|
| 1722 | + |
1721 | 1723 | def test_index_equals_string_vs_object():
|
1722 | 1724 | # GH 61099
|
1723 |
| - idx1 = pd.Index(['a', 'b', 'c']) |
1724 |
| - idx2 = pd.Index(['a', 'b', 'c'], dtype='string') |
| 1725 | + idx1 = Index(["a", "b", "c"]) |
| 1726 | + idx2 = Index(["a", "b", "c"], dtype="string") |
1725 | 1727 | assert idx1.equals(idx2)
|
1726 | 1728 | assert idx2.equals(idx1)
|
1727 | 1729 |
|
| 1730 | + |
1728 | 1731 | def test_compare_string_vs_object_index_equality():
|
1729 | 1732 | # GH 61099
|
1730 |
| - s1 = pd.Series([1, 2, 3], index=['a', 'b', 'c']) # dtype=object |
1731 |
| - s2 = pd.Series([0, 1, 2], index=pd.Index(['a', 'b', 'c'], dtype='string')) # dtype=string |
| 1733 | + s1 = Series([1, 2, 3], index=["a", "b", "c"]) # dtype=object |
| 1734 | + s2 = Series([0, 1, 2], index=Index(["a", "b", "c"], dtype="string")) # dtype=string |
1732 | 1735 | result = s1 > s2
|
1733 |
| - expected = pd.Series([True, True, True], index=['a', 'b', 'c']) |
1734 |
| - pd.testing.assert_series_equal(result, expected) |
| 1736 | + expected = Series([True, True, True], index=["a", "b", "c"]) |
| 1737 | + assert_series_equal(result, expected) |
| 1738 | + |
1735 | 1739 |
|
1736 | 1740 | def test_align_string_vs_object_index():
|
1737 | 1741 | # GH 61099
|
1738 |
| - s1 = pd.Series([1, 2, 3], index=['a', 'b', 'c']) # object |
1739 |
| - s2 = pd.Series([1, 2, 3], index=pd.Index(['a', 'b', 'c'], dtype='string')) # string |
| 1742 | + s1 = Series([1, 2, 3], index=["a", "b", "c"]) # object |
| 1743 | + s2 = Series([1, 2, 3], index=Index(["a", "b", "c"], dtype="string")) # string |
1740 | 1744 | s1_aligned, s2_aligned = s1.align(s2)
|
1741 | 1745 | assert list(s1_aligned.index) == list(s2_aligned.index)
|
1742 | 1746 |
|
| 1747 | + |
1743 | 1748 | def test_comparison_without_manual_casting():
|
1744 | 1749 | # GH 61099
|
1745 |
| - s1 = pd.Series([1, 2, 3], index=['a', 'b', 'c']) # object index |
1746 |
| - s2 = pd.Series([1, 2, 3], index=pd.Index(['a', 'b', 'c'], dtype='string')) |
| 1750 | + s1 = Series([1, 2, 3], index=["a", "b", "c"]) # object index |
| 1751 | + s2 = Series([1, 2, 3], index=Index(["a", "b", "c"], dtype="string")) |
1747 | 1752 | # Should not raise
|
1748 | 1753 | result = s1 == s2
|
1749 |
| - expected = pd.Series([True, True, True], index=['a', 'b', 'c']) |
1750 |
| - pd.testing.assert_series_equal(result, expected) |
| 1754 | + expected = Series([True, True, True], index=["a", "b", "c"]) |
| 1755 | + assert_series_equal(result, expected) |
0 commit comments