@@ -5626,6 +5626,11 @@ def test_apply(self):
5626
5626
self .assertEqual (applied [d ], np .mean (self .frame .xs (d )))
5627
5627
self .assert_ (applied .index is self .frame .index ) # want this
5628
5628
5629
+ #invalid axis
5630
+ df = DataFrame ([[1 ,2 ,3 ], [4 ,5 ,6 ], [7 ,8 ,9 ]], index = ['a' ,'a' ,'c' ])
5631
+ self .assertRaises (ValueError , df .apply , lambda x : x , 2 )
5632
+
5633
+ def test_apply_empty (self ):
5629
5634
# empty
5630
5635
applied = self .empty .apply (np .sqrt )
5631
5636
self .assert_ (applied .empty )
@@ -5643,9 +5648,10 @@ def test_apply(self):
5643
5648
expected = Series (np .nan , index = self .frame .index )
5644
5649
assert_series_equal (result , expected )
5645
5650
5646
- #invalid axis
5647
- df = DataFrame ([[1 ,2 ,3 ], [4 ,5 ,6 ], [7 ,8 ,9 ]], index = ['a' ,'a' ,'c' ])
5648
- self .assertRaises (ValueError , df .apply , lambda x : x , 2 )
5651
+ #2476
5652
+ xp = DataFrame (index = ['a' ])
5653
+ rs = xp .apply (lambda x : x ['a' ], axis = 1 )
5654
+ assert_frame_equal (xp , rs )
5649
5655
5650
5656
def test_apply_standard_nonunique (self ):
5651
5657
df = DataFrame ([[1 ,2 ,3 ], [4 ,5 ,6 ], [7 ,8 ,9 ]], index = ['a' ,'a' ,'c' ])
0 commit comments