Skip to content

Commit 4c337f6

Browse files
committed
fix groupby value_counts
1 parent ea22074 commit 4c337f6

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
@@ -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):
942943
def 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

Comments
 (0)