TST: add test to cover bug GH pandas-dev#14580

Add test to verify type of returned value of iloc() Series of Categorical data.

Author: nathalier

pandas/tests/series/test_indexing.py

@@ -409,6 +409,23 @@ def test_getitem_callable(self):
         result = s[lambda x: [True, False, True, True]]
         tm.assert_series_equal(result, s.iloc[[0, 2, 3]])
 
+    def test_getitem_category_type(self):
+        # GH 14580
+
+        s = pd.Series([1, 2, 3]).astype('category')
+
+        # get slice
+        result = s.iloc[0:1]
+        self.assertEqual(type(result), type(s))
+
+        # get list of indexes
+        result = s.iloc[[0, 1]]
+        self.assertEqual(type(result), type(s))
+
+        # get boolean array
+        result = s.iloc[[True, False, False]]
+        self.assertEqual(type(result), type(s))
+
     def test_setitem_ambiguous_keyerror(self):
         s = Series(lrange(10), index=lrange(0, 20, 2))