TST: df.info(null_counts=False) header as Dtype

Author: Pratap Vardhan

doc/source/whatsnew/v0.24.0.txt

@@ -71,7 +71,7 @@ Current Behavior:
 Output Formatting Enhancements
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 
-- `df.info()` now shows line numbers for the columns summary (:issue:`17304`)
+- :func:`DataFrame.info` now shows line numbers for the columns summary (:issue:`17304`)
 
 .. ipython:: python
 

pandas/core/frame.py

@@ -2121,8 +2121,8 @@ def info(self, verbose=None, buf=None, max_cols=None, memory_usage=None,
         <class 'pandas.core.frame.DataFrame'>
         RangeIndex: 5 entries, 0 to 4
         Data columns (total 3 columns):
-         #.  Column       Non-Null Count
-        ---  ------       --------------
+         #.  Column       Non-Null Count & Dtype
+        ---  ------       ----------------------
          0   int_col      5 non-null int64
          1   text_col     5 non-null object
          2   float_col    5 non-null float64
@@ -2163,8 +2163,8 @@ def info(self, verbose=None, buf=None, max_cols=None, memory_usage=None,
         <class 'pandas.core.frame.DataFrame'>
         RangeIndex: 1000000 entries, 0 to 999999
         Data columns (total 3 columns):
-         #.  Column      Non-Null Count
-        ---  ------      --------------
+         #.  Column      Non-Null Count & Dtype
+        ---  ------      ----------------------
          0   column_1    1000000 non-null object
          1   column_2    1000000 non-null object
          2   column_3    1000000 non-null object
@@ -2175,8 +2175,8 @@ def info(self, verbose=None, buf=None, max_cols=None, memory_usage=None,
         <class 'pandas.core.frame.DataFrame'>
         RangeIndex: 1000000 entries, 0 to 999999
         Data columns (total 3 columns):
-         #.  Column      Non-Null Count
-        ---  ------      --------------
+         #.  Column      Non-Null Count & Dtype
+        ---  ------      ----------------------
          0   column_1    1000000 non-null object
          1   column_2    1000000 non-null object
          2   column_3    1000000 non-null object
@@ -2232,10 +2232,10 @@ def _verbose_repr():
                         'Columns must equal counts '
                         '({cols_count} != {count})'.format(
                             cols_count=cols_count, count=len(counts)))
-                col_header = 'Non-Null Count'
+                col_header = 'Non-Null Count & Dtype'
                 tmpl = '{count} non-null {dtype}'
             else:
-                col_header = 'dtype'
+                col_header = 'Dtype'
                 tmpl = '{count}{dtype}'
             header += col_header
 

pandas/tests/frame/test_repr_info.py

@@ -217,15 +217,33 @@ def test_info_memory(self):
         <class 'pandas.core.frame.DataFrame'>
         RangeIndex: 2 entries, 0 to 1
         Data columns (total 1 columns):
-         #.  Column    Non-Null Count
-        ---  ------    --------------
+         #.  Column    Non-Null Count & Dtype
+        ---  ------    ----------------------
          0   a         2 non-null int64
         dtypes: int64(1)
         memory usage: {} bytes
         """.format(bytes))
 
         assert result == expected
 
+    def test_info_without_null_counts(self):
+        df = pd.DataFrame({'a': [1, 2]})
+        buf = StringIO()
+        df.info(buf=buf, null_counts=False)
+        buf.seek(0)
+        lines = buf.readlines()
+        result = ''.join(lines[:-1])
+        expected = textwrap.dedent('''\
+        <class 'pandas.core.frame.DataFrame'>
+        RangeIndex: 2 entries, 0 to 1
+        Data columns (total 1 columns):
+         #.  Column    Dtype
+        ---  ------    -----
+         0   a         int64
+        dtypes: int64(1)
+        ''')
+        assert result == expected
+
     def test_info_wide(self):
         from pandas import set_option, reset_option
         io = StringIO()