Skip to content

Commit 9788bf4

Browse files
author
Marco Gorelli
committed
🎨 shorten test for the sake of legibility
1 parent 948f247 commit 9788bf4

File tree

1 file changed

+4
-36
lines changed

1 file changed

+4
-36
lines changed

pandas/tests/groupby/aggregate/test_aggregate.py

+4-36
Original file line numberDiff line numberDiff line change
@@ -245,43 +245,11 @@ def test_agg_multiple_functions_same_name(df):
245245
result = df.resample("3D").agg(
246246
{"A": [functools.partial(np.std, ddof=0), functools.partial(np.std, ddof=1)]}
247247
)
248-
expected_index = pd.DatetimeIndex(
249-
[
250-
"2000-01-03",
251-
"2000-01-06",
252-
"2000-01-09",
253-
"2000-01-12",
254-
"2000-01-15",
255-
"2000-01-18",
256-
"2000-01-21",
257-
"2000-01-24",
258-
"2000-01-27",
259-
"2000-01-30",
260-
"2000-02-02",
261-
"2000-02-05",
262-
"2000-02-08",
263-
"2000-02-11",
264-
],
265-
dtype="datetime64[ns]",
266-
freq="3D",
267-
)
248+
expected_index = pd.date_range("2000-01-03", "2000-02-11", freq="3D")
268249
expected_columns = pd.MultiIndex.from_tuples([("A", "std"), ("A", "std")])
269-
expected_values = [
270-
[1.03497007, 1.26757429],
271-
[0.96918813, 1.37063899],
272-
[2.02317523, 2.86120185],
273-
[0.44121013, 0.54036984],
274-
[0.0, np.nan],
275-
[0.80503606, 0.98596379],
276-
[0.0, np.nan],
277-
[0.3954067, 0.48427232],
278-
[0.27030073, 0.38226296],
279-
[1.12267144, 1.58769718],
280-
[0.18796289, 0.23020659],
281-
[0.0, np.nan],
282-
[0.3540029, 0.43356324],
283-
[0.0, np.nan],
284-
]
250+
expected_values = np.array(
251+
[df.resample("3D").A.std(ddof=i).values for i in range(2)]
252+
).T
285253
expected = pd.DataFrame(
286254
expected_values, columns=expected_columns, index=expected_index
287255
)

0 commit comments

Comments
 (0)