Skip to content

Commit 8040915

Browse files
Add test
Signed-off-by: Devin Petersohn <[email protected]>
1 parent 13395f0 commit 8040915

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

modin/pandas/test/test_groupby.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2003,3 +2003,25 @@ def test_sum_with_level():
20032003
}
20042004
modin_df, pandas_df = pd.DataFrame(data), pandas.DataFrame(data)
20052005
eval_general(modin_df, pandas_df, lambda df: df.set_index("C").groupby("C").sum())
2006+
2007+
2008+
def test_reset_index_groupby():
2009+
frame_data = np.random.randint(97, 198, size=(2**6, 2**4))
2010+
pandas_df = pandas.DataFrame(
2011+
frame_data,
2012+
index=pandas.MultiIndex.from_tuples(
2013+
[(i // 4, i // 2, i) for i in range(2**6)]
2014+
),
2015+
).add_prefix("col")
2016+
pandas_df.index.names = [f"index_{i}" for i in range(len(pandas_df.index.names))]
2017+
# Convert every other column to string
2018+
for col in pandas_df.iloc[
2019+
:, [i for i in range(len(pandas_df.columns)) if i % 2 == 0]
2020+
]:
2021+
pandas_df[col] = [str(chr(i)) for i in pandas_df[col]]
2022+
modin_df = from_pandas(pandas_df)
2023+
eval_general(
2024+
modin_df,
2025+
pandas_df,
2026+
lambda df: df.reset_index().groupby(["index_0", "index_1"]).count(),
2027+
)

0 commit comments

Comments
 (0)