cft

Author: h-vetinari

pandas/tests/frame/conftest.py

@@ -8,37 +8,67 @@
 
 
 @pytest.fixture
-def frame():
+def float_frame():
+    """
+    Fixture for DataFrame of floats with index of unique strings
+
+    Columns are ['A', 'B', 'C', 'D'].
+    """
     return DataFrame(tm.getSeriesData())
 
 
 @pytest.fixture
-def frame2():
+def float_frame2():
+    """
+    Fixture for DataFrame of floats with index of unique strings
+
+    Columns are ['D', 'C', 'B', 'A']
+    """
     return DataFrame(tm.getSeriesData(), columns=['D', 'C', 'B', 'A'])
 
 
 @pytest.fixture
-def intframe():
+def int_frame():
+    """
+    Fixture for DataFrame of ints with index of unique strings
+
+    Columns are ['A', 'B', 'C', 'D']
+    """
     df = DataFrame({k: v.astype(int)
                    for k, v in compat.iteritems(tm.getSeriesData())})
     # force these all to int64 to avoid platform testing issues
     return DataFrame({c: s for c, s in compat.iteritems(df)}, dtype=np.int64)
 
 
 @pytest.fixture
-def tsframe():
+def datetime_frame():
+    """
+    Fixture for DataFrame of floats with DatetimeIndex
+
+    Columns are ['A', 'B', 'C', 'D']
+    """
     return DataFrame(tm.getTimeSeriesData())
 
 
 @pytest.fixture
-def mixed_frame():
+def float_string_frame():
+    """
+    Fixture for DataFrame of floats and strings with index of unique strings
+
+    Columns are ['A', 'B', 'C', 'D', 'foo'].
+    """
     df = DataFrame(tm.getSeriesData())
     df['foo'] = 'bar'
     return df
 
 
 @pytest.fixture
-def mixed_float():
+def mixed_float_frame():
+    """
+    Fixture for DataFrame of different float types with index of unique strings
+
+    Columns are ['A', 'B', 'C', 'D'].
+    """
     df = DataFrame(tm.getSeriesData())
     df.A = df.A.astype('float16')
     df.B = df.B.astype('float32')
@@ -47,7 +77,12 @@ def mixed_float():
 
 
 @pytest.fixture
-def mixed_float2():
+def mixed_float_frame2():
+    """
+    Fixture for DataFrame of different float types with index of unique strings
+
+    Columns are ['A', 'B', 'C', 'D'].
+    """
     df = DataFrame(tm.getSeriesData())
     df.D = df.D.astype('float16')
     df.C = df.C.astype('float32')
@@ -56,7 +91,12 @@ def mixed_float2():
 
 
 @pytest.fixture
-def mixed_int():
+def mixed_int_frame():
+    """
+    Fixture for DataFrame of different int types with index of unique strings
+
+    Columns are ['A', 'B', 'C', 'D'].
+    """
     df = DataFrame({k: v.astype(int)
                    for k, v in compat.iteritems(tm.getSeriesData())})
     df.A = df.A.astype('uint8')
@@ -67,15 +107,25 @@ def mixed_int():
 
 
 @pytest.fixture
-def all_mixed():
+def mixed_type_frame():
+    """
+    Fixture for DataFrame of float/int/string columns with RangeIndex
+
+    Columns are ['a', 'b', 'c', 'float32', 'int32'].
+    """
     return DataFrame({'a': 1., 'b': 2, 'c': 'foo',
                       'float32': np.array([1.] * 10, dtype='float32'),
                       'int32': np.array([1] * 10, dtype='int32')},
                      index=np.arange(10))
 
 
 @pytest.fixture
-def tzframe():
+def timezone_frame():
+    """
+    Fixture for DataFrame of date_range Series with different time zones
+
+    Columns are ['A', 'B', 'C']; some entries are missing
+    """
     df = DataFrame({'A': date_range('20130101', periods=3),
                     'B': date_range('20130101', periods=3,
                                     tz='US/Eastern'),
@@ -87,22 +137,36 @@ def tzframe():
 
 
 @pytest.fixture
-def empty():
+def empty_frame():
+    """
+    Fixture for empty DataFrame
+    """
     return DataFrame({})
 
 
 @pytest.fixture
-def ts1():
+def datetime_series():
+    """
+    Fixture for Series of floats with DatetimeIndex
+    """
     return tm.makeTimeSeries(nper=30)
 
 
 @pytest.fixture
-def ts2():
+def datetime_series_short():
+    """
+    Fixture for Series of floats with DatetimeIndex
+    """
     return tm.makeTimeSeries(nper=30)[5:]
 
 
 @pytest.fixture
-def simple():
+def simple_frame():
+    """
+    Fixture for simple 3x3 DataFrame
+
+    Columns are ['one', 'two', 'three'], index is ['a', 'b', 'c'].
+    """
     arr = np.array([[1., 2., 3.],
                     [4., 5., 6.],
                     [7., 8., 9.]])
@@ -113,6 +177,12 @@ def simple():
 
 @pytest.fixture
 def frame_of_index_cols():
+    """
+    Fixture for DataFrame of columns that can be used for indexing
+
+    Columns are ['A', 'B', 'C', 'D', 'E']; 'A' & 'B' contain duplicates (but
+    are jointly unique), the rest are unique.
+    """
     df = DataFrame({'A': ['foo', 'foo', 'foo', 'bar', 'bar'],
                     'B': ['one', 'two', 'three', 'one', 'two'],
                     'C': ['a', 'b', 'c', 'd', 'e'],