Skip to content

Commit 5581710

Browse files
committed
Remove week/weekofyear attributes from tests
1 parent 90995ca commit 5581710

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
@@ -148,8 +148,8 @@ def test_datetimeindex_accessors(self):
148148
assert dti.dayofyear[0] == 1
149149
assert dti.dayofyear[120] == 121
150150

151-
assert dti.weekofyear[0] == 1
152-
assert dti.weekofyear[120] == 18
151+
assert dti.isocalendar().week[0] == 1
152+
assert dti.isocalendar().week[120] == 18
153153

154154
assert dti.quarter[0] == 1
155155
assert dti.quarter[120] == 2
@@ -184,7 +184,7 @@ def test_datetimeindex_accessors(self):
184184
assert len(dti.microsecond) == 365
185185
assert len(dti.dayofweek) == 365
186186
assert len(dti.dayofyear) == 365
187-
assert len(dti.weekofyear) == 365
187+
assert len(dti.isocalendar()) == 365
188188
assert len(dti.quarter) == 365
189189
assert len(dti.is_month_start) == 365
190190
assert len(dti.is_month_end) == 365
@@ -277,7 +277,7 @@ def test_datetimeindex_accessors(self):
277277
dates = ["2013/12/29", "2013/12/30", "2013/12/31"]
278278
dates = DatetimeIndex(dates, tz="Europe/Brussels")
279279
expected = [52, 1, 1]
280-
assert dates.weekofyear.tolist() == expected
280+
assert dates.isocalendar().week.tolist() == expected
281281
assert [d.weekofyear for d in dates] == expected
282282

283283
# GH 12806

0 commit comments

Comments
 (0)