@@ -644,21 +644,21 @@ def test_mode(self, dtype, writable):
644
644
values = np .repeat (np .arange (N ).astype (dtype ), 5 )
645
645
values [0 ] = 42
646
646
values .flags .writeable = writable
647
- result = ht .mode (values , False )
647
+ result = ht .mode (values , False )[ 0 ]
648
648
assert result == 42
649
649
650
650
def test_mode_stable (self , dtype , writable ):
651
651
values = np .array ([2 , 1 , 5 , 22 , 3 , - 1 , 8 ]).astype (dtype )
652
652
values .flags .writeable = writable
653
- keys = ht .mode (values , False )
653
+ keys = ht .mode (values , False )[ 0 ]
654
654
tm .assert_numpy_array_equal (keys , values )
655
655
656
656
657
657
def test_modes_with_nans ():
658
658
# GH42688, nans aren't mangled
659
659
nulls = [pd .NA , np .nan , pd .NaT , None ]
660
660
values = np .array ([True ] + nulls * 2 , dtype = np .object_ )
661
- modes = ht .mode (values , False )
661
+ modes = ht .mode (values , False )[ 0 ]
662
662
assert modes .size == len (nulls )
663
663
664
664
@@ -724,8 +724,8 @@ def test_ismember_no(self, dtype):
724
724
725
725
def test_mode (self , dtype ):
726
726
values = np .array ([42 , np .nan , np .nan , np .nan ], dtype = dtype )
727
- assert ht .mode (values , True ) == 42
728
- assert np .isnan (ht .mode (values , False ))
727
+ assert ht .mode (values , True )[ 0 ] == 42
728
+ assert np .isnan (ht .mode (values , False )[ 0 ] )
729
729
730
730
731
731
def test_ismember_tuple_with_nans ():
0 commit comments