Skip to content

Commit 66c3c6f

Browse files
authored
TST: tighten assert_index_equal calls (#38054)
1 parent f15c4bc commit 66c3c6f

File tree

7 files changed

+17
-14
lines changed

7 files changed

+17
-14
lines changed

pandas/tests/frame/methods/test_reset_index.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def test_reset_index(self, float_frame):
130130
float_frame.index.name = "index"
131131
deleveled = float_frame.reset_index()
132132
tm.assert_series_equal(deleveled["index"], Series(float_frame.index))
133-
tm.assert_index_equal(deleveled.index, Index(np.arange(len(deleveled))))
133+
tm.assert_index_equal(deleveled.index, Index(range(len(deleveled))), exact=True)
134134

135135
# preserve column names
136136
float_frame.columns.name = "columns"

pandas/tests/frame/test_constructors.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,7 @@ def test_constructor_dict(self):
352352

353353
# with dict of empty list and Series
354354
frame = DataFrame({"A": [], "B": []}, columns=["A", "B"])
355-
tm.assert_index_equal(frame.index, Index([], dtype=np.int64))
355+
tm.assert_index_equal(frame.index, RangeIndex(0), exact=True)
356356

357357
# GH 14381
358358
# Dict with None value
@@ -802,14 +802,14 @@ def _check_basic_constructor(self, empty):
802802

803803
# automatic labeling
804804
frame = DataFrame(mat)
805-
tm.assert_index_equal(frame.index, pd.Int64Index(range(2)))
806-
tm.assert_index_equal(frame.columns, pd.Int64Index(range(3)))
805+
tm.assert_index_equal(frame.index, Index(range(2)), exact=True)
806+
tm.assert_index_equal(frame.columns, Index(range(3)), exact=True)
807807

808808
frame = DataFrame(mat, index=[1, 2])
809-
tm.assert_index_equal(frame.columns, pd.Int64Index(range(3)))
809+
tm.assert_index_equal(frame.columns, Index(range(3)), exact=True)
810810

811811
frame = DataFrame(mat, columns=["A", "B", "C"])
812-
tm.assert_index_equal(frame.index, pd.Int64Index(range(2)))
812+
tm.assert_index_equal(frame.index, Index(range(2)), exact=True)
813813

814814
# 0-length axis
815815
frame = DataFrame(empty((0, 3)))

pandas/tests/indexes/common.py

+3
Original file line numberDiff line numberDiff line change
@@ -529,6 +529,7 @@ def test_equals_op(self):
529529
# assuming the 2nd to last item is unique in the data
530530
item = index_a[-2]
531531
tm.assert_numpy_array_equal(index_a == item, expected3)
532+
# For RangeIndex we can convert to Int64Index
532533
tm.assert_series_equal(series_a == item, Series(expected3))
533534

534535
def test_format(self):
@@ -656,6 +657,7 @@ def test_map(self):
656657
expected = index
657658

658659
result = index.map(lambda x: x)
660+
# For RangeIndex we convert to Int64Index
659661
tm.assert_index_equal(result, expected)
660662

661663
@pytest.mark.parametrize(
@@ -680,6 +682,7 @@ def test_map_dictlike(self, mapper):
680682
expected = index
681683

682684
result = index.map(identity)
685+
# For RangeIndex we convert to Int64Index
683686
tm.assert_index_equal(result, expected)
684687

685688
# empty mappable

pandas/tests/indexes/test_setops.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -385,8 +385,8 @@ def test_difference_preserves_type_empty(self, index, sort):
385385
if not index.is_unique:
386386
return
387387
result = index.difference(index, sort=sort)
388-
expected = index.drop(index)
389-
tm.assert_index_equal(result, expected)
388+
expected = index[:0]
389+
tm.assert_index_equal(result, expected, exact=True)
390390

391391
def test_intersection_difference_match_empty(self, index, sort):
392392
# GH#20040
@@ -395,6 +395,6 @@ def test_intersection_difference_match_empty(self, index, sort):
395395
# of an index with itself. Test for all types
396396
if not index.is_unique:
397397
return
398-
inter = index.intersection(index.drop(index))
398+
inter = index.intersection(index[:0])
399399
diff = index.difference(index, sort=sort)
400-
tm.assert_index_equal(inter, diff)
400+
tm.assert_index_equal(inter, diff, exact=True)

pandas/tests/reshape/concat/test_series.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,9 @@ def test_concat_series_axis1_same_names_ignore_index(self):
106106
s2 = Series(np.random.randn(len(dates)), index=dates, name="value")
107107

108108
result = concat([s1, s2], axis=1, ignore_index=True)
109-
expected = Index([0, 1])
109+
expected = Index(range(2))
110110

111-
tm.assert_index_equal(result.columns, expected)
111+
tm.assert_index_equal(result.columns, expected, exact=True)
112112

113113
@pytest.mark.parametrize(
114114
"s1name,s2name", [(np.int64(190), (43, 0)), (190, (43, 0))]

pandas/tests/reshape/merge/test_join.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,7 @@ def test_join_sort(self):
504504

505505
# smoke test
506506
joined = left.join(right, on="key", sort=False)
507-
tm.assert_index_equal(joined.index, Index(list(range(4))))
507+
tm.assert_index_equal(joined.index, Index(range(4)), exact=True)
508508

509509
def test_join_mixed_non_unique_index(self):
510510
# GH 12814, unorderable types in py3 with a non-unique index

pandas/tests/series/test_constructors.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -556,7 +556,7 @@ def test_series_ctor_plus_datetimeindex(self):
556556

557557
def test_constructor_default_index(self):
558558
s = Series([0, 1, 2])
559-
tm.assert_index_equal(s.index, Index(np.arange(3)))
559+
tm.assert_index_equal(s.index, Index(range(3)), exact=True)
560560

561561
@pytest.mark.parametrize(
562562
"input",

0 commit comments

Comments
 (0)