rremove copy

Author: ryankarlos

pandas/tests/groupby/test_transform.py

@@ -356,54 +356,49 @@ def test_transform_transformation_func(transformation_func):
  tm.assert_frame_equal(result, expected)
 
 
-def test_groupby_corrwith(transformation_func, df_for_transformation_func):
+def test_groupby_corrwith(df_for_transformation_func):
 
  # GH 27905
- df = df_for_transformation_func.copy()
+ df = df_for_transformation_func
  g = df.groupby("A")
 
- if transformation_func == "corrwith":
- op = lambda x: getattr(x, transformation_func)(df)
- result = op(g)
- expected = pd.DataFrame(dict(B=[1, np.nan, np.nan], A=[np.nan] * 3))
- expected.index = pd.Index([121, 231, 676], name="A")
- tm.assert_frame_equal(result, expected)
+ op = lambda x: getattr(x, "corrwith")(df)
+ result = op(g)
+ expected = pd.DataFrame(dict(B=[1, np.nan, np.nan], A=[np.nan] * 3))
+ expected.index = pd.Index([121, 231, 676], name="A")
+ tm.assert_frame_equal(result, expected)
 
 
-def test_groupby_transform_nan(transformation_func, df_for_transformation_func):
+def test_groupby_transform_nan(df_for_transformation_func):
 
  # GH 27905
- df = df_for_transformation_func.copy()
+ df = df_for_transformation_func
  g = df.groupby("A")
 
- if transformation_func == "fillna":
-
- df["B"] = [1, np.nan, np.nan, 3, np.nan, 3, 4]
- result = g.transform(transformation_func, value=1)
- expected = pd.DataFrame({"B": [1.0, 1.0, 1.0, 3.0, 1.0, 3.0, 4.0]})
- tm.assert_frame_equal(result, expected)
- op = lambda x: getattr(x, transformation_func)(1)
- result = op(g)
- tm.assert_frame_equal(result, expected)
+ df["B"] = [1, np.nan, np.nan, 3, np.nan, 3, 4]
+ result = g.transform("fillna", value=1)
+ expected = pd.DataFrame({"B": [1.0, 1.0, 1.0, 3.0, 1.0, 3.0, 4.0]})
+ tm.assert_frame_equal(result, expected)
+ op = lambda x: getattr(x, "fillna")(1)
+ result = op(g)
+ tm.assert_frame_equal(result, expected)
 
 
-def test_groupby_tshift(transformation_func, df_for_transformation_func):
+def test_groupby_tshift(df_for_transformation_func):
 
  # GH 27905
- df = df_for_transformation_func.copy()
+ df = df_for_transformation_func
  dt_periods = pd.date_range("2013-11-03", periods=7, freq="D")
  df["C"] = dt_periods
  g = df.set_index("C").groupby("A")
 
- if transformation_func == "tshift":
-
- op = lambda x: getattr(x, transformation_func)(2, "D")
- result = op(g)
- df["C"] = dt_periods + dt_periods.freq * 2
- expected = df
- tm.assert_frame_equal(
- result.reset_index().reindex(columns=["A", "B", "C"]), expected
- )
+ op = lambda x: getattr(x, "tshift")(2, "D")
+ result = op(g)
+ df["C"] = dt_periods + dt_periods.freq * 2
+ expected = df
+ tm.assert_frame_equal(
+ result.reset_index().reindex(columns=["A", "B", "C"]), expected
+ )
 
 
 def test_check_original_and_transformed_index(transformation_func):