Skip to content

Commit 8b55ddf

Browse files
gfyoungpcluo
authored andcommitted
MAINT: Remove vestigial self.assert* (pandas-dev#16190)
Remove all remaining self.assert* method calls originating from unittest. Any that are left are calls to methods directly defined in the test class or a higher derived pandas test class.
1 parent 4fdb2cf commit 8b55ddf

File tree

4 files changed

+71
-85
lines changed

4 files changed

+71
-85
lines changed

pandas/tests/frame/test_repr_info.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -331,13 +331,13 @@ def test_info_memory_usage(self):
331331
res = buf.getvalue().splitlines()
332332
assert re.match(r"memory usage: [^+]+$", res[-1])
333333

334-
self.assertGreater(df_with_object_index.memory_usage(index=True,
335-
deep=True).sum(),
336-
df_with_object_index.memory_usage(index=True).sum())
334+
assert (df_with_object_index.memory_usage(
335+
index=True, deep=True).sum() > df_with_object_index.memory_usage(
336+
index=True).sum())
337337

338338
df_object = pd.DataFrame({'a': ['a']})
339-
self.assertGreater(df_object.memory_usage(deep=True).sum(),
340-
df_object.memory_usage().sum())
339+
assert (df_object.memory_usage(deep=True).sum() >
340+
df_object.memory_usage().sum())
341341

342342
# Test a DataFrame with duplicate columns
343343
dtypes = ['int64', 'int64', 'int64', 'float64']

pandas/tests/io/formats/test_style.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -374,7 +374,7 @@ def test_bar_align_mid_pos_and_neg(self):
374374
'#5fba7d 10.0%, #5fba7d 100.0%, '
375375
'transparent 100.0%)']}
376376

377-
self.assertEqual(result, expected)
377+
assert result == expected
378378

379379
def test_bar_align_mid_all_pos(self):
380380
df = pd.DataFrame({'A': [10, 20, 50, 100]})
@@ -399,7 +399,7 @@ def test_bar_align_mid_all_pos(self):
399399
'transparent 0%, transparent 0.0%, #5fba7d 0.0%, '
400400
'#5fba7d 100.0%, transparent 100.0%)']}
401401

402-
self.assertEqual(result, expected)
402+
assert result == expected
403403

404404
def test_bar_align_mid_all_neg(self):
405405
df = pd.DataFrame({'A': [-100, -60, -30, -20]})

pandas/tests/test_nanops.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -914,11 +914,11 @@ def test_constant_series(self):
914914
def test_all_finite(self):
915915
alpha, beta = 0.3, 0.1
916916
left_tailed = self.prng.beta(alpha, beta, size=100)
917-
self.assertLess(nanops.nanskew(left_tailed), 0)
917+
assert nanops.nanskew(left_tailed) < 0
918918

919919
alpha, beta = 0.1, 0.3
920920
right_tailed = self.prng.beta(alpha, beta, size=100)
921-
self.assertGreater(nanops.nanskew(right_tailed), 0)
921+
assert nanops.nanskew(right_tailed) > 0
922922

923923
def test_ground_truth(self):
924924
skew = nanops.nanskew(self.samples)
@@ -964,11 +964,11 @@ def test_constant_series(self):
964964
def test_all_finite(self):
965965
alpha, beta = 0.3, 0.1
966966
left_tailed = self.prng.beta(alpha, beta, size=100)
967-
self.assertLess(nanops.nankurt(left_tailed), 0)
967+
assert nanops.nankurt(left_tailed) < 0
968968

969969
alpha, beta = 0.1, 0.3
970970
right_tailed = self.prng.beta(alpha, beta, size=100)
971-
self.assertGreater(nanops.nankurt(right_tailed), 0)
971+
assert nanops.nankurt(right_tailed) > 0
972972

973973
def test_ground_truth(self):
974974
kurt = nanops.nankurt(self.samples)

0 commit comments

Comments
 (0)