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

fangchenli · fangchenli · commit 787d9ba4a183 · 2020-06-09T00:31:09.000-05:00
diff --git a/pandas/tests/groupby/test_apply.py b/pandas/tests/groupby/test_apply.py
@@ -968,9 +968,14 @@ def test_apply_function_with_indexing_return_column():
     df = DataFrame(
         {
             "foo1": ["one", "two", "two", "three", "one", "two"],
-            "foo2": [1, 2, 3, 4, 5, 6],
+            "foo2": [1, 2, 4, 4, 5, 6],
         }
     )
     result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())
-    expected = df.groupby("foo1", as_index=False).mean()
+    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
`@@ -968,9 +968,14 @@ def test_apply_function_with_indexing_return_column():`
`968`	`968`	`df = DataFrame(`
`969`	`969`	`{`
`970`	`970`	`"foo1": ["one", "two", "two", "three", "one", "two"],`
`971`		`- "foo2": [1, 2, 3, 4, 5, 6],`
	`971`	`+ "foo2": [1, 2, 4, 4, 5, 6],`
`972`	`972`	`}`
`973`	`973`	`)`
`974`	`974`	`result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())`
`975`		`- expected = df.groupby("foo1", as_index=False).mean()`
	`975`	`+ expected = DataFrame(`
	`976`	`+ {`
	`977`	`+ "foo1": ["one", "three", "two"],`
	`978`	`+ "foo2": [3.0, 4.0, 4.0],`
	`979`	`+ }`
	`980`	`+ )`
`976`	`981`	`tm.assert_frame_equal(result, expected)`