@@ -381,21 +381,26 @@ def _check_output(result, values_col, index=['A', 'B'],
381
381
df = df .set_index (['JOB' , 'NAME' , 'YEAR' , 'MONTH' ], drop = False ,
382
382
append = False )
383
383
384
- result = df .pivot_table (index = ['JOB' , 'NAME' ],
385
- columns = ['YEAR' , 'MONTH' ],
386
- values = ['DAYS' , 'SALARY' ],
387
- aggfunc = {'DAYS' : 'mean' , 'SALARY' : 'sum' },
388
- margins = True )
389
-
390
- expected = df .pivot_table (index = ['JOB' , 'NAME' ],
391
- columns = ['YEAR' , 'MONTH' ], values = ['DAYS' ],
392
- aggfunc = 'mean' , margins = True )
384
+ with tm .assert_produces_warning (FutureWarning , check_stacklevel = False ):
385
+ result = df .pivot_table (index = ['JOB' , 'NAME' ],
386
+ columns = ['YEAR' , 'MONTH' ],
387
+ values = ['DAYS' , 'SALARY' ],
388
+ aggfunc = {'DAYS' : 'mean' , 'SALARY' : 'sum' },
389
+ margins = True )
390
+
391
+ with tm .assert_produces_warning (FutureWarning , check_stacklevel = False ):
392
+ expected = df .pivot_table (index = ['JOB' , 'NAME' ],
393
+ columns = ['YEAR' , 'MONTH' ],
394
+ values = ['DAYS' ],
395
+ aggfunc = 'mean' , margins = True )
393
396
394
397
tm .assert_frame_equal (result ['DAYS' ], expected ['DAYS' ])
395
398
396
- expected = df .pivot_table (index = ['JOB' , 'NAME' ],
397
- columns = ['YEAR' , 'MONTH' ], values = ['SALARY' ],
398
- aggfunc = 'sum' , margins = True )
399
+ with tm .assert_produces_warning (FutureWarning , check_stacklevel = False ):
400
+ expected = df .pivot_table (index = ['JOB' , 'NAME' ],
401
+ columns = ['YEAR' , 'MONTH' ],
402
+ values = ['SALARY' ],
403
+ aggfunc = 'sum' , margins = True )
399
404
400
405
tm .assert_frame_equal (result ['SALARY' ], expected ['SALARY' ])
401
406
0 commit comments