Remove TestData from frame-tests test_repr_info.py

Author: SaturnFromTitan

pandas/tests/frame/conftest.py

@@ -5,6 +5,23 @@
 import pandas.util.testing as tm
 
 
+@pytest.fixture
+def empty_frame():
+    """
+    Fixture for DataFrame that is empty
+    """
+    return DataFrame()
+
+
+@pytest.fixture
+def float_frame():
+    """
+    Fixture for DataFrame of floats with index of unique strings
+    """
+    df = DataFrame(tm.getSeriesData())
+    return df
+
+
 @pytest.fixture
 def float_frame_with_na():
     """

pandas/tests/frame/test_repr_info.py

@@ -18,7 +18,6 @@
     option_context,
     period_range,
 )
-from pandas.tests.frame.common import TestData
 import pandas.util.testing as tm
 
 import pandas.io.formats.format as fmt
@@ -27,21 +26,21 @@
 # structure
 
 
-class TestDataFrameReprInfoEtc(TestData):
-    def test_repr_empty(self):
+class TestDataFrameReprInfoEtc:
+    def test_repr_empty(self, empty_frame):
         # empty
-        foo = repr(self.empty)  # noqa
+        foo = repr(empty_frame)  # noqa
 
         # empty with index
         frame = DataFrame(index=np.arange(1000))
         foo = repr(frame)  # noqa
 
-    def test_repr_mixed(self):
+    def test_repr_mixed(self, float_string_frame):
         buf = StringIO()
 
         # mixed
-        foo = repr(self.mixed_frame)  # noqa
-        self.mixed_frame.info(verbose=False, buf=buf)
+        foo = repr(float_string_frame)  # noqa
+        float_string_frame.info(verbose=False, buf=buf)
 
     @pytest.mark.slow
     def test_repr_mixed_big(self):
@@ -54,16 +53,16 @@ def test_repr_mixed_big(self):
 
         foo = repr(biggie)  # noqa
 
-    def test_repr(self):
+    def test_repr(self, float_frame, empty_frame):
         buf = StringIO()
 
         # small one
-        foo = repr(self.frame)
-        self.frame.info(verbose=False, buf=buf)
+        foo = repr(float_frame)
+        float_frame.info(verbose=False, buf=buf)
 
         # even smaller
-        self.frame.reindex(columns=["A"]).info(verbose=False, buf=buf)
-        self.frame.reindex(columns=["A", "B"]).info(verbose=False, buf=buf)
+        float_frame.reindex(columns=["A"]).info(verbose=False, buf=buf)
+        float_frame.reindex(columns=["A", "B"]).info(verbose=False, buf=buf)
 
         # exhausting cases in DataFrame.info
 
@@ -72,7 +71,7 @@ def test_repr(self):
         foo = repr(no_index)  # noqa
 
         # no columns or index
-        self.empty.info(buf=buf)
+        empty_frame.info(buf=buf)
 
         df = DataFrame(["a\n\r\tb"], columns=["a\n\r\td"], index=["a\n\r\tf"])
         assert "\t" not in repr(df)
@@ -96,7 +95,7 @@ def test_repr_big(self):
         biggie = DataFrame(np.zeros((200, 4)), columns=range(4), index=range(200))
         repr(biggie)
 
-    def test_repr_unsortable(self):
+    def test_repr_unsortable(self, float_frame):
         # columns are not sortable
         import warnings
 
@@ -115,13 +114,13 @@ def test_repr_unsortable(self):
         repr(unsortable)
 
         fmt.set_option("display.precision", 3, "display.column_space", 10)
-        repr(self.frame)
+        repr(float_frame)
 
         fmt.set_option("display.max_rows", 10, "display.max_columns", 2)
-        repr(self.frame)
+        repr(float_frame)
 
         fmt.set_option("display.max_rows", 1000, "display.max_columns", 1000)
-        repr(self.frame)
+        repr(float_frame)
 
         tm.reset_display_options()
 
@@ -196,10 +195,10 @@ def test_latex_repr(self):
         # GH 12182
         assert df._repr_latex_() is None
 
-    def test_info(self):
+    def test_info(self, float_frame, datetime_frame):
         io = StringIO()
-        self.frame.info(buf=io)
-        self.tsframe.info(buf=io)
+        float_frame.info(buf=io)
+        datetime_frame.info(buf=io)
 
         frame = DataFrame(np.random.randn(5, 3))