Use simpler criterion for deleting info['class']; fix docstring fail

taldcroft · taldcroft · commit ed8a3e197bc7 · 2015-06-12T17:06:01.000-04:00
diff --git a/astropy/table/info.py b/astropy/table/info.py
@@ -95,9 +95,9 @@ def table_info(tbl, option='attributes', out=''):
 
     if 'class' in info.colnames:
         # Remove 'class' info column if all table columns are the same class
-        # and they are not mixin columns.
+        # and they are the default column class for that table.
         uniq_types = set(type(col) for col in cols)
-        if len(uniq_types) == 1 and not tbl._is_mixin_column(cols[0]):
+        if len(uniq_types) == 1 and isinstance(cols[0], tbl.ColumnClass):
             del info['class']
 
     if 'n_bad' in info.colnames and np.all(info['n_bad'] == 0):
diff --git a/astropy/utils/data_info.py b/astropy/utils/data_info.py
@@ -230,12 +230,14 @@ def __call__(self, option='attributes', out=''):
         >>> c.info()
         dtype = int32
         unit = m
+        class = Column
         n_bad = 0
         length = 2
 
         >>> c.info(['attributes', 'stats'])
         dtype = int32
         unit = m
+        class = Column
         mean = 1.5
         std = 0.5
         min = 1
