@@ -333,64 +333,73 @@ def test_drop_duplicates_inplace():
333
333
)
334
334
# single column
335
335
df = orig .copy ()
336
- df .drop_duplicates ("A" , inplace = True )
336
+ return_value = df .drop_duplicates ("A" , inplace = True )
337
337
expected = orig [:2 ]
338
338
result = df
339
339
tm .assert_frame_equal (result , expected )
340
+ assert return_value is None
340
341
341
342
df = orig .copy ()
342
- df .drop_duplicates ("A" , keep = "last" , inplace = True )
343
+ return_value = df .drop_duplicates ("A" , keep = "last" , inplace = True )
343
344
expected = orig .loc [[6 , 7 ]]
344
345
result = df
345
346
tm .assert_frame_equal (result , expected )
347
+ assert return_value is None
346
348
347
349
df = orig .copy ()
348
- df .drop_duplicates ("A" , keep = False , inplace = True )
350
+ return_value = df .drop_duplicates ("A" , keep = False , inplace = True )
349
351
expected = orig .loc [[]]
350
352
result = df
351
353
tm .assert_frame_equal (result , expected )
352
354
assert len (df ) == 0
355
+ assert return_value is None
353
356
354
357
# multi column
355
358
df = orig .copy ()
356
- df .drop_duplicates (["A" , "B" ], inplace = True )
359
+ return_value = df .drop_duplicates (["A" , "B" ], inplace = True )
357
360
expected = orig .loc [[0 , 1 , 2 , 3 ]]
358
361
result = df
359
362
tm .assert_frame_equal (result , expected )
363
+ assert return_value is None
360
364
361
365
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 )
363
367
expected = orig .loc [[0 , 5 , 6 , 7 ]]
364
368
result = df
365
369
tm .assert_frame_equal (result , expected )
370
+ assert return_value is None
366
371
367
372
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 )
369
374
expected = orig .loc [[0 ]]
370
375
result = df
371
376
tm .assert_frame_equal (result , expected )
377
+ assert return_value is None
372
378
373
379
# consider everything
374
380
orig2 = orig .loc [:, ["A" , "B" , "C" ]].copy ()
375
381
376
382
df2 = orig2 .copy ()
377
- df2 .drop_duplicates (inplace = True )
383
+ return_value = df2 .drop_duplicates (inplace = True )
378
384
# in this case only
379
385
expected = orig2 .drop_duplicates (["A" , "B" ])
380
386
result = df2
381
387
tm .assert_frame_equal (result , expected )
388
+ assert return_value is None
382
389
383
390
df2 = orig2 .copy ()
384
- df2 .drop_duplicates (keep = "last" , inplace = True )
391
+ return_value = df2 .drop_duplicates (keep = "last" , inplace = True )
385
392
expected = orig2 .drop_duplicates (["A" , "B" ], keep = "last" )
386
393
result = df2
387
394
tm .assert_frame_equal (result , expected )
395
+ assert return_value is None
388
396
389
397
df2 = orig2 .copy ()
390
- df2 .drop_duplicates (keep = False , inplace = True )
398
+ return_value = df2 .drop_duplicates (keep = False , inplace = True )
391
399
expected = orig2 .drop_duplicates (["A" , "B" ], keep = False )
392
400
result = df2
393
401
tm .assert_frame_equal (result , expected )
402
+ assert return_value is None
394
403
395
404
396
405
@pytest .mark .parametrize ("inplace" , [True , False ])
0 commit comments