Skip to content

added test to indexing on groupby, #32464 #44046

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 16 commits into from
Closed

added test to indexing on groupby, #32464 #44046

Changes from 6 commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
239d42a
added test to indexing on groupby, #32464
gabrieldi95 Oct 16, 2021
b42f6e5
added comment
gabrieldi95 Oct 16, 2021
b2df552
Merge remote-tracking branch 'upstream/master' into tst_issue_32464
gabrieldi95 Oct 16, 2021
57f86c3
changed file and asserting method
gabrieldi95 Oct 17, 2021
c89f4ad
removed test to see if df has multiindex
gabrieldi95 Oct 17, 2021
84e8330
Merge remote-tracking branch 'upstream/master' into tst_issue_32464
gabrieldi95 Oct 17, 2021
4bf0563
changed assert method
gabrieldi95 Oct 17, 2021
be119e6
Merge remote-tracking branch 'upstream/master' into tst_issue_32464
gabrieldi95 Oct 17, 2021
8bd532f
Merge remote-tracking branch 'upstream/master' into tst_issue_32464
gabrieldi95 Nov 17, 2021
9ee9c8e
using multiindex constructor
gabrieldi95 Nov 17, 2021
c1ed828
Merge remote-tracking branch 'upstream/master' into tst_issue_32464
gabrieldi95 Nov 19, 2021
28fbe02
Merge remote-tracking branch 'upstream/master' into tst_issue_32464
gabrieldi95 Nov 28, 2021
0df3f33
Merge remote-tracking branch 'upstream/main' into tst_issue_32464
gabrieldi95 Feb 16, 2022
de65783
changed test name
gabrieldi95 Feb 22, 2022
e22b830
Merge remote-tracking branch 'upstream/main' into tst_issue_32464
gabrieldi95 Feb 22, 2022
cd2d3fd
Merge remote-tracking branch 'upstream/main' into tst_issue_32464
gabrieldi95 Mar 16, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions pandas/tests/groupby/test_function.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1155,3 +1155,15 @@ def test_groupby_sum_below_mincount_nullable_integer():
result = grouped.sum(min_count=2)
expected = DataFrame({"b": [pd.NA] * 3, "c": [pd.NA] * 3}, dtype="Int64", index=idx)
tm.assert_frame_equal(result, expected)


def test_if_is_multiindex():
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you rename to something like test_empty_multiindex

# GH 32464
# Test if index after groupby with more then one column is always MultiIndex
a = DataFrame({"a": [], "b": [], "c": []})

agg_1 = a.groupby(["a", "b"]).sum().iloc[:, :0]
agg_2 = a.groupby(["a", "b", "c"]).sum().droplevel("c")
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Okay could you change the assertion back to testing the MultiIndex result if the new assertion can't test the whole frame?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Sure! Like this?

# GH 32464
# Test if index after groupby with more then one column is always MultiIndex
a = DataFrame({"a": [], "b": [], "c": []})

agg_1 = a.groupby(["a", "b"]).sum()
agg_2 = a.groupby(["a", "b", "c"]).sum()

# Tests if group by with all columns has a MultiIndex
assert isinstance(agg_2.index, pd.core.indexes.multi.MultiIndex)

result = agg_1.iloc[:, :0]
expected = agg_2.droplevel("c")

# Tests if both agreggations have multiindex
tm.assert_frame_equal(result, expected)

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You can use tm.assert_index_equal


# Tests if both agreggations have multiindex
tm.assert_frame_equal(agg_1, agg_2)