Skip to content

Commit 49aa096

Browse files
committed
Remove week/weekofyear attributes from tests
1 parent adb9cff commit 49aa096

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

pandas/tests/groupby/transform/test_transform.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1025,7 +1025,9 @@ def test_groupby_transform_with_datetimes(func, values):
10251025
dates = pd.date_range("1/1/2011", periods=10, freq="D")
10261026

10271027
stocks = pd.DataFrame({"price": np.arange(10.0)}, index=dates)
1028-
stocks["week_id"] = pd.to_datetime(stocks.index).week
1028+
stocks["week_id"] = (
1029+
pd.to_datetime(stocks.index).isocalendar().set_index(stocks.index).week
1030+
)
10291031

10301032
result = stocks.groupby(stocks["week_id"])["price"].transform(func)
10311033

pandas/tests/indexes/datetimes/test_misc.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,8 @@ def test_datetimeindex_accessors(self):
156156
assert dti.dayofyear[0] == 1
157157
assert dti.dayofyear[120] == 121
158158

159-
assert dti.weekofyear[0] == 1
160-
assert dti.weekofyear[120] == 18
159+
assert dti.isocalendar().week[0] == 1
160+
assert dti.isocalendar().week[120] == 18
161161

162162
assert dti.quarter[0] == 1
163163
assert dti.quarter[120] == 2
@@ -192,7 +192,7 @@ def test_datetimeindex_accessors(self):
192192
assert len(dti.microsecond) == 365
193193
assert len(dti.dayofweek) == 365
194194
assert len(dti.dayofyear) == 365
195-
assert len(dti.weekofyear) == 365
195+
assert len(dti.isocalendar()) == 365
196196
assert len(dti.quarter) == 365
197197
assert len(dti.is_month_start) == 365
198198
assert len(dti.is_month_end) == 365
@@ -288,7 +288,7 @@ def test_datetimeindex_accessors(self):
288288
dates = ["2013/12/29", "2013/12/30", "2013/12/31"]
289289
dates = DatetimeIndex(dates, tz="Europe/Brussels")
290290
expected = [52, 1, 1]
291-
assert dates.weekofyear.tolist() == expected
291+
assert dates.isocalendar().week.tolist() == expected
292292
assert [d.weekofyear for d in dates] == expected
293293

294294
# GH 12806

0 commit comments

Comments
 (0)