Skip to content

Commit e5b5034

Browse files
SaturnFromTitanproost
authored andcommitted
Fixed test util imports in pandas/tests/frame - part 4 (pandas-dev#29303)
1 parent 9b6b381 commit e5b5034

File tree

4 files changed

+278
-269
lines changed

4 files changed

+278
-269
lines changed

pandas/tests/frame/test_sort_values_level_as_str.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,7 @@
44
from pandas.errors import PerformanceWarning
55

66
from pandas import DataFrame
7-
from pandas.util import testing as tm
8-
from pandas.util.testing import assert_frame_equal
7+
import pandas.util.testing as tm
98

109

1110
@pytest.fixture
@@ -62,7 +61,7 @@ def test_sort_index_level_and_column_label(df_none, df_idx, sort_names, ascendin
6261
# Compute result sorting on mix on columns and index levels
6362
result = df_idx.sort_values(by=sort_names, ascending=ascending, axis=0)
6463

65-
assert_frame_equal(result, expected)
64+
tm.assert_frame_equal(result, expected)
6665

6766

6867
def test_sort_column_level_and_index_label(df_none, df_idx, sort_names, ascending):
@@ -88,6 +87,6 @@ def test_sort_column_level_and_index_label(df_none, df_idx, sort_names, ascendin
8887
# Accessing multi-level columns that are not lexsorted raises a
8988
# performance warning
9089
with tm.assert_produces_warning(PerformanceWarning, check_stacklevel=False):
91-
assert_frame_equal(result, expected)
90+
tm.assert_frame_equal(result, expected)
9291
else:
93-
assert_frame_equal(result, expected)
92+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)