ó
šxŠ\c           @   s×   d  d l  Z d  d l Z d  d l j Z d  d l m Z d  d l m Z d  d l	 Z
 d  d l	 m Z m Z m Z m Z m Z m Z m Z d  d l m Z d  d l m Z d  d l m Z m Z d e f d „  ƒ  YZ d S(	   iÿÿÿÿN(   t   is_categorical_dtype(   t   CategoricalDtype(   t   Categoricalt   CategoricalIndext	   DataFramet   Indext   Intervalt   Seriest	   Timestamp(   t   testing(   t   assert_frame_equalt   assert_series_equalt   TestCategoricalIndexc           B   sÎ   e  Z d  „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z	 d „  Z
 d	 „  Z d
 „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   c      	   C   s†  t  i t j d d d ƒd 6t t d ƒ ƒ j t t d ƒ ƒ ƒ d 6ƒ j d ƒ |  _ t  i t j d d d ƒd 6t t d ƒ ƒ j t t d ƒ ƒ ƒ d 6ƒ j d ƒ |  _	 t  i t j d d d ƒd 6t d	 d	 d
 d	 d d
 g ƒ j t d d
 d	 g d t
 ƒƒ d 6ƒ j d ƒ |  _ t  i t j d d d ƒd 6t d	 d	 d
 d	 d d
 g ƒ j t d d
 d	 g d t ƒƒ d 6ƒ j d ƒ |  _ d  S(   Ni   t   dtypet   int64t   At   aabbcat   cabt   Bt   cabei   i   i   t   ordered(   R   t   npt   arangeR   t   listt   astypet   CDTt	   set_indext   dft   df2t   Truet   df3t   Falset   df4(   t   selft   method(    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   setup_method   s    ((%%c            sc  |  j  j d } t i d d d g d 6t t d ƒ ƒ j t t d ƒ ƒ ƒ d 6ƒ j d ƒ } t | | ƒ |  j  j	 ƒ  ‰  d	 ˆ  j d <t i d	 d	 d
 d d d	 g d 6t t d ƒ ƒ j t t d ƒ ƒ ƒ d 6ƒ j d ƒ } t ˆ  | ƒ t
 j t ‡  f d †  ƒ ‡  f d †  } t
 j t | ƒ ‡  f d †  } t
 j t | ƒ ‡  f d †  } t
 j t | ƒ d  S(   Nt   ai    i   i   R   t   aaaR   R   i   i   i   i   R   c              s   ˆ  j  d S(   Nt   d(   t   loc(    (   R   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   <lambda>8   s    c              s   d ˆ  j  d <d  S(   Ni
   R&   (   R'   (    (   R   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   f:   s    c              s   d ˆ  j  d <d  S(   Ni
   R&   R   (   R&   R   (   R'   (    (   R   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR)   ?   s    c              s   d ˆ  j  d <d  S(   Ni
   R&   t   C(   R&   R*   (   R'   (    (   R   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR)   D   s    (   R   R'   R   R   R   R   R   R   R
   t   copyt   pytestt   raisest   KeyErrort	   TypeError(   R!   t   resultt   expectedR)   (    (   R   sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_loc_scalar'   s(    		c         C   sg   t  t d ƒ t d ƒ g ƒ } t d d g d | ƒ} | j d } | | d } | | k sc t ‚ d  S(   Ns
   12-31-1999s
   12-31-2000i   i   t   indexi    (   R   R   R   t   iloct   AssertionError(   R!   t   catst   sR1   R0   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_getitem_scalarI   s    c         C   sž   t  d d d d d d d g ƒ } | d } | d k s= t ‚ | d d !} t  d d g d d d d d g ƒ} t j | j | j ƒ t j | j | j ƒ d  S(   NR$   t   bt   cR&   i   i   t
   categories(   R   R5   t   tmt   assert_numpy_array_equalt   _codest   assert_index_equalR;   (   R!   t   catt   slicedR1   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_slicing_directlyT   s    !
$c         C   så  t  t d d d d g ƒ ƒ } | d  d  d … } t j d d d d g d t j ƒ} t j | j ƒ  | ƒ t i t j	 d ƒ d j
 d ƒ d	 6ƒ } t j | j d
 d d d d d g ƒ| d <t  d t d d ƒ g d d	 d g d d ƒ} | j d } t j | | ƒ t i t j	 d d ƒ j
 d ƒ d	 6d t j	 d d ƒ j
 d ƒ ƒ} t j | j d
 d d d d d g ƒ| d <| j d d !} t j | | ƒ t  d t d d ƒ g d d	 d g d d ƒ} | j d } t j | | ƒ d  S(   Ni   i   i   i   iÿÿÿÿR   id   R   t   valuet   binsi    i   i2   iK   t   Di   R3   t   namei
   i   i   i	   i   (   R   R   R   t   arrayR   R<   R=   t	   __array__R   R   R   t   pdt   cutRC   R   R4   R   R
   R'   (   R!   R@   t   reversedt   expR   R1   R0   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_slicing]   s"    $)+-%+-c         C   s  t  d d d d d d d g d d d d g ƒ} t d d d d d	 d
 d g ƒ } d d d d d d d g } t i | d 6| d 6d | ƒ} t  d d g d d d d g ƒ} t d d g ƒ } d d g } t i | d 6| d 6d | ƒ} t | d | d d ƒ}	 t d d g d d d g d d d d ƒ}
 d } | j d d … d  d  … f } t j | | ƒ t | d ƒ sxt ‚ | j d d  d  … f } t j	 | |
 ƒ t
 | d t j ƒ s½t ‚ | j d  d  … d f } t j	 | |	 ƒ t | ƒ søt ‚ | j d } | | k st ‚ | j d d … d  d  … f } t j | | ƒ t | d ƒ s\t ‚ | j d d  d  … f } t j	 | |
 ƒ t
 | d t j ƒ s¡t ‚ | j d  d  … d f } t j	 | |	 ƒ t | ƒ sÜt ‚ | j d } | | k sût ‚ | j d d … d  d  … f } t j | | ƒ t | d ƒ s@t ‚ | j d d  d  … f } t j	 | |
 ƒ t
 | d t j ƒ s…t ‚ | j d  d  … d f } t j	 | |	 ƒ t | ƒ sÀt ‚ | j d | j d f } | | k sìt ‚ | j d } | | k st ‚ | j d } | | k s*t ‚ | j d g } | | d d k } t j | | ƒ | | d d k } t j | | ƒ | j d } | | k s¡t ‚ | j d } t j	 | |
 ƒ t
 | d t j ƒ sÚt ‚ | j t d d ƒ } t j | | ƒ t | d ƒ st ‚ | j d d g } t j | | ƒ t | d ƒ sOt ‚ | j d  d  … d f } t j	 | |	 ƒ t | ƒ sŠt ‚ | j d  d  … t d d ƒ f } t j | | ƒ t | d ƒ sÒt ‚ | j d  d  … d d g f } t j | | ƒ t | d ƒ st ‚ d  S(   NR$   R:   R9   R;   t   ht   it   jt   kt   lt   mt   ni   i   i   i   i   i   i   R6   t   valuesR3   RF   R   t   objecti    (   i   i    (   RP   R6   (   i   i    (   RP   R6   (   RP   R6   (   R   R   R   R   R4   R<   R
   R    R5   R   t
   isinstancet   compatt   string_typesR'   t   columnst   iatt   att   slice(   R!   R6   t   idxRU   R   t   cats2t   idx2t   values2t   exp_dft   exp_colt   exp_rowt   exp_valt   res_dft   res_rowt   res_colt   res_valt	   exp_fancyt	   res_fancy(    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_slicing_and_getting_opst   s–    	-! ! !	"c         C   s.  t  d d d d d d d g d d d d g ƒ} t d d d d d	 d
 d g ƒ } d d d d d d d g } t i | d 6| d 6d | ƒ} | j d d … d  d  … f } t i t  d d g d d d d g ƒd 6d d g d 6d d d g ƒ} t j | | ƒ | j d d … d  d  … f j } t d d g d d g ƒ } t j | | ƒ | j	 d d … d f } t t  d d d g d d d d g ƒd d d d g d d ƒ} t j | | ƒ | j	 d d … | j
 d d !f } t i t  d d d g d d d d g ƒd 6d d d d g ƒ} t j | | ƒ d  S(   NR$   R9   R:   R;   RN   RO   RP   RQ   RR   RS   RT   i   i   i   i   i   R6   RU   R3   t   categoryR   RF   i    (   R   R   R   R4   R<   R
   t   dtypesR   R   R'   RZ   (   R!   R6   R^   RU   R   R0   R1   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_slicing_doc_examplesÿ   s.    ! %"#c         C   s  t  d d d g ƒ j d ƒ } | j d d !} t  d d g ƒ j t d d d g ƒ ƒ } t j | | ƒ | j d d g } t  d d g ƒ j t d d d g ƒ ƒ } t j | | ƒ | j t t t g } t  d g ƒ j t d d d g ƒ ƒ } t j | | ƒ d  S(   Ni   i   i   Rm   i    (   R   R   R4   R   R<   R   R   R   (   R!   R7   R0   R1   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_getitem_category_type  s    **'c            s"  ˆ  j  j d d g } ˆ  j  j d d d d g } t | | d t ƒˆ  j j d d d	 g } t t d
 ƒ d t d ƒ d d ƒ} t i d d d d d t	 j
 g d 6d | ƒ} t | | d t ƒt j t ‡  f d †  ƒ ˆ  j j ƒ  } d | j d	 <| j d d d	 g } t t d
 ƒ d t d ƒ d d ƒ} t i d d d d d d g d 6d | ƒ} t | | ƒ ˆ  j j ƒ  } | j d d d	 g } t t d
 ƒ d t d ƒ d d ƒ} t i d d d d d t	 j
 g d 6d | ƒ} t | | d t ƒt j t ƒ  ˆ  j j d d g Wd  QXd  S(   NR:   R$   i   i    i   i   t   check_index_typeR9   t   et   aaabbeR;   R   RF   R   i   i   R   R3   c              s   ˆ  j  j d S(   NRr   (   R   R'   (    (   R!   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   A  s    i   R&   (   R   R'   R4   R
   R   R   R   R   R   R   t   nanR,   R-   R.   R+   (   R!   R0   R1   t	   exp_indexR   (    (   R!   sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_loc_listlike3  s0    !.!+!.c         C   sÙ  t  d d d g ƒ } t i d d d g d 6d d	 d
 g d 6d | ƒ} | j d d g } t  d d g d | j ƒ} t i d d g d 6d d	 g d 6d | ƒ} t j | | d t ƒ| j d d d g } t  d d d g d | j ƒ} t i d d d g d 6d d d	 g d 6d | ƒ} t j | | d t ƒd } t j t	 d | ƒ | j d d g Wd  QXt  d d d g ƒ } t i d d d g d 6d d	 d
 g d 6d | ƒ} | j d d g } t i d d d g d 6d d
 d	 g d 6d t  d d d g ƒ ƒ} t j | | d t ƒ| j d d d g } t i d d d d d g d 6d d
 d d
 d	 g d 6d t  d d d d d g ƒ ƒ} t j | | d t ƒd } t j t	 d | ƒ | j d d g Wd  QXt  d d d d g d t
 d ƒ ƒ} t i d d d d g d 6d	 d
 d d g d 6d | ƒ} | j d d g } t i d d d g d 6d	 d d
 g d 6d t  d d d g d t
 d ƒ ƒƒ} t j | | d t ƒ| j d d g } t i d d t j g d 6d	 d t j g d 6d t  d d d g d t
 d ƒ ƒƒ} t j | | d t ƒ| j d d d g } t i d d d d d g d 6d	 d d	 d d
 g d 6d t  d d d d d g d t
 d ƒ ƒƒ} t j | | d t ƒd } t j t	 d | ƒ | j d d g Wd  QXd  S(   NR$   R9   R:   i   i   i   R   i   i   i   R   R3   R;   Rq   sB   a list-indexer must only include values that are in the categoriest   matcht   xt   abcdei   i   Rr   (   R   R   R'   R;   R<   R
   R   R,   R-   R.   R   R   Rt   (   R!   R3   R   t   resRu   RL   t   msg(    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_loc_listlike_dtypesW  sh    2,22!8)/5c         C   s›   t  j t d ƒ t d ƒ g d t ƒ} t d ƒ t d ƒ g } t | d | d t d d ƒ} | j | ƒ } t  j d d g d d	 ƒ} t j | | ƒ d  S(
   Ns   1999-12-31 00:00:00s   2000-12-31 00:00:00R   R;   R   Rm   i    i   t   intp(	   R   RG   R   RV   R   R   t   get_indexerR<   R=   (   R!   t   arrR6   t   ciR0   R1   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_get_indexer_array§  s    		c         C   st   t  d d g d d d g ƒ} | j t  d d g d d d g ƒƒ } t j d d g d d ƒ} t j | | ƒ d  S(   NR$   R9   R;   i   R   R}   (   R   R~   R   RG   R<   R=   (   R!   R€   R0   R1   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt+   test_get_indexer_same_categories_same_order³  s
    c         C   st   t  d d g d d d g ƒ} | j t  d d g d d d g ƒƒ } t j d d g d d ƒ} t j | | ƒ d  S(   NR$   R9   R;   i   R   R}   (   R   R~   R   RG   R<   R=   (   R!   R€   R0   R1   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt0   test_get_indexer_same_categories_different_order»  s
    c         C   s”   t  t d ƒ t d ƒ g ƒ } t d d g d d g g d d d d d g d | ƒ} t j | ƒ } | g  | j D] } | ^ qp } t | | ƒ d  S(	   Ns
   12-31-1999s
   12-31-2000i   i    R   t   uint8R3   RZ   (   R   R   R   RI   t   get_dummiesRZ   R
   (   R!   R6   R1   t   dummiesR:   R0   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_getitem_with_listlikeÄ  s    ! c         C   s¼   t  j j d ƒ t t  j j d d d d ƒj t  j ƒ ƒ j d g ƒ } t  j d g ƒ j t  j	 ƒ } d | | <| j
 t  j d g ƒ j t  j	 ƒ } t j | t  j d g d d	 ƒƒ d  S(
   Ni   i    i   t   sizeiðI iüÿÿi † R   t   int8(   R   t   randomt   seedR   t   randintR   R‰   t   add_categoriesRG   R   t   codesR<   R=   (   R!   R:   t   indexerR0   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_setitem_listlikeÏ  s    
%c         C   s   t  t j j d d ƒ d t d ƒ d t d ƒ ƒ} | j ƒ  } t | j ƒ | _ t | j ƒ | _ t	 | j
 d d  d  … f d | j d d ƒ} t | j
 d d  d  … f | ƒ t	 | j
 d  d  … d f d | j d d ƒ} t | j
 d  d  … d f | ƒ t t d	 ƒ d
 d d d g ƒ} t  | j
 d d g d  d  … f d | j d | ƒ} t | j
 d d g d  d  … f | ƒ t t d ƒ d
 d d d g ƒ} t  | j
 d  d  … d d g f d | j d | ƒ} t | j
 d  d  … d d g f | ƒ t  t j j d d ƒ d t d ƒ d t d ƒ ƒ} | j ƒ  } t | j ƒ | _ t | j ƒ | _ t t d ƒ d
 d d g ƒ} t  | j
 d d  d  … f d | j d | ƒ} t | j
 d d  d  … f | ƒ t t d ƒ d
 d d g ƒ} t  | j
 d  d  … d f d | j d | ƒ} t | j
 d  d  … d f | ƒ t  | j
 d d g d  d  … f d | j d t t d ƒ ƒ ƒ} t | j
 d d g d  d  … f | ƒ t  | j
 d  d  … d d g f d | j d t t d ƒ ƒ ƒ} t | j
 d  d  … d d g f | ƒ d  S(   Ni   R3   t   ABCRZ   t   XYZR   RF   t   Xt   ABR;   R   R*   t   XYt   Yt   Zt   ABAt   XYXt   AAt   XXt   AABt   XXY(   R   R   RŠ   t   randnR   R+   R   R3   RZ   R   R'   R   R
   (   R!   R   t   cdft   expectRu   t   exp_columns(    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_ix_categorical_indexÞ  sJ    . . !+	&+	&%	 %	 +&+c         C   s^  t  j d ƒ } t | ƒ } t  j d ƒ } | j d t ƒ t | ƒ } t | j d d d g | j d d d g ƒ t | j d g | j d g ƒ t | j d | j d ƒ t | j d d !| j d d !ƒ t | j d d d g | j d d d g ƒ t | j d g | j d g ƒ t | j d | j d ƒ t | j d d !| j d d !ƒ d  S(   Ni
   t   writei   i   i   (	   R   t   eyeR   t   setflagsR   R
   R4   R   R'   (   R!   t   rw_arrayt   rw_dft   ro_arrayt   ro_df(    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_read_only_source  s    -!!-!c            sW  ˆ  j  j d d d g ƒ } t i d d d d d t j g d	 6t t d
 ƒ ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j d d g ƒ } t i d d d d d g d	 6t t d ƒ ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j d g ƒ } t i t j g d	 6t d g ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j d g ƒ } t i t j g d	 6t d g ƒ d 6ƒ j d ƒ } t | | d t	 ƒt d ƒ } ˆ  j  j t
 d d g d | ƒƒ } t i d d d t j g d	 6t t d ƒ ƒ j t | ƒ ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j t
 d g d | ƒƒ } t i d d d g d	 6t t d ƒ ƒ j t | ƒ ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j d d d g ƒ } t i d d d d d t j g d	 6t t d
 ƒ ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j d d g ƒ } t i d d d d d g d	 6t t d ƒ ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j d g ƒ } t i t j g d	 6t d g ƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j t
 d d g d | d t	 ƒƒ } t i d d d t j g d	 6t t d ƒ ƒ j t | d t	 ƒƒ d 6ƒ j d ƒ } t | | d t	 ƒˆ  j  j t
 d d g d d d g ƒƒ } t i d d d t j g d	 6t t d ƒ ƒ j t d d g ƒ ƒ d 6ƒ j d ƒ } t | | d t	 ƒt j t ‡  f d †  ƒ t j t ‡  f d †  ƒ t j t ‡  f d †  ƒ t j t ‡  f d †  ƒ d  S(   NR$   R9   Rr   i    i   i   i   i   R   Rs   R   Rq   t   aaabbR&   R   R;   t   aaadR%   R   c              s   ˆ  j  j d d g ƒ S(   NR$   (   R   t   reindex(    (   R!   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   k  s    c              s   ˆ  j  j d g d d ƒS(   NR$   R"   t   ffill(   R   R­   (    (   R!   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   o  s    c              s   ˆ  j  j d g d d ƒS(   NR$   t   leveli   (   R   R­   (    (   R!   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   q  s    c              s   ˆ  j  j d g d d ƒS(   NR$   t   limiti   (   R   R­   (    (   R!   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   s  s    (   R   R­   R   R   Rt   R   R   R   R
   R   R   R   R   R,   R-   t
   ValueErrort   NotImplementedError(   R!   R0   R1   R6   (    (   R!   sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_reindexing#  sv    """$!"""%%			c            s   t  j t ‡  f d †  ƒ d  S(   Nc              s   ˆ  j  j d d !S(   Ni   i   (   R   R'   (    (   R!   (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   z  s    (   R,   R-   R/   (   R!   (    (   R!   sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_loc_sliceu  s    c            sk  |  j  } |  j ‰  | | j d k } | j g  } t | | ƒ ˆ  ˆ  j d k } ˆ  j g  } t | | ƒ | | j d k } | j d d d g } t | | ƒ ˆ  ˆ  j d k } ˆ  j d d d g } t | | ƒ | | j d k  } | j d g } t | | ƒ | | j d k } | j g  } t | | ƒ t j t ‡  f d †  ƒ t j t ‡  f d †  ƒ d  S(	   NR$   i   i    i   i   i   c              s   ˆ  ˆ  j  d k  S(   Ni   (   R3   (    (   R    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   ª  s    c              s   ˆ  ˆ  j  d k S(   Ni   (   R3   (    (   R    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR(   «  s    (   R   R    R3   R4   R
   R,   R-   R/   (   R!   R   R0   R1   (    (   R    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_boolean_selection€  s,    			c         C   s™   t  i d d d g d 6ƒ } t  i t t t g d 6ƒ } | d g d k } t j | | ƒ | d j d ƒ | d <| d g d k } t j | | ƒ d  S(   Nt   foot   bart   bazR   Rm   (   R   R   R   R<   R
   R   (   R!   R@   RL   Rz   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_indexing_with_category­  s    c         C   sÝ   d d d d g } d d d d g } t  j | d d ƒ} t  j | d d d	 d d d g ƒ} t  j | d
  d | d
  ƒ} | j | ƒ } t j | | ƒ d „  t | d
  | d
  ƒ Dƒ } | j | ƒ } t j | | ƒ d  S(   NR$   R   i   t   onei   g      @RF   t   XXXR;   iÿÿÿÿR3   c         S   s   i  |  ] \ } } | | “ q S(    (    (   t   .0t   oRT   (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pys
   <dictcomp>É  s   	 (   RI   R   R   t   mapR<   R?   t   zip(   R!   t   orig_valuest
   new_valuest	   cur_indexR1   t   mappert   output(    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   test_map_with_dict_or_series½  s    (   t   __name__t
   __module__R#   R2   R8   RB   RM   Rl   Ro   Rp   Rv   R|   R   R‚   Rƒ   R‡   R   R¢   Rª   R³   R´   Rµ   R¹   RÅ   (    (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyR      s,   		"					‹			$	P							2		R		-	(   t   numpyR   R,   t   pandas.compatRX   t   pandas.core.dtypes.commonR    t   pandas.core.dtypes.dtypesR   t   pandasRI   R   R   R   R   R   R   R   t   pandas.api.typesR   t   pandas.utilR	   R<   t   pandas.util.testingR
   R   RV   R   (    (    (    sE   lib/python2.7/site-packages/pandas/tests/indexing/test_categorical.pyt   <module>   s   4