Fixturize tests/frame/test_apply

pandas-dev · Sep 17, 2018 · bbc92b2 · bbc92b2
1 parent 8a1c8ad
commit bbc92b2
Show file tree

Hide file tree

Showing 2 changed files with 171 additions and 190 deletions.
diff --git a/pandas/tests/frame/conftest.py b/pandas/tests/frame/conftest.py
@@ -40,6 +40,18 @@ def int_frame():
     return DataFrame({c: s for c, s in compat.iteritems(df)}, dtype=np.int64)
 
 
+@pytest.fixture
+def int_frame_const_col():
+    """
+    Fixture for DataFrame of ints which are constant per column
+
+    Columns are ['A', 'B', 'C'], with values (per column): [1, 2, 3]
+    """
+    df = DataFrame(np.tile(np.arange(3, dtype='int64'), 6).reshape(6, -1) + 1,
+                   columns=['A', 'B', 'C'])
+    return df
+
+
 @pytest.fixture
 def datetime_frame():
     """