Skip to content

Commit 27de044

Browse files
authored
TST add test case for drop_duplicates (#35121)
1 parent e9d90b4 commit 27de044

File tree

1 file changed

+18
-9
lines changed

1 file changed

+18
-9
lines changed

pandas/tests/frame/methods/test_drop_duplicates.py

+18-9
Original file line numberDiff line numberDiff line change
@@ -333,64 +333,73 @@ def test_drop_duplicates_inplace():
333333
)
334334
# single column
335335
df = orig.copy()
336-
df.drop_duplicates("A", inplace=True)
336+
return_value = df.drop_duplicates("A", inplace=True)
337337
expected = orig[:2]
338338
result = df
339339
tm.assert_frame_equal(result, expected)
340+
assert return_value is None
340341

341342
df = orig.copy()
342-
df.drop_duplicates("A", keep="last", inplace=True)
343+
return_value = df.drop_duplicates("A", keep="last", inplace=True)
343344
expected = orig.loc[[6, 7]]
344345
result = df
345346
tm.assert_frame_equal(result, expected)
347+
assert return_value is None
346348

347349
df = orig.copy()
348-
df.drop_duplicates("A", keep=False, inplace=True)
350+
return_value = df.drop_duplicates("A", keep=False, inplace=True)
349351
expected = orig.loc[[]]
350352
result = df
351353
tm.assert_frame_equal(result, expected)
352354
assert len(df) == 0
355+
assert return_value is None
353356

354357
# multi column
355358
df = orig.copy()
356-
df.drop_duplicates(["A", "B"], inplace=True)
359+
return_value = df.drop_duplicates(["A", "B"], inplace=True)
357360
expected = orig.loc[[0, 1, 2, 3]]
358361
result = df
359362
tm.assert_frame_equal(result, expected)
363+
assert return_value is None
360364

361365
df = orig.copy()
362-
df.drop_duplicates(["A", "B"], keep="last", inplace=True)
366+
return_value = df.drop_duplicates(["A", "B"], keep="last", inplace=True)
363367
expected = orig.loc[[0, 5, 6, 7]]
364368
result = df
365369
tm.assert_frame_equal(result, expected)
370+
assert return_value is None
366371

367372
df = orig.copy()
368-
df.drop_duplicates(["A", "B"], keep=False, inplace=True)
373+
return_value = df.drop_duplicates(["A", "B"], keep=False, inplace=True)
369374
expected = orig.loc[[0]]
370375
result = df
371376
tm.assert_frame_equal(result, expected)
377+
assert return_value is None
372378

373379
# consider everything
374380
orig2 = orig.loc[:, ["A", "B", "C"]].copy()
375381

376382
df2 = orig2.copy()
377-
df2.drop_duplicates(inplace=True)
383+
return_value = df2.drop_duplicates(inplace=True)
378384
# in this case only
379385
expected = orig2.drop_duplicates(["A", "B"])
380386
result = df2
381387
tm.assert_frame_equal(result, expected)
388+
assert return_value is None
382389

383390
df2 = orig2.copy()
384-
df2.drop_duplicates(keep="last", inplace=True)
391+
return_value = df2.drop_duplicates(keep="last", inplace=True)
385392
expected = orig2.drop_duplicates(["A", "B"], keep="last")
386393
result = df2
387394
tm.assert_frame_equal(result, expected)
395+
assert return_value is None
388396

389397
df2 = orig2.copy()
390-
df2.drop_duplicates(keep=False, inplace=True)
398+
return_value = df2.drop_duplicates(keep=False, inplace=True)
391399
expected = orig2.drop_duplicates(["A", "B"], keep=False)
392400
result = df2
393401
tm.assert_frame_equal(result, expected)
402+
assert return_value is None
394403

395404

396405
@pytest.mark.parametrize("inplace", [True, False])

0 commit comments

Comments
 (0)