Skip to content

Commit c698913

Browse files
committed
REF: extract function _assert_series_values_equal
1 parent 0be8f99 commit c698913

File tree

1 file changed

+50
-18
lines changed

1 file changed

+50
-18
lines changed

pandas/_testing.py

Lines changed: 50 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1289,7 +1289,6 @@ def assert_series_equal(
12891289
rtol=1.0e-5,
12901290
atol=1.0e-8,
12911291
obj="Series",
1292-
check_index=True,
12931292
):
12941293
"""
12951294
Check that left and right Series are equal.
@@ -1363,6 +1362,55 @@ def assert_series_equal(
13631362
"""
13641363
__tracebackhide__ = True
13651364

1365+
_assert_series_values_equal(
1366+
left,
1367+
right,
1368+
check_dtype=check_dtype,
1369+
check_series_type=check_series_type,
1370+
check_less_precise=check_less_precise,
1371+
check_names=check_names,
1372+
check_exact=check_exact,
1373+
check_datetimelike_compat=check_datetimelike_compat,
1374+
check_categorical=check_categorical,
1375+
check_category_order=check_category_order,
1376+
check_freq=check_freq,
1377+
check_flags=check_flags,
1378+
rtol=rtol,
1379+
atol=atol,
1380+
obj=obj,
1381+
)
1382+
1383+
assert_index_equal(
1384+
left.index,
1385+
right.index,
1386+
exact=check_index_type,
1387+
check_names=check_names,
1388+
check_exact=check_exact,
1389+
check_categorical=check_categorical,
1390+
rtol=rtol,
1391+
atol=atol,
1392+
obj=f"{obj}.index",
1393+
)
1394+
1395+
1396+
def _assert_series_values_equal(
1397+
left,
1398+
right,
1399+
*,
1400+
check_dtype=True,
1401+
check_series_type=True,
1402+
check_less_precise=no_default,
1403+
check_names=True,
1404+
check_exact=False,
1405+
check_datetimelike_compat=False,
1406+
check_categorical=True,
1407+
check_category_order=True,
1408+
check_freq=True,
1409+
check_flags=True,
1410+
rtol=1.0e-5,
1411+
atol=1.0e-8,
1412+
obj="Series",
1413+
):
13661414
if check_less_precise is not no_default:
13671415
warnings.warn(
13681416
"The 'check_less_precise' keyword in testing.assert_*_equal "
@@ -1388,20 +1436,6 @@ def assert_series_equal(
13881436
if check_flags:
13891437
assert left.flags == right.flags, f"{repr(left.flags)} != {repr(right.flags)}"
13901438

1391-
if check_index:
1392-
# GH #38183
1393-
assert_index_equal(
1394-
left.index,
1395-
right.index,
1396-
exact=check_index_type,
1397-
check_names=check_names,
1398-
check_exact=check_exact,
1399-
check_categorical=check_categorical,
1400-
rtol=rtol,
1401-
atol=atol,
1402-
obj=f"{obj}.index",
1403-
)
1404-
14051439
if check_freq and isinstance(left.index, (pd.DatetimeIndex, pd.TimedeltaIndex)):
14061440
lidx = left.index
14071441
ridx = right.index
@@ -1697,11 +1731,10 @@ def assert_frame_equal(
16971731
assert col in right
16981732
lcol = left.iloc[:, i]
16991733
rcol = right.iloc[:, i]
1700-
assert_series_equal(
1734+
_assert_series_values_equal(
17011735
lcol,
17021736
rcol,
17031737
check_dtype=check_dtype,
1704-
check_index_type=check_index_type,
17051738
check_exact=check_exact,
17061739
check_names=check_names,
17071740
check_datetimelike_compat=check_datetimelike_compat,
@@ -1710,7 +1743,6 @@ def assert_frame_equal(
17101743
obj=f'{obj}.iloc[:, {i}] (column name="{col}")',
17111744
rtol=rtol,
17121745
atol=atol,
1713-
check_index=False,
17141746
)
17151747

17161748

0 commit comments

Comments
 (0)