4
4
from pandas .errors import PerformanceWarning
5
5
6
6
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
9
8
10
9
11
10
@pytest .fixture
@@ -62,7 +61,7 @@ def test_sort_index_level_and_column_label(df_none, df_idx, sort_names, ascendin
62
61
# Compute result sorting on mix on columns and index levels
63
62
result = df_idx .sort_values (by = sort_names , ascending = ascending , axis = 0 )
64
63
65
- assert_frame_equal (result , expected )
64
+ tm . assert_frame_equal (result , expected )
66
65
67
66
68
67
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
88
87
# Accessing multi-level columns that are not lexsorted raises a
89
88
# performance warning
90
89
with tm .assert_produces_warning (PerformanceWarning , check_stacklevel = False ):
91
- assert_frame_equal (result , expected )
90
+ tm . assert_frame_equal (result , expected )
92
91
else :
93
- assert_frame_equal (result , expected )
92
+ tm . assert_frame_equal (result , expected )
0 commit comments