tests for CI

Author: jreback

doc/source/whatsnew/v0.20.0.txt

@@ -598,52 +598,68 @@ ndarray, you can always convert explicitly using ``np.asarray(idx.hour)``.
 pd.unique will now be consistent with extension types
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
 
-In prior versions, using ``Series.unique()`` and ``pd.unique(Series)`` on ``Categorical`` and tz-aware datatypes would yield different return types. These are now consistent to return the extension type. Note that the behavior of ``Index.unique()`` and ``pd.unique(Index)`` has not changed. (:issue:`15939`)
+In prior versions, using ``Series.unique()`` and ``pd.unique(Series)`` on ``Categorical`` and tz-aware datatypes would yield different return types. These are now consistent to return the extension type.
 
-Previous behaviour:
+- Datetime tz-aware
 
-Datetime tz-aware
+  Previous behaviour:
 
-.. code-block:: ipython
+  .. code-block:: ipython
 
-   In [5]: Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
-                            pd.Timestamp('20160101', tz='US/Eastern')])).unique()
-   Out[5]: array([Timestamp('2016-01-01 00:00:00-0500', tz='US/Eastern')], dtype=object)
+     # Series
+     In [5]: Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+                              pd.Timestamp('20160101', tz='US/Eastern')])).unique()
+     Out[5]: array([Timestamp('2016-01-01 00:00:00-0500', tz='US/Eastern')], dtype=object)
 
-   In [7]: pd.unique(Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
-                                      pd.Timestamp('20160101', tz='US/Eastern')])))
-   Out[7]: array(['2016-01-01T05:00:00.000000000'], dtype='datetime64[ns]')
+     In [6]: pd.unique(Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+                                        pd.Timestamp('20160101', tz='US/Eastern')])))
+     Out[6]: array(['2016-01-01T05:00:00.000000000'], dtype='datetime64[ns]')
 
-Categoricals
+     # Index
+     In [7]: pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+                       pd.Timestamp('20160101', tz='US/Eastern')]).unique()
+     Out[7]: DatetimeIndex(['2016-01-01 00:00:00-05:00'], dtype='datetime64[ns, US/Eastern]', freq=None)
 
-.. code-block:: ipython
+     In [8]: pd.unique(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+                       pd.Timestamp('20160101', tz='US/Eastern')]))
+     Out[8]: array(['2016-01-01T05:00:00.000000000'], dtype='datetime64[ns]')
 
+  New Behavior:
 
-   In [1]: pd.Series(pd.Categorical(list('aabc'))).unique()
-   Out[1]:
-   [a, b, c]
-   Categories (3, object): [a, b, c]
+  .. ipython:: python
 
-   In [2]: pd.unique(pd.Series(pd.Categorical(list('aabc'))).unique())
-   Out[2]: array(['a', 'b', 'c'], dtype=object)
+     # Series
+     Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+                      pd.Timestamp('20160101', tz='US/Eastern')])).unique()
+     pd.unique(Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+                                pd.Timestamp('20160101', tz='US/Eastern')])))
 
-New Behavior:
+     # Index
+     pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+               pd.Timestamp('20160101', tz='US/Eastern')])
+     pd.unique(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
+                         pd.Timestamp('20160101', tz='US/Eastern')]))
 
-Datetime tz-aware
+- Categoricals
 
-.. ipython:: python
+  Previous behaviour:
+
+  .. code-block:: ipython
 
-   Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
-                    pd.Timestamp('20160101', tz='US/Eastern')])).unique()
-   pd.unique(Series(pd.Index([pd.Timestamp('20160101', tz='US/Eastern'),
-                              pd.Timestamp('20160101', tz='US/Eastern')])))
+     In [1]: pd.Series(pd.Categorical(list('aabc'))).unique()
+     Out[1]:
+     [a, b, c]
+     Categories (3, object): [a, b, c]
 
-Categoricals
+     In [2]: pd.unique(pd.Series(pd.Categorical(list('aabc'))).unique())
+     Out[2]: array(['a', 'b', 'c'], dtype=object)
 
-.. ipython:: python
+  New Behavior:
+
+  .. ipython:: python
 
-   pd.Series(pd.Categorical(list('aabc'))).unique()
-   pd.unique(pd.Series(pd.Categorical(list('aabc'))).unique())
+     pd.Series(pd.Categorical(list('aabc'))).unique()
+     pd.unique(pd.Series(pd.Categorical(list('aabc'))).unique())
 
 .. _whatsnew_0200.api_breaking.s3:
 

pandas/tests/test_algos.py

@@ -399,6 +399,15 @@ def test_categorical(self):
         expected = Series(expected, name='foo')
         tm.assert_series_equal(result, expected)
 
+        # CI
+        ci = pd.CategoricalIndex(pd.Categorical(list('aabc')))
+        result = ci.unique()
+        expected = pd.CategoricalIndex(pd.Categorical(list('abc')))
+        tm.assert_index_equal(result, expected)
+
+        result = pd.unique(ci)
+        tm.assert_index_equal(result, expected)
+
     def test_datetime64tz_aware(self):
         # GH 15939