@@ -880,7 +880,7 @@ def test_mixed_groupings(normalize, expected_label, expected_values):
880880 result = gp .value_counts (sort = True , normalize = normalize )
881881 expected = DataFrame (
882882 {
883- "level_0" : [4 , 4 , 5 ],
883+ "level_0" : np . array ( [4 , 4 , 5 ], dtype = np . int_ ) ,
884884 "A" : [1 , 1 , 2 ],
885885 "level_2" : [8 , 8 , 7 ],
886886 "B" : [1 , 3 , 2 ],
@@ -903,7 +903,8 @@ def test_column_label_duplicates(test, columns, expected_names, as_index):
903903 # Test for duplicate input column labels and generated duplicate labels
904904 df = DataFrame ([[1 , 3 , 5 , 7 , 9 ], [2 , 4 , 6 , 8 , 10 ]], columns = columns )
905905 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 ()
907908 if as_index :
908909 expected = Series (
909910 data = (1 , 1 ),
@@ -942,7 +943,7 @@ def test_result_label_duplicates(normalize, expected_label):
942943def test_ambiguous_grouping ():
943944 # Test that groupby is not confused by groupings length equal to row count
944945 df = DataFrame ({"a" : [1 , 1 ]})
945- gb = df .groupby ([1 , 1 ])
946+ gb = df .groupby (np . array ( [1 , 1 ], dtype = np . int64 ) )
946947 result = gb .value_counts ()
947948 expected = Series ([2 ], index = MultiIndex .from_tuples ([[1 , 1 ]], names = [None , "a" ]))
948949 tm .assert_series_equal (result , expected )
0 commit comments