@@ -82,22 +82,24 @@ def test_apply_empty(self):
82
82
assert_frame_equal (xp , rs )
83
83
84
84
# reduce with an empty DataFrame
85
- x = []
86
- result = self .empty .apply (x .append , axis = 1 , reduce = False )
87
- assert_frame_equal (result , self .empty )
88
- result = self .empty .apply (x .append , axis = 1 , reduce = True )
89
- assert_series_equal (result , Series (
90
- [], index = pd .Index ([], dtype = object )))
91
-
92
- empty_with_cols = DataFrame (columns = ['a' , 'b' , 'c' ])
93
- result = empty_with_cols .apply (x .append , axis = 1 , reduce = False )
94
- assert_frame_equal (result , empty_with_cols )
95
- result = empty_with_cols .apply (x .append , axis = 1 , reduce = True )
96
- assert_series_equal (result , Series (
97
- [], index = pd .Index ([], dtype = object )))
98
-
99
- # Ensure that x.append hasn't been called
100
- assert x == []
85
+ with tm .assert_produces_warning (FutureWarning ,
86
+ check_stacklevel = False ):
87
+ x = []
88
+ result = self .empty .apply (x .append , axis = 1 , reduce = False )
89
+ assert_frame_equal (result , self .empty )
90
+ result = self .empty .apply (x .append , axis = 1 , reduce = True )
91
+ assert_series_equal (result , Series (
92
+ [], index = pd .Index ([], dtype = object )))
93
+
94
+ empty_with_cols = DataFrame (columns = ['a' , 'b' , 'c' ])
95
+ result = empty_with_cols .apply (x .append , axis = 1 , reduce = False )
96
+ assert_frame_equal (result , empty_with_cols )
97
+ result = empty_with_cols .apply (x .append , axis = 1 , reduce = True )
98
+ assert_series_equal (result , Series (
99
+ [], index = pd .Index ([], dtype = object )))
100
+
101
+ # Ensure that x.append hasn't been called
102
+ assert x == []
101
103
102
104
def test_apply_standard_nonunique (self ):
103
105
df = DataFrame (
@@ -371,14 +373,16 @@ def test_apply_dict(self):
371
373
B_dicts = pd .Series ([dict ([(0 , 0 ), (1 , 2 )]), dict ([(0 , 1 ), (1 , 3 )])])
372
374
fn = lambda x : x .to_dict ()
373
375
374
- for df , dicts in [(A , A_dicts ), (B , B_dicts )]:
375
- reduce_true = df .apply (fn , reduce = True )
376
- reduce_false = df .apply (fn , reduce = False )
377
- reduce_none = df .apply (fn , reduce = None )
376
+ with tm .assert_produces_warning (FutureWarning ,
377
+ check_stacklevel = False ):
378
+ for df , dicts in [(A , A_dicts ), (B , B_dicts )]:
379
+ reduce_true = df .apply (fn , reduce = True )
380
+ reduce_false = df .apply (fn , reduce = False )
381
+ reduce_none = df .apply (fn , reduce = None )
378
382
379
- assert_series_equal (reduce_true , dicts )
380
- assert_frame_equal (reduce_false , df )
381
- assert_series_equal (reduce_none , dicts )
383
+ assert_series_equal (reduce_true , dicts )
384
+ assert_frame_equal (reduce_false , df )
385
+ assert_series_equal (reduce_none , dicts )
382
386
383
387
def test_applymap (self ):
384
388
applied = self .frame .applymap (lambda x : x * 2 )
0 commit comments