Skip to content

Commit 75c6089

Browse files
author
Rehan Durrani
committed
lint
Signed-off-by: Rehan Durrani <[email protected]>
1 parent a5bb81e commit 75c6089

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

modin/pandas/test/test_groupby.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2026,47 +2026,48 @@ def test_reset_index_groupby():
20262026
lambda df: df.reset_index().groupby(["index_0", "index_1"]).count(),
20272027
)
20282028

2029+
20292030
def test_by_in_index_and_columns():
20302031
pandas_df = pandas.DataFrame(
2031-
[[1, 2, 3]], index=pd.Index([0], name="a"), columns=['a', 'b', 'c']
2032+
[[1, 2, 3]], index=pd.Index([0], name="a"), columns=["a", "b", "c"]
20322033
)
20332034
modin_df = from_pandas(pandas_df)
20342035
eval_general(
20352036
modin_df,
20362037
pandas_df,
2037-
lambda df: df.groupby(by='a').count(),
2038+
lambda df: df.groupby(by="a").count(),
20382039
raising_exceptions=True,
20392040
check_exception_type=True,
20402041
)
20412042
eval_general(
20422043
modin_df,
20432044
pandas_df,
2044-
lambda df: df.groupby(by=['a', 'b']).count(),
2045+
lambda df: df.groupby(by=["a", "b"]).count(),
20452046
raising_exceptions=True,
20462047
check_exception_type=True,
20472048
)
20482049
pandas_df = pandas.DataFrame(
2049-
[[1, 2, 3]], index=pd.Index([(0, 1)], names=["a", 'b']), columns=['a', 'b', 'c']
2050+
[[1, 2, 3]], index=pd.Index([(0, 1)], names=["a", "b"]), columns=["a", "b", "c"]
20502051
)
20512052
modin_df = from_pandas(pandas_df)
20522053
eval_general(
20532054
modin_df,
20542055
pandas_df,
2055-
lambda df: df.groupby(by='a').count(),
2056+
lambda df: df.groupby(by="a").count(),
20562057
raising_exceptions=True,
20572058
check_exception_type=True,
20582059
)
20592060
eval_general(
20602061
modin_df,
20612062
pandas_df,
2062-
lambda df: df.groupby(by=['a', 'c']).count(),
2063+
lambda df: df.groupby(by=["a", "c"]).count(),
20632064
raising_exceptions=True,
20642065
check_exception_type=True,
20652066
)
20662067
eval_general(
20672068
modin_df,
20682069
pandas_df,
2069-
lambda df: df.groupby(by=['a', 'b']).count(),
2070+
lambda df: df.groupby(by=["a", "b"]).count(),
20702071
raising_exceptions=True,
20712072
check_exception_type=True,
20722073
)

0 commit comments

Comments
 (0)