Refactor test to improve readability

pandas-dev · Apr 20, 2020 · 29ce5ac · 29ce5ac
1 parent e6d0c79
commit 29ce5ac
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pandas/tests/groupby/transform/test_transform.py b/pandas/tests/groupby/transform/test_transform.py
@@ -1025,9 +1025,7 @@ def test_groupby_transform_with_datetimes(func, values):
     dates = pd.date_range("1/1/2011", periods=10, freq="D")
 
     stocks = pd.DataFrame({"price": np.arange(10.0)}, index=dates)
-    stocks["week_id"] = (
-        pd.to_datetime(stocks.index).isocalendar().set_index(stocks.index).week
-    )
+    stocks["week_id"] = dates.isocalendar().set_index(dates).week
 
     result = stocks.groupby(stocks["week_id"])["price"].transform(func)