Skip to content

Commit 6655dfe

Browse files
committed
fix groupby value_counts
1 parent eb0ecbe commit 6655dfe

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pandas/tests/groupby/test_value_counts.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -860,7 +860,7 @@ def test_mixed_groupings(normalize, expected_label, expected_values):
860860
result = gp.value_counts(sort=True, normalize=normalize)
861861
expected = DataFrame(
862862
{
863-
"level_0": [4, 4, 5],
863+
"level_0": np.array([4, 4, 5], dtype=np.int_),
864864
"A": [1, 1, 2],
865865
"level_2": [8, 8, 7],
866866
"B": [1, 3, 2],
@@ -883,7 +883,8 @@ def test_column_label_duplicates(test, columns, expected_names, as_index):
883883
# Test for duplicate input column labels and generated duplicate labels
884884
df = DataFrame([[1, 3, 5, 7, 9], [2, 4, 6, 8, 10]], columns=columns)
885885
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()
887888
if as_index:
888889
expected = Series(
889890
data=(1, 1),
@@ -922,7 +923,7 @@ def test_result_label_duplicates(normalize, expected_label):
922923
def test_ambiguous_grouping():
923924
# Test that groupby is not confused by groupings length equal to row count
924925
df = DataFrame({"a": [1, 1]})
925-
gb = df.groupby([1, 1])
926+
gb = df.groupby(np.array([1, 1], dtype=np.int64))
926927
result = gb.value_counts()
927928
expected = Series([2], index=MultiIndex.from_tuples([[1, 1]], names=[None, "a"]))
928929
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)