Skip to content

Commit 6d0118e

Browse files
gfyoungjreback
authored andcommitted
MAINT: Remove self.assertNotEqual from testing (#16176)
1 parent 668f91a commit 6d0118e

18 files changed

+40
-42
lines changed

pandas/tests/frame/test_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,7 @@ def test_deepcopy(self):
247247
series = cp['A']
248248
series[:] = 10
249249
for idx, value in compat.iteritems(series):
250-
self.assertNotEqual(self.frame['A'][idx], value)
250+
assert self.frame['A'][idx] != value
251251

252252
# ---------------------------------------------------------------------
253253
# Transposing

pandas/tests/frame/test_indexing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1890,7 +1890,7 @@ def test_nested_exception(self):
18901890
try:
18911891
repr(df)
18921892
except Exception as e:
1893-
self.assertNotEqual(type(e), UnboundLocalError)
1893+
assert type(e) != UnboundLocalError
18941894

18951895
def test_reindex_methods(self):
18961896
df = pd.DataFrame({'x': list(range(5))})

pandas/tests/frame/test_sorting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -361,7 +361,7 @@ def test_sort_index_inplace(self):
361361
df.sort_index(inplace=True)
362362
expected = frame
363363
assert_frame_equal(df, expected)
364-
self.assertNotEqual(a_id, id(df['A']))
364+
assert a_id != id(df['A'])
365365

366366
df = unordered.copy()
367367
df.sort_index(ascending=False, inplace=True)

pandas/tests/groupby/test_groupby.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1032,7 +1032,7 @@ def test_frame_set_name_single(self):
10321032
assert result.index.name == 'A'
10331033

10341034
result = self.df.groupby('A', as_index=False).mean()
1035-
self.assertNotEqual(result.index.name, 'A')
1035+
assert result.index.name != 'A'
10361036

10371037
result = grouped.agg(np.mean)
10381038
assert result.index.name == 'A'

pandas/tests/indexes/test_base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def test_constructor(self):
8181
assert index.name == 'name'
8282
tm.assert_numpy_array_equal(arr, index.values)
8383
arr[0] = "SOMEBIGLONGSTRING"
84-
self.assertNotEqual(index[0], "SOMEBIGLONGSTRING")
84+
assert index[0] != "SOMEBIGLONGSTRING"
8585

8686
# what to do here?
8787
# arr = np.array(5.)

pandas/tests/indexes/test_numeric.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -653,7 +653,7 @@ def test_constructor(self):
653653

654654
# this should not change index
655655
arr[0] = val
656-
self.assertNotEqual(new_index[0], val)
656+
assert new_index[0] != val
657657

658658
# interpret list-like
659659
expected = Int64Index([5, 0])

pandas/tests/io/formats/test_format.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -918,7 +918,7 @@ def test_wide_repr(self):
918918
assert "10 rows x %d columns" % (max_cols - 1) in rep_str
919919
set_option('display.expand_frame_repr', True)
920920
wide_repr = repr(df)
921-
self.assertNotEqual(rep_str, wide_repr)
921+
assert rep_str != wide_repr
922922

923923
with option_context('display.width', 120):
924924
wider_repr = repr(df)
@@ -944,7 +944,7 @@ def test_wide_repr_named(self):
944944
rep_str = repr(df)
945945
set_option('display.expand_frame_repr', True)
946946
wide_repr = repr(df)
947-
self.assertNotEqual(rep_str, wide_repr)
947+
assert rep_str != wide_repr
948948

949949
with option_context('display.width', 150):
950950
wider_repr = repr(df)
@@ -966,7 +966,7 @@ def test_wide_repr_multiindex(self):
966966
rep_str = repr(df)
967967
set_option('display.expand_frame_repr', True)
968968
wide_repr = repr(df)
969-
self.assertNotEqual(rep_str, wide_repr)
969+
assert rep_str != wide_repr
970970

971971
with option_context('display.width', 150):
972972
wider_repr = repr(df)
@@ -990,7 +990,7 @@ def test_wide_repr_multiindex_cols(self):
990990
rep_str = repr(df)
991991
set_option('display.expand_frame_repr', True)
992992
wide_repr = repr(df)
993-
self.assertNotEqual(rep_str, wide_repr)
993+
assert rep_str != wide_repr
994994

995995
with option_context('display.width', 150):
996996
wider_repr = repr(df)
@@ -1006,7 +1006,7 @@ def test_wide_repr_unicode(self):
10061006
rep_str = repr(df)
10071007
set_option('display.expand_frame_repr', True)
10081008
wide_repr = repr(df)
1009-
self.assertNotEqual(rep_str, wide_repr)
1009+
assert rep_str != wide_repr
10101010

10111011
with option_context('display.width', 150):
10121012
wider_repr = repr(df)

pandas/tests/io/formats/test_style.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,9 @@ def test_deepcopy(self):
8585

8686
self.styler._update_ctx(self.attrs)
8787
self.styler.highlight_max()
88-
self.assertNotEqual(self.styler.ctx, s2.ctx)
88+
assert self.styler.ctx != s2.ctx
8989
assert s2._todo == []
90-
self.assertNotEqual(self.styler._todo, s2._todo)
90+
assert self.styler._todo != s2._todo
9191

9292
def test_clear(self):
9393
s = self.df.style.highlight_max()._compute()

pandas/tests/io/test_common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def test_expand_user(self):
3838
filename = '~/sometest'
3939
expanded_name = common._expand_user(filename)
4040

41-
self.assertNotEqual(expanded_name, filename)
41+
assert expanded_name != filename
4242
assert isabs(expanded_name)
4343
assert os.path.expanduser(filename) == expanded_name
4444

@@ -68,7 +68,7 @@ def test_stringify_path_localpath(self):
6868
def test_get_filepath_or_buffer_with_path(self):
6969
filename = '~/sometest'
7070
filepath_or_buffer, _, _ = common.get_filepath_or_buffer(filename)
71-
self.assertNotEqual(filepath_or_buffer, filename)
71+
assert filepath_or_buffer != filename
7272
assert isabs(filepath_or_buffer)
7373
assert os.path.expanduser(filename) == filepath_or_buffer
7474

pandas/tests/scalar/test_period.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -938,8 +938,8 @@ def test_equal_Raises_Value(self):
938938
self.january1 == self.day
939939

940940
def test_notEqual(self):
941-
self.assertNotEqual(self.january1, 1)
942-
self.assertNotEqual(self.january1, self.february)
941+
assert self.january1 != 1
942+
assert self.january1 != self.february
943943

944944
def test_greater(self):
945945
assert self.february > self.january1

0 commit comments

Comments
 (0)