@@ -880,7 +880,7 @@ def test_mixed_groupings(normalize, expected_label, expected_values):
880
880
result = gp .value_counts (sort = True , normalize = normalize )
881
881
expected = DataFrame (
882
882
{
883
- "level_0" : [4 , 4 , 5 ],
883
+ "level_0" : np . array ( [4 , 4 , 5 ], dtype = np . int_ ) ,
884
884
"A" : [1 , 1 , 2 ],
885
885
"level_2" : [8 , 8 , 7 ],
886
886
"B" : [1 , 3 , 2 ],
@@ -903,7 +903,8 @@ def test_column_label_duplicates(test, columns, expected_names, as_index):
903
903
# Test for duplicate input column labels and generated duplicate labels
904
904
df = DataFrame ([[1 , 3 , 5 , 7 , 9 ], [2 , 4 , 6 , 8 , 10 ]], columns = columns )
905
905
expected_data = [(1 , 0 , 7 , 3 , 5 , 9 ), (2 , 1 , 8 , 4 , 6 , 10 )]
906
- result = df .groupby (["a" , [0 , 1 ], "d" ], as_index = as_index ).value_counts ()
906
+ keys = ["a" , np .array ([0 , 1 ], dtype = np .int64 ), "d" ]
907
+ result = df .groupby (keys , as_index = as_index ).value_counts ()
907
908
if as_index :
908
909
expected = Series (
909
910
data = (1 , 1 ),
@@ -942,7 +943,7 @@ def test_result_label_duplicates(normalize, expected_label):
942
943
def test_ambiguous_grouping ():
943
944
# Test that groupby is not confused by groupings length equal to row count
944
945
df = DataFrame ({"a" : [1 , 1 ]})
945
- gb = df .groupby ([1 , 1 ])
946
+ gb = df .groupby (np . array ( [1 , 1 ], dtype = np . int64 ) )
946
947
result = gb .value_counts ()
947
948
expected = Series ([2 ], index = MultiIndex .from_tuples ([[1 , 1 ]], names = [None , "a" ]))
948
949
tm .assert_series_equal (result , expected )
0 commit comments