@@ -809,18 +809,23 @@ def test_mode(self):
809
809
"E" : [8 , 8 , 1 , 1 , 3 , 3 ]})
810
810
tm .assert_frame_equal (df [["A" ]].mode (),
811
811
pd .DataFrame ({"A" : [12 ]}))
812
- expected = pd .Series ([0 , 1 , 2 , 3 , 4 , 5 ], dtype = 'int64' , name = 'D' ).to_frame ()
812
+ expected = pd .Series ([0 , 1 , 2 , 3 , 4 , 5 ], dtype = 'int64' , name = 'D' ).\
813
+ to_frame ()
813
814
tm .assert_frame_equal (df [["D" ]].mode (), expected )
814
815
expected = pd .Series ([1 , 3 , 8 ], dtype = 'int64' , name = 'E' ).to_frame ()
815
816
tm .assert_frame_equal (df [["E" ]].mode (), expected )
816
817
tm .assert_frame_equal (df [["A" , "B" ]].mode (),
817
818
pd .DataFrame ({"A" : [12 ], "B" : [10. ]}))
818
819
tm .assert_frame_equal (df .mode (),
819
- pd .DataFrame ({"A" : [12 , np .nan , np .nan , np .nan , np .nan , np .nan ],
820
- "B" : [10 , np .nan , np .nan , np .nan , np .nan , np .nan ],
821
- "C" : [8 , 9 , np .nan , np .nan , np .nan , np .nan ],
820
+ pd .DataFrame ({"A" : [12 , np .nan , np .nan , np .nan ,
821
+ np .nan , np .nan ],
822
+ "B" : [10 , np .nan , np .nan , np .nan ,
823
+ np .nan , np .nan ],
824
+ "C" : [8 , 9 , np .nan , np .nan , np .nan ,
825
+ np .nan ],
822
826
"D" : [0 , 1 , 2 , 3 , 4 , 5 ],
823
- "E" : [1 , 3 , 8 , np .nan , np .nan , np .nan ]}))
827
+ "E" : [1 , 3 , 8 , np .nan , np .nan ,
828
+ np .nan ]}))
824
829
825
830
# outputs in sorted order
826
831
df ["C" ] = list (reversed (df ["C" ]))
@@ -837,9 +842,12 @@ def test_mode(self):
837
842
df = pd .DataFrame ({"A" : np .arange (6 , dtype = 'int64' ),
838
843
"B" : pd .date_range ('2011' , periods = 6 ),
839
844
"C" : list ('abcdef' )})
840
- exp = pd .DataFrame ({"A" : pd .Series (np .arange (6 , dtype = 'int64' ), dtype = df ["A" ].dtype ),
841
- "B" : pd .Series (pd .date_range ('2011' , periods = 6 ), dtype = df ["B" ].dtype ),
842
- "C" : pd .Series (list ('abcdef' ), dtype = df ["C" ].dtype )})
845
+ exp = pd .DataFrame ({"A" : pd .Series (np .arange (6 , dtype = 'int64' ),
846
+ dtype = df ["A" ].dtype ),
847
+ "B" : pd .Series (pd .date_range ('2011' , periods = 6 ),
848
+ dtype = df ["B" ].dtype ),
849
+ "C" : pd .Series (list ('abcdef' ),
850
+ dtype = df ["C" ].dtype )})
843
851
tm .assert_frame_equal (df .mode (), exp )
844
852
845
853
def test_operators_timedelta64 (self ):
0 commit comments