@@ -860,7 +860,7 @@ def test_mixed_groupings(normalize, expected_label, expected_values):
860
860
result = gp .value_counts (sort = True , normalize = normalize )
861
861
expected = DataFrame (
862
862
{
863
- "level_0" : [4 , 4 , 5 ],
863
+ "level_0" : np . array ( [4 , 4 , 5 ], dtype = np . int_ ) ,
864
864
"A" : [1 , 1 , 2 ],
865
865
"level_2" : [8 , 8 , 7 ],
866
866
"B" : [1 , 3 , 2 ],
@@ -883,7 +883,8 @@ def test_column_label_duplicates(test, columns, expected_names, as_index):
883
883
# Test for duplicate input column labels and generated duplicate labels
884
884
df = DataFrame ([[1 , 3 , 5 , 7 , 9 ], [2 , 4 , 6 , 8 , 10 ]], columns = columns )
885
885
expected_data = [(1 , 0 , 7 , 3 , 5 , 9 ), (2 , 1 , 8 , 4 , 6 , 10 )]
886
- result = df .groupby (["a" , [0 , 1 ], "d" ], as_index = as_index ).value_counts ()
886
+ keys = ["a" , np .array ([0 , 1 ], dtype = np .int64 ), "d" ]
887
+ result = df .groupby (keys , as_index = as_index ).value_counts ()
887
888
if as_index :
888
889
expected = Series (
889
890
data = (1 , 1 ),
@@ -922,7 +923,7 @@ def test_result_label_duplicates(normalize, expected_label):
922
923
def test_ambiguous_grouping ():
923
924
# Test that groupby is not confused by groupings length equal to row count
924
925
df = DataFrame ({"a" : [1 , 1 ]})
925
- gb = df .groupby ([1 , 1 ])
926
+ gb = df .groupby (np . array ( [1 , 1 ], dtype = np . int64 ) )
926
927
result = gb .value_counts ()
927
928
expected = Series ([2 ], index = MultiIndex .from_tuples ([[1 , 1 ]], names = [None , "a" ]))
928
929
tm .assert_series_equal (result , expected )
0 commit comments