Skip to content

Commit 9ca8d3c

Browse files
committed
TST: use assert_frame_equal directly
1 parent 219ad75 commit 9ca8d3c

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

pandas/tests/test_frame.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6238,7 +6238,7 @@ def test_equals_different_blocks(self):
62386238
self.assertTrue(df0._data.blocks[0].dtype !=
62396239
df1._data.blocks[0].dtype)
62406240
# do the real tests
6241-
self.assert_frame_equal(df0, df1)
6241+
assert_frame_equal(df0, df1)
62426242
self.assertTrue(df0.equals(df1))
62436243
self.assertTrue(df1.equals(df0))
62446244

@@ -10776,7 +10776,7 @@ def test_diff(self):
1077610776
df.insert(0, 'x', 1)
1077710777
result = df.diff(axis=1)
1077810778
expected = pd.DataFrame({'x':np.nan, 'y':pd.Series(1), 'z':pd.Series(1)}).astype('float64')
10779-
self.assert_frame_equal(result, expected)
10779+
assert_frame_equal(result, expected)
1078010780

1078110781

1078210782
def test_diff_timedelta(self):
@@ -11382,11 +11382,11 @@ def test_filter(self):
1138211382
df = DataFrame(0., index=[0, 1, 2], columns=['A1', 1, 'B', 2, 'C'])
1138311383
expected = DataFrame(0., index=[0, 1, 2], columns=[1, 2])
1138411384
filtered = df.filter(regex='^[0-9]+$')
11385-
self.assert_frame_equal(filtered, expected)
11385+
assert_frame_equal(filtered, expected)
1138611386

1138711387
expected = DataFrame(0., index=[0, 1, 2], columns=[0, '0', 1, '1'])
1138811388
filtered = expected.filter(regex='^[0-9]+$') # shouldn't remove anything
11389-
self.assert_frame_equal(filtered, expected)
11389+
assert_frame_equal(filtered, expected)
1139011390

1139111391
# pass in None
1139211392
with assertRaisesRegexp(TypeError, 'Must pass'):
@@ -13438,7 +13438,7 @@ def test_unstack_level_binding(self):
1343813438
index=expected_mi,
1343913439
columns=pd.Index(['a', 'b'], name='third'))
1344013440

13441-
self.assert_frame_equal(result, expected)
13441+
assert_frame_equal(result, expected)
1344213442

1344313443
def test_unstack_to_series(self):
1344413444
# check reversibility

0 commit comments

Comments
 (0)