TST: explicitly constructed expected DataFrame (pandas-dev#7002)

fangchenli · fangchenli · commit 8139d8ee9c04 · 2020-06-09T01:02:15.000-05:00
diff --git a/pandas/tests/groupby/test_apply.py b/pandas/tests/groupby/test_apply.py
@@ -972,10 +972,5 @@ def test_apply_function_with_indexing_return_column():
         }
     )
     result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())
-    expected = DataFrame(
-        {
-            "foo1": ["one", "three", "two"],
-            "foo2": [3.0, 4.0, 4.0],
-        }
-    )
+    expected = DataFrame({"foo1": ["one", "three", "two"], "foo2": [3.0, 4.0, 4.0],})
     tm.assert_frame_equal(result, expected)

Original file line number	Diff line number	Diff line change
`@@ -972,10 +972,5 @@ def test_apply_function_with_indexing_return_column():`
`972`	`972`	`}`
`973`	`973`	`)`
`974`	`974`	`result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())`
`975`		`- expected = DataFrame(`
`976`		`- {`
`977`		`- "foo1": ["one", "three", "two"],`
`978`		`- "foo2": [3.0, 4.0, 4.0],`
`979`		`- }`
`980`		`- )`
	`975`	`+ expected = DataFrame({"foo1": ["one", "three", "two"], "foo2": [3.0, 4.0, 4.0],})`
`981`	`976`	`tm.assert_frame_equal(result, expected)`