@@ -6568,14 +6568,14 @@ def test_group_shift_with_null_key(self):
6568
6568
# values in column `B`, and then group by [`A`, `B`]. This should
6569
6569
# force `-1` in `labels` array of `gr_.grouper.group_info` exactly
6570
6570
# at those places, where the group-by key is partilly missing.
6571
- df = pd .DataFrame ([(i % 12 , i % 3 if i % 3 else float ("nan" ), i )
6571
+ df = pd .DataFrame ([(i % 12 , i % 3 if i % 3 else float ("nan" ), i )
6572
6572
for i in range (n_rows )], dtype = float ,
6573
6573
columns = ["A" , "B" , "Z" ], index = None )
6574
6574
gr_ = df .groupby (["A" , "B" ])
6575
6575
6576
6576
# Generate teh expected dataframe
6577
- expected = pd .DataFrame ([(i % 12 , i % 3 if i % 3 else float ("nan" ),
6578
- i + 12 if i % 3 and i < n_rows - 12 \
6577
+ expected = pd .DataFrame ([(i % 12 , i % 3 if i % 3 else float ("nan" ),
6578
+ i + 12 if i % 3 and i < n_rows - 12
6579
6579
else float ("nan" ))
6580
6580
for i in range (n_rows )], dtype = float ,
6581
6581
columns = ["A" , "B" , "Z" ], index = None )
0 commit comments