TST: Test describe method of Categorical

Author: jseabold

pandas/tests/test_factor.py

@@ -9,6 +9,7 @@
 from pandas.core.api import value_counts
 from pandas.core.categorical import Categorical
 from pandas.core.index import Index, Int64Index, MultiIndex
+from pandas.core.frame import DataFrame
 from pandas.util.testing import assert_almost_equal
 import pandas.core.common as com
 
@@ -116,6 +117,23 @@ def test_levels_none(self):
  'a', 'c', 'c', 'c'])
  self.assert_(factor.equals(self.factor))
 
+ def test_describe(self):
+ # string type
+ desc = self.factor.describe()
+ expected = DataFrame.from_dict(dict(counts=[3, 2, 3],
+ freqs=[3/8., 2/8., 3/8.],
+ levels=['a', 'b', 'c'])
+ ).set_index('levels')
+ tm.assert_frame_equal(desc, expected)
+
+ # check an integer one
+ desc = Categorical([1,2,3,1,2,3,3,2,1,1,1]).describe()
+ expected = DataFrame.from_dict(dict(counts=[5, 3, 3],
+ freqs=[5/11., 3/11., 3/11.],
+ levels=[1,2,3]
+ )
+ ).set_index('levels')
+ tm.assert_frame_equal(desc, expected)
 
 if __name__ == '__main__':
  import nose