Skip to content

Commit 74f91e0

Browse files
committed
TST: Fix tests which used tuples to pass multiple keys
1 parent 201a4fe commit 74f91e0

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

pandas/tests/groupby/test_groupby.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -264,7 +264,7 @@ def test_len(self):
264264
df = pd.DataFrame(dict(a=[np.nan] * 3, b=[1, 2, 3]))
265265
assert len(df.groupby(('a'))) == 0
266266
assert len(df.groupby(('b'))) == 3
267-
assert len(df.groupby(('a', 'b'))) == 3
267+
assert len(df.groupby(['a', 'b'])) == 3
268268

269269
def test_basic_regression(self):
270270
# regression

pandas/tests/groupby/test_nth.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def test_nth(self):
202202
freq='B')
203203
df = DataFrame(1, index=business_dates, columns=['a', 'b'])
204204
# get the first, fourth and last two business days for each month
205-
key = (df.index.year, df.index.month)
205+
key = [df.index.year, df.index.month]
206206
result = df.groupby(key, as_index=False).nth([0, 3, -2, -1])
207207
expected_dates = pd.to_datetime(
208208
['2014/4/1', '2014/4/4', '2014/4/29', '2014/4/30', '2014/5/1',

pandas/tests/groupby/test_value_counts.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def seed_df(seed_nans, n, m):
4343

4444
df = seed_df(seed_nans, n, m)
4545
bins = None, np.arange(0, max(5, df['3rd'].max()) + 1, 2)
46-
keys = '1st', '2nd', ('1st', '2nd')
46+
keys = '1st', '2nd', ['1st', '2nd']
4747
for k, b in product(keys, bins):
4848
binned.append((df, k, b, n, m))
4949
ids.append("{}-{}-{}".format(k, n, m))

0 commit comments

Comments
 (0)