|
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