TST: deterministic test case (pandas-dev#7002)

fangchenli · fangchenli · commit 0bd567c5b17d · 2020-06-08T15:53:13.000-05:00
diff --git a/pandas/tests/groupby/test_apply.py b/pandas/tests/groupby/test_apply.py
@@ -968,10 +968,9 @@ def test_apply_function_with_indexing_return_column():
     df = DataFrame(
         {
             "foo1": ["one", "two", "two", "three", "one", "two"],
-            "foo2": np.random.randn(6),
+            "foo2": [1, 2, 3, 4, 5, 6],
         }
     )
     result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())
     expected = df.groupby("foo1", as_index=False).mean()
     tm.assert_frame_equal(result, expected)
-    assert "foo1" in result.columns

Original file line number	Diff line number	Diff line change
`@@ -968,10 +968,9 @@ 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": np.random.randn(6),`
	`971`	`+ "foo2": [1, 2, 3, 4, 5, 6],`
`972`	`972`	`}`
`973`	`973`	`)`
`974`	`974`	`result = df.groupby("foo1", as_index=False).apply(lambda x: x.mean())`
`975`	`975`	`expected = df.groupby("foo1", as_index=False).mean()`
`976`	`976`	`tm.assert_frame_equal(result, expected)`
`977`		`- assert "foo1" in result.columns`