ó
—W[c           @   sX  d  d l  Z d  d l Z d  d l j Z d  d l Z d  d l	 j
 Z d  d l j Z y d  d l j Z Wn# e k
 r‰ d  d l j j Z n Xd  d l m Z d  d l m Z y d  d l j j Z e Z Wn e k
 rß e Z n Xd d l m Z  d d l! m" Z" e j# j$ d ƒ Z% d e& f d	 „  ƒ  YZ' d
 e& f d „  ƒ  YZ( d e& f d „  ƒ  YZ) d S(   iÿÿÿÿN(   t   skipif(   t   SkipTesti   (   t
   regression(   t   color_palettei    t   TestLinearPlotterc           B   s  e  Z e j j d  ƒ Z e j e d e j	 d d ƒ d e j
 d d d ƒ d e j d d d ƒd	 e j e d
 ƒ d ƒ ƒ ƒ Z e j e j d ƒ e d <e j j ƒ  e d <e j e j d d d g d f <d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   iM   t   xt   sizei<   t   diþÿÿÿi   t   yi   t   st
   abcdefghiji   t   zt   y_nai
   i   i   c         C   st   t  j ƒ  } | j |  j d d d d ƒt j | j |  j j ƒ t j | j |  j j ƒ t j | j	 |  j ƒ d  S(   NR   R   (
   t   lmt   _LinearPlottert   establish_variablest   dft   pdtt   assert_series_equalR   R   t   assert_frame_equalt   data(   t   selft   p(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt#   test_establish_variables_from_frame&   s
    c         C   sz   t  j ƒ  } | j d  d |  j j d |  j j ƒt j | j |  j j ƒ t j | j |  j j ƒ t	 j
 | j d  ƒ d  S(   NR   R   (   R   R   R   t   NoneR   R   R   R   R   t   ntt	   assert_isR   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt$   test_establish_variables_from_series.   s
    %c         C   s€   t  j ƒ  } | j d  d |  j j j d |  j j j ƒt j	 | j |  j j ƒ t j	 | j |  j j ƒ t
 j | j d  ƒ d  S(   NR   R   (   R   R   R   R   R   R   t   valuesR   t   nptt   assert_array_equalR   R   R   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt#   test_establish_variables_from_array6   s    c         C   sŒ   t  j ƒ  } | j d  d |  j j j j ƒ  d |  j j j j ƒ  ƒt	 j
 | j |  j j ƒ t	 j
 | j |  j j ƒ t j | j d  ƒ d  S(   NR   R   (   R   R   R   R   R   R   R   t   tolistR   R   R   R   R   R   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt#   test_establish_variables_from_lists@   s    c         C   sz   t  j ƒ  } | j |  j d d d |  j j ƒt j | j |  j j ƒ t j | j |  j j ƒ t j | j	 |  j ƒ d  S(   NR   R   (
   R   R   R   R   R   R   R   R   R   R   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt!   test_establish_variables_from_mixJ   s
    "c      
   C   sE   t  j ƒ  } t j t ƒ $ | j d  d d d |  j j ƒWd  QXd  S(   NR   R   (	   R   R   R   t   assert_raisest
   ValueErrorR   R   R   R   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt!   test_establish_variables_from_badR   s    c         C   sº   t  j ƒ  } | j |  j d d d d ƒt j | j |  j j ƒ t j | j |  j j ƒ | j d d ƒ |  j j j	 ƒ  } t j | j |  j j | ƒ t j | j |  j j | ƒ d  S(   NR   R   (
   R   R   R   R   R   R   R   R   t   dropnat   notnull(   R   R   t   mask(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_dropnaX   s    (   t   __name__t
   __module__t   npt   randomt   RandomStatet   rst   pdt	   DataFramet   dictt   normalt   randintt   gammat   tilet   listR   R   t   randnt   copyt   nant   locR   R   R   R!   R"   R%   R)   (    (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyR      s   			
	
		t   TestRegressionPlotterc           B   s»  e  Z e j j d  ƒ Z e j d d d ƒ Z d Z d Z	 d d d g Z
 e j e d e j d	 d
 ƒ d e j d d d
 ƒ d e j d d	 d
 ƒd e j e e d ƒ ƒ d ƒ ƒ ƒ Z e j e j d
 ƒ e d <e j j ƒ  e d <e j d ƒ e j j d Z e j e 7_ d d e j e j d e j d
 ƒ ƒ Z g  e D] Z e j d e ƒ ^ q@e d <e j  e j! d d d g d f <d „  Z" d „  Z# d „  Z$ d „  Z% d „  Z& e' e( ƒ d „  ƒ Z) e' e( ƒ d „  ƒ Z* d „  Z+ e' e( ƒ d „  ƒ Z, e' e( ƒ d  „  ƒ Z- d! „  Z. d" „  Z/ d# „  Z0 d$ „  Z1 d% „  Z2 d& „  Z3 d' „  Z4 d( „  Z5 e' e( ƒ d) „  ƒ Z6 e' e( ƒ d* „  ƒ Z7 e' e( ƒ d+ „  ƒ Z8 e' e( ƒ d, „  ƒ Z9 d- „  Z: d. „  Z; RS(/   i1   iýÿÿÿi   i   id   iÿÿÿÿi    i   R   R   i<   R   iþÿÿÿR   i   R	   i   i
   R   R   i   t   ci   c         C   s†   t  j d d d |  j d d ƒ} t j | j |  j j ƒ t j | j |  j j ƒ t j | j |  j j ƒ t j	 | j
 |  j ƒ d  S(   NR   R   R   t   unitsR	   (   R   t   _RegressionPlotterR   R   R   R   R   R>   R	   R   R   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_variables_from_frame|   s
    !c         C   sŒ   t  j |  j j |  j j d |  j j ƒ} t j | j |  j j ƒ t j | j |  j j ƒ t j | j |  j j ƒ t	 j
 | j d  ƒ d  S(   NR>   (   R   R?   R   R   R   R	   R   R   R>   R   R   R   R   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_variables_from_series…   s
    *c         C   su   t  j d |  j j d d |  j ƒ} t j | j |  j j ƒ t j | j |  j j d ƒ t j | j	 |  j ƒ d  S(   NR   i   R   (
   R   R?   R   R   R   R   R   R   R   R   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_variables_from_mixŽ   s    %c         C   s“   t  j d d d |  j ƒ} t j t | j ƒ t j |  j j	 ƒ j
 ƒ  ƒ t  j d d d |  j d t ƒ} t j t | j ƒ t |  j j	 ƒ ƒ d  S(   NR   R   R   R&   (   R   R?   R   R   t   assert_equalt   lenR   R0   R'   R   t   sumt   False(   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyR)   –   s    .!c      	   C   så   t  j d d d |  j d d ƒ} t j | j d ƒ t j | j d ƒ t  j d d d |  j d d d d ƒ} t j | j d ƒ t j | j d ƒ t  j d d d |  j d d d d ƒ} t j | j d ƒ t j | j d ƒ d  S(	   NR   R   R   t   cii_   t   x_ciiD   t   sd(   R   R?   R   R   RC   RG   RH   (   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_ciž   s    !''c         C   sn   t  j d d d |  j d |  j ƒ} | j |  j ƒ \ } } | j |  j t j ƒ \ } } t	 j
 | | ƒ d  S(   NR   R   R   t   n_boot(   R   R?   R   RK   t   fit_fastt   gridt   fit_statsmodelst   smlmt   OLSR   t   assert_array_almost_equal(   R   R   t	   yhat_fastt   _t	   yhat_smod(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_fast_regression¬   s    $c         C   sq   t  j d d d |  j d |  j ƒ} | j |  j d ƒ \ } } | j |  j t j ƒ \ } } t	 j
 | | ƒ d  S(   NR   R   R   RK   i   (   R   R?   R   RK   t   fit_polyRM   RN   RO   RP   R   RQ   (   R   R   t	   yhat_polyRS   RT   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regress_polyº   s    $c         C   sÊ   t  j d d ƒ } t  j d d ƒ } t  j d d d ƒ } t j | | d |  j ƒ} | j | ƒ \ } } | j | ƒ \ } } t j	 | d | d ƒ t j	 | d | d ƒ t j	 | d | d ƒ d  S(   Ni   i
   id   RK   i    i   iZ   (
   R,   t   aranget   linspaceR   R?   RK   RL   t   fit_logxR   t   assert_greater(   R   R   R   RM   R   t   yhat_linRS   t   yhat_log(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regress_logxÈ   s    c         C   sß   t  j d d d |  j d |  j ƒ} | j |  j ƒ \ } } t j | j |  j |  j j	 f ƒ | j
 |  j d ƒ \ } } t j | j |  j |  j j	 f ƒ | j |  j t j ƒ \ } } t j | j |  j |  j j	 f ƒ d  S(   NR   R   R   RK   i   (   R   R?   R   RK   RL   RM   R   RC   t   shapeR   RV   RN   RO   RP   (   R   R   RS   t
   boots_fastt
   boots_polyt
   boots_smod(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regress_n_bootÖ   s    $""c      	   C   s¯   t  j d d d |  j d |  j d d  ƒ} | j |  j ƒ \ } } t j | d  ƒ | j	 |  j d ƒ \ } } t j | d  ƒ | j
 |  j t j ƒ \ } } t j | d  ƒ d  S(   NR   R   R   RK   RG   i   (   R   R?   R   RK   R   RL   RM   R   R   RV   RN   RO   RP   (   R   R   RS   Ra   Rb   Rc   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regress_without_bootstrapç   s    c         C   sl   t  j |  j j |  j j ƒ } | j |  j ƒ \ } } t j t	 | ƒ |  j ƒ t j
 t j | ƒ | ƒ d  S(   N(   R   R?   R   R   R   t   bin_predictort   bins_numericR   RC   RD   R   R,   t   unique(   R   R   t   x_binnedt   bins(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_numeric_binsù   s    c         C   sV   t  j |  j j |  j j ƒ } | j |  j ƒ \ } } t j t	 j
 | ƒ |  j ƒ d  S(   N(   R   R?   R   R   R   Rf   t
   bins_givenR   R   R,   Rh   (   R   R   Ri   Rj   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_provided_bins   s    c         C   s²   t  j |  j j |  j j ƒ } | j |  j ƒ \ } } t j |  j j | d k j	 ƒ  |  j j | d k j
 ƒ  ƒ t j |  j j | d k j	 ƒ  |  j j | d k j
 ƒ  ƒ d  S(   Ni    iÿÿÿÿi   (   R   R?   R   R   R   Rf   Rl   R   R\   t   mint   max(   R   R   Ri   Rj   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_bin_results  s    c         C   sÑ  t  j |  j j |  j j ƒ } | j \ } } t j | |  j j ƒ t j | |  j j ƒ t  j |  j j |  j j ƒ } | j \ } } t j | |  j j ƒ t j | |  j j ƒ t  j |  j j |  j j d d ƒ} | j \ } } t	 j
 | |  j j k j ƒ  ƒ t j t j |  j j | ƒ t j d t | ƒ ƒ ƒ t j | |  j j ƒ t  j |  j j |  j j d d ƒ} | j \ } } t j | |  j j ƒ t j t j |  j j | ƒ t j d t | ƒ ƒ ƒ d  S(   Nt   x_jittergš™™™™™¹?t   y_jittergš™™™™™©?(   R   R?   R   R   R   t   scatter_dataR   R   R   R   t   assert_truet   anyt   assert_array_lessR,   t   abst   repeatRD   (   R   R   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_scatter_data  s"    $5$c         C   sÜ   t  j |  j j |  j j d t j ƒ} | j \ } } } t j	 | t j
 t j |  j j ƒ ƒ ƒ t j | |  j j d ƒ j j ƒ  ƒ t j t j | ƒ d  d  … d f | ƒ t j | t j | ƒ d  d  … d f ƒ d  S(   Nt   x_estimatorR   i    i   (   R   R?   R   R   R   R,   t   meant   estimate_dataR   R   t   sortRh   RQ   t   groupbyRv   t   array(   R   R   R   R   RG   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_estimate_data&  s    '(%)c         C   s  t  j j d ƒ t j |  j j |  j j d t  j d d ƒ} | j	 \ } } } t j |  j j |  j j d t  j d d ƒ} | j	 \ } } } t
 j t  j | ƒ t  j | ƒ ƒ t j |  j j |  j j d t  j d d  ƒ} | j	 \ } } } t
 j | d  g t | ƒ ƒ d  S(   Ni{   Rz   RG   i_   i2   (   R,   R-   t   seedR   R?   R   R   R   R{   R|   R   Rv   t   diffR   R   RD   (   R   R   RS   t   ci_bigt   ci_weet   ci_nil(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_estimate_cis1  s    "c      	   C   sº   t  j j d ƒ t j d d d |  j d d d d ƒ} | j \ } } } t  j | d	 d
 ƒ} t j d d d |  j d d ƒ} | j \ } } } t  j | d	 d
 ƒ} t j	 | | ƒ d  S(   NiY  R   R   R   R>   R	   t   x_binsi   t   axisi   (
   R,   R-   R   R   R?   R   R|   R‚   R   Rv   (   R   R   RS   Rƒ   R„   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_estimate_unitsD  s    !c   	      C   s  |  j  j d ƒ } | |  j  j d ƒ } | |  j  j d ƒ } t j | | ƒ } t j | j | j ƒ d \ } } t j | | d | ƒ} t j | j | j ƒ d \ } } t j	 | | ƒ t j | | d | d | ƒ} t j | j | j ƒ d \ } } t j	 | | ƒ d  S(   Nid   i    t	   y_partialt	   x_partial(
   R/   R8   R   R?   R,   t   corrcoefR   R   R   t   assert_less(	   R   R   R   R   R   RS   t   r_origt   r_semipartialt	   r_partial(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_partialT  s    """c      	   C   si   t  j d d d |  j d t d |  j ƒ} | j d d ƒ \ } } } t j | d	 ƒ t j d
 | ƒ d  S(   NR   R=   R   t   logisticRK   t   x_rangeiýÿÿÿi   i   i    (   iýÿÿÿi   (   R   R?   R   t   TrueRK   t   fit_regressionR   Rv   (   R   R   RS   t   yhat(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_logistic_regressione  s
    c      	   C   s™   |  j  j |  j  j j ƒ  k } t j d | d |  j  d t d d ƒ} t j d d ƒ   | j d d ƒ \ } } } Wd  QXt	 j
 t j | ƒ j ƒ  ƒ d  S(   NR   R   R’   RK   i
   t   allt   ignoreR“   iýÿÿÿi   (   iýÿÿÿi   (   R   R   R{   R   R?   R”   R,   t   errstateR•   R   Rt   t   isnanR˜   (   R   R   R   RS   R–   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt    test_logistic_perfect_separationn  s    !c      	   C   s¤   t  j d d d |  j d |  j ƒ} | j d d	 ƒ \ } } } t  j d d d |  j d t d |  j ƒ} | j d d
 ƒ \ } } } t j t | ƒ t | ƒ ƒ d  S(   NR   R   R   RK   R“   iýÿÿÿi   t   robust(   iýÿÿÿi   (   iýÿÿÿi   (	   R   R?   R   RK   R•   R”   R   RC   RD   (   R   t   p_olsRS   t   ols_yhatt   p_robustt   robust_yhat(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_robust_regressionx  s    c         C   sl   t  j d d d |  j d t ƒ} | j d d ƒ \ } } } t j t | ƒ t | ƒ ƒ t j | d  ƒ d  S(	   NR   R   R   t   lowessR“   iýÿÿÿi   (   iýÿÿÿi   (
   R   R?   R   R”   R•   R   RC   RD   R   R   (   R   R   RM   R–   t	   err_bands(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lowess_regression…  s    !c         C   sz   t  j t ƒ * t j d d d |  j d t d d ƒWd  QXt  j t ƒ * t j d d d |  j d t d t ƒWd  QXd  S(   NR   R   R   R£   t   orderi   R’   (   R   R#   R$   R   R?   R   R”   (   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regression_optionsŽ  s    c         C   s  t  j ƒ  \ } } | j |  j j |  j j ƒ t j d d d |  j ƒ} | j | ƒ \ } } } | j	 ƒ  } t
 j | j ƒ  | d ƒ t
 j | j ƒ  | d ƒ t j d d d |  j d t ƒ} | j ƒ  \ } } } t
 j | j ƒ  |  j j j ƒ  ƒ t
 j | j ƒ  |  j j j ƒ  ƒ d  S(   NR   R   R   i    i   t   truncate(   t   pltt   subplotst   scatterR   R   R   R   R?   R•   t   get_xlimR   RC   Rn   Ro   R”   (   R   t   ft   axR   RM   RS   t   xlim(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regression_limits˜  s    !"(<   R*   R+   R,   R-   R.   R/   RZ   RM   RK   Rg   Rl   R0   R1   R2   R3   R4   R5   R6   R7   t   rangeR   R   R8   R9   R	   R   t   bw_errt   expR   R   t   p_it   binomialR:   R;   R@   RA   RB   R)   RJ   R    t   _no_statsmodelsRU   RX   R_   Rd   Re   Rk   Rm   Rp   Ry   R€   R†   R‰   R‘   R—   Rœ   R¢   R¥   R§   R°   (    (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyR<   e   sP   $,)																		
		
t   TestRegressionPlotsc           B   s`  e  Z e j j d  ƒ Z e j e d e j	 d ƒ d e j	 d ƒ d d e j
 d d d ƒ d e j e d	 ƒ d
 ƒ d e j e d ƒ d ƒ d e j e j d ƒ d ƒ ƒ ƒ Z e j	 d ƒ e j j Z e j e 7_ d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z d „  Z e  e! ƒ d „  ƒ Z" d „  Z# RS(   i8   R   iZ   R   i   R   i    i   t   gt   abci   t   ht   xyi-   t   ui   i   c         C   s¦   t  j ƒ  \ } } t j d d |  j ƒ t j t | j ƒ d ƒ t j t | j	 ƒ d ƒ | j	 d j
 ƒ  j \ } } t j | |  j j ƒ t j | |  j j ƒ d  S(   NR   R   i   i   i    (   R©   Rª   R   t   regplotR   R   RC   RD   t   linest   collectionst   get_offsetst   TR   R   R   R   (   R   R­   R®   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regplot_basic´  s    c         C   sN  t  j ƒ  \ } } t j d d |  j d t d | ƒ} t j t | j	 ƒ d ƒ t j t | j
 ƒ d ƒ | j ƒ  t  j ƒ  \ } } t j d d |  j d t ƒ} t j t | j	 ƒ d ƒ t j t | j
 ƒ d ƒ | j ƒ  t  j ƒ  \ } } t j d d |  j d d  ƒ} t j t | j	 ƒ d ƒ t j t | j
 ƒ d ƒ | j ƒ  d  S(	   NR   R   R«   R®   i   t   fit_regi    RG   (   R©   Rª   R   R½   R   RF   R   RC   RD   R¾   R¿   t   clearR   (   R   R­   R®   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regplot_selective¿  s    $

c         C   sÇ  t  j ƒ  \ } } t j d d d d g g ƒ } t j d d |  j d i | d 6ƒ} t j | j	 d j
 d  ƒ t j | j	 d j d d ƒ t  j ƒ  \ } } t j d d d g g ƒ } t j d d |  j d i | d 6ƒ} t j | j	 d j
 d ƒ t  j ƒ  \ } } t j d d d g g ƒ } t j d d |  j d i | d 6d
 d 6ƒ} t j | j	 d j
 d
 ƒ t  j ƒ  \ } } d } t j d d |  j d i | d 6ƒ} t j | j	 d j
 d ƒ d  S(   Ng333333Ó?gš™™™™™é?g      à?R   R   t   scatter_kwst   colori    i   gš™™™™™Ù?t   alphat   r(   i    i   (   R©   Rª   R,   R   R   R½   R   R   R   R¿   t   _alphaR   RC   t   _facecolors(   R   R­   R®   RÇ   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regplot_scatter_kws_alphaÓ  s$    %%%c         C   sT   t  j d d |  j d d ƒ} t j t | j ƒ d ƒ t j t | j ƒ d ƒ d  S(   NR   R   R‡   i   i   i   (   R   R½   R   R   RC   RD   R¾   R¿   (   R   R®   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_regplot_binnedë  s    c         C   s£   t  j d d |  j ƒ } | j d } t j t | j ƒ d ƒ t j t | j ƒ d ƒ | j d j	 ƒ  j
 \ } } t j | |  j j ƒ t j | |  j j ƒ d  S(   NR   R   i    i   i   (   i    i    (   R   t   lmplotR   t   axesR   RC   RD   R¾   R¿   RÀ   RÁ   R   R   R   R   (   R   R¸   R®   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lmplot_basicñ  s    c         C   sd   t  j d d d |  j d d ƒ} | j d	 } t j t | j ƒ d ƒ t j t | j ƒ d ƒ d  S(
   NR   R   R   t   hueRº   i    i   i   (   i    i    (	   R   RÎ   R   RÏ   R   RC   RD   R¾   R¿   (   R   R¸   R®   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lmplot_hueü  s    !c         C   sÜ   t  j d d d |  j d d d d ƒ} t j | j i d d g d 6ƒ t  j d d d |  j d d d d	 d g ƒ} t j | j i d	 d g d 6ƒ t j t ƒ 3 t  j d d d |  j d d d d	 d d
 g ƒWd  QXd  S(   NR   R   R   RÑ   Rº   t   markersR	   t   markert   oR   (   R   RÎ   R   R   RC   t   hue_kwsR#   R$   (   R   t   g1t   g2(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lmplot_markers  s    ' - c         C   s„   t  j d k r t ‚ n  t j d d d |  j d d d t d d	 d
 g ƒ} | j d j } t	 j
 | d j ƒ  d t  j d ƒ d  S(   Ns   1.4.2R   R   R   RÑ   Rº   RÃ   RÓ   RÕ   t   +i    i   s   lines.linewidth(   i    i    (   t   mplt   __version__R   R   RÎ   R   RF   RÏ   R¿   R   RC   t   get_linewidthst   rcParams(   R   R¸   R=   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lmplot_marker_linewidths  s    	c      	   C   s»   t  j d d d |  j d d d d ƒ} t j | j j d ƒ t  j d d d |  j d d
 d d ƒ} t j | j j d ƒ t  j d d d |  j d d d d
 ƒ} t j | j j d ƒ d  S(   NR   R   R   t   rowR¸   t   colRº   i   i   R¼   t   col_wrapi   i   RÑ   i   (   i   i   (   i   (   i   i   (   R   RÎ   R   R   RC   RÏ   R`   (   R   R¸   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lmplot_facets  s    '''c      	   C   s>   t  j d d d |  j d d d d ƒ} t j | j d  ƒ d  S(   NR   R   R   Rá   Rº   RÑ   (   R   RÎ   R   R   R   t   _legendR   (   R   R¸   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lmplot_hue_col_nolegend%  s    'c      	   C   s¢   t  j d d d d d |  j d d  ƒ} | j d j \ } } t d d	 ƒ \ } } t j | | j	 ƒ  d d  d
 … f ƒ t j | | j	 ƒ  d d  d
 … f ƒ d  S(   NR   R   RÑ   Rº   R   RG   i    t   n_colorsi   i   (   i    i    (
   R   RÎ   R   R   RÏ   R¿   R   R   R   t   get_facecolors(   R   R¸   t   red_scattert   blue_scattert   redt   blue(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_lmplot_scatter_kws*  s
    '&c         C   s   |  j  j |  j  j } } t j | | ƒ } | t j t j | | d ƒ | ƒ } | j d j	 ƒ  j
 \ } } t j | | ƒ t j | | ƒ d  S(   Ni   i    (   R   R   R   R   t	   residplotR,   t   polyvalt   polyfitR¿   RÀ   RÁ   R   R   RQ   (   R   R   R   R®   t   residt   x_plott   y_plot(    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_residplot3  s    %c         C   sv   t  j d d |  j d t ƒ} t j t | j ƒ d ƒ | j d j ƒ  j	 \ } } t
 j | t j |  j j ƒ ƒ d  S(   NR   R   R£   i   i   (   R   Rí   R   R”   R   RC   RD   R¾   t
   get_xydataRÁ   R   R   R,   R}   R   (   R   R®   R   R   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_residplot_lowess>  s    c         C   sj   t  j j d d ƒ \ } } t j | | d d ƒ} | j d j ƒ  } t j | d d  d … f d ƒ d  S(   Ni   i   RÇ   i   i    (   i   i    i    (   i   i    i    (	   R,   R-   R8   R   R½   R¿   Rç   R   t   assert_almost_equal(   R   R   R   R®   RÇ   (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   test_three_point_colorsG  s
    ($   R*   R+   R,   R-   R.   R/   R0   R1   R2   R8   R4   Rx   R7   R6   RY   R   R¼   R   R²   R   RÂ   RÅ   RÌ   RÍ   RÐ   RÒ   RÙ   Rß   Rã   Rå   Rì   Ró   R    R¶   Rõ   R÷   (    (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyR·   ¨  s.   !														(*   t   numpyR,   t
   matplotlibRÛ   t   matplotlib.pyplott   pyplotR©   t   pandasR0   t
   nose.toolst   toolsR   t   numpy.testingt   testingR   t   pandas.testingR   t   ImportErrort   pandas.util.testingt   utilt   numpy.testing.decoratorsR    t   noseR   t#   statsmodels.regression.linear_modelR   t   linear_modelRO   RF   R¶   R”   t    R   t   palettesR   R-   R.   R/   t   objectR   R<   R·   (    (    (    s<   lib/python2.7/site-packages/seaborn/tests/test_regression.pyt   <module>   s.   

Jÿ D