ó
‡ˆ\c           @   sÀ  d  d l  Z  d  d l Z d  d l Z d  d l Z d  d l Z d  d l m Z d  d l m Z d  d l m	 Z	 d  d l
 Z
 d  d l m Z d  d l m Z d  d l m Z d  d l m Z d  d	 l m Z d  d
 l m Z d  d l m Z d  d l m Z d  d l m Z d  d l m Z d  d l m Z d  d l m Z d  d l m Z d  d l m Z d  d l m Z m Z d  d l m  Z  d  d l! m" Z" d  d l! m# Z# d  d l! m$ Z$ d „  Z% e
 j& d d ƒ d „  ƒ Z' e
 j& d d ƒ d „  ƒ Z( e
 j& d „  ƒ Z) e
 j& d „  ƒ Z* d „  Z+ d  „  Z, d! „  Z- d" „  Z. d# „  Z/ d$ „  Z0 d% „  Z1 d& „  Z2 d' „  Z3 d( „  Z4 d) „  Z5 d* „  Z6 d+ „  Z7 d, „  Z8 d- „  Z9 d. „  Z: d/ „  Z; d0 „  Z< d1 „  Z= d S(2   iÿÿÿÿN(   t   loads(   t   dumps(   t   partial(   t   get_data_home(   t   clear_data_home(   t
   load_files(   t   load_sample_images(   t   load_sample_image(   t   load_digits(   t   load_diabetes(   t   load_linnerud(   t	   load_iris(   t   load_breast_cancer(   t   load_boston(   t	   load_wine(   t   Bunch(   t   check_return_X_y(   t   bt   u(   t   pillow_installed(   t   assert_false(   t   assert_equal(   t   assert_raisesc         C   s&   t  j j |  ƒ r" t j |  ƒ n  d  S(   N(   t   ost   patht   isdirt   shutilt   rmtree(   R   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   _remove_dir#   s    t   scopet   modulec         c   s(   t  |  j d ƒ ƒ } | Vt | ƒ d  S(   Nt   scikit_learn_data_home_test(   t   strt   mktempR   (   t   tmpdir_factoryt   tmp_file(    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt	   data_home(   s    c         c   s(   t  |  j d ƒ ƒ } | Vt | ƒ d  S(   Nt   scikit_learn_load_files_test(   R    R!   R   (   R"   R#   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   load_files_root/   s    c         c   s`   t  j d |  ƒ } t  j d | d t ƒ } | j t d ƒ ƒ | j ƒ  t | ƒ Vt | ƒ d  S(   Nt   dirt   deletes   Hello World!
(	   t   tempfilet   mkdtempt   NamedTemporaryFilet   Falset   writeR   t   closeR    R   (   R&   t   test_category_dir1t   sample_file(    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_category_dir_16   s    	
c         c   s+   t  j d |  ƒ } t | ƒ Vt | ƒ d  S(   NR'   (   R)   R*   R    R   (   R&   t   test_category_dir2(    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_category_dir_2A   s    c         C   s‚   t  d |  ƒ }  t |  |  ƒ t j j |  ƒ s4 t ‚ t d |  ƒ t t j j |  ƒ ƒ t  d |  ƒ }  t j j |  ƒ s~ t ‚ d  S(   NR$   (   R   R   R   R   t   existst   AssertionErrorR   R   (   R$   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_data_homeH   s    c         C   sL   t  |  ƒ } t t | j ƒ d ƒ t t | j ƒ d ƒ t | j d  ƒ d  S(   Ni    (   R   R   t   lent	   filenamest   target_namest   DESCRt   None(   R&   t   res(    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_default_empty_load_filesW   s    c         C   se   t  | ƒ } t t | j ƒ d ƒ t t | j ƒ d ƒ t | j d  ƒ t | j t d ƒ g ƒ d  S(   Ni   i   s   Hello World!
(	   R   R   R7   R8   R9   R:   R;   t   dataR   (   R1   R3   R&   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_default_load_files^   s
    c         C   s˜   t  j j |  ƒ j d ƒ j ƒ  } t | d d d | d d ƒ} t t | j ƒ d ƒ t t | j	 ƒ d ƒ t | j
 d ƒ t | j t d ƒ g ƒ d  S(	   Nt   /t   descriptiont   testt
   categoriest   encodings   utf-8i   s   Hello World!
(   R   R   t   abspatht   splitt   popR   R   R7   R8   R9   R:   R>   R   (   R1   R3   R&   t   categoryR<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt.   test_load_files_w_categories_desc_and_encodingg   s    !c         C   sh   t  | d t ƒ} t t | j ƒ d ƒ t t | j ƒ d ƒ t | j d  ƒ t | j d ƒ d  ƒ d  S(   Nt   load_contenti   i   R>   (	   R   R,   R   R7   R8   R9   R:   R;   t   get(   R1   R3   R&   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_files_wo_load_contentr   s
    c          C   sm   yH t  ƒ  }  t t |  j ƒ d ƒ t t |  j ƒ d ƒ |  j sG t ‚ Wn t k
 rh t j	 d ƒ n Xd  S(   Ni   s3   Could not load sample images, PIL is not available.(
   R   R   R7   t   imagesR8   R:   R5   t   ImportErrort   warningst   warn(   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_sample_images{   s    	c          C   sO   t  ƒ  }  t |  j j d ƒ t t j |  j ƒ j d ƒ t |  t	 t  ƒ ƒ d  S(   Ni  i@   i
   (   i  i@   (
   R   R   R>   t   shapet   numpyt   uniquet   targett   sizeR   R   (   t   digits(    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_digits…   s    	c          C   s?   t  d ƒ }  t |  j j d ƒ t t j |  j ƒ j d ƒ d  S(   Ni	   iQ  i@   (   iQ  i@   (   R   R   R>   RR   RS   RT   RU   RV   (   RW   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_digits_n_class_lt_10Ž   s    c          C   sU   y0 t  d ƒ }  t |  j d ƒ t |  j d ƒ Wn t k
 rP t j d ƒ n Xd  S(   Ns	   china.jpgt   uint8i«  i€  i   s3   Could not load sample images, PIL is not available.(   i«  i€  i   (   R   R   t   dtypeRR   RN   RO   RP   (   t   china(    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_sample_image”   s    c           C   s*   t  r t t t d ƒ n t j d ƒ d  S(   Ns   blop.jpgs3   Could not load sample images, PIL is not available.(   R   R   t   AttributeErrorR   RO   RP   (    (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt$   test_load_missing_sample_image_error   s    	
c          C   sp   t  ƒ  }  t |  j j d ƒ |  j j s4 t d ƒ ‚ t t |  j ƒ d ƒ |  j	 sY t ‚ t
 |  t t  ƒ ƒ d  S(   Niº  i
   (   iº  i
   (   R	   R   R>   RR   RU   RV   R5   R7   t   feature_namesR:   R   R   (   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_diabetes¥   s    	c          C   s¡   t  ƒ  }  t |  j j d ƒ t |  j j d ƒ t t |  j ƒ d ƒ |  j sT t ‚ t	 j
 j |  j ƒ so t ‚ t	 j
 j |  j ƒ sŠ t ‚ t |  t t  ƒ ƒ d  S(   Ni   i   (   i   i   (   i   i   (   R
   R   R>   RR   RU   R7   R9   R:   R5   R   R   R4   t   data_filenamet   target_filenameR   R   (   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_linnerud°   s    	c          C   sƒ   t  ƒ  }  t |  j j d ƒ t |  j j d ƒ t |  j j d ƒ |  j sQ t ‚ t	 j
 j |  j ƒ sl t ‚ t |  t t  ƒ ƒ d  S(   Ni–   i   i   (   i–   i   (   R   R   R>   RR   RU   RV   R9   R:   R5   R   R   R4   t   filenameR   R   (   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_iris½   s    	c          C   sh   t  ƒ  }  t |  j j d ƒ t |  j j d ƒ t |  j j d ƒ |  j sQ t ‚ t	 |  t
 t  ƒ ƒ d  S(   Ni²   i   i   (   i²   i   (   R   R   R>   RR   RU   RV   R9   R:   R5   R   R   (   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_wineÉ   s    	c          C   sƒ   t  ƒ  }  t |  j j d ƒ t |  j j d ƒ t |  j j d ƒ |  j sQ t ‚ t	 j
 j |  j ƒ sl t ‚ t |  t t  ƒ ƒ d  S(   Ni9  i   i   (   i9  i   (   R   R   R>   RR   RU   RV   R9   R:   R5   R   R   R4   Re   R   R   (   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_breast_cancerÔ   s    	c          C   sƒ   t  ƒ  }  t |  j j d ƒ t |  j j d ƒ t |  j j d ƒ |  j sQ t ‚ t	 j
 j |  j ƒ sl t ‚ t |  t t  ƒ ƒ d  S(   Niú  i   (   iú  i   (   R   R   R>   RR   RU   RV   R`   R:   R5   R   R   R4   Re   R   R   (   R<   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_load_bostonà   s    	c          C   sB   t  d d ƒ }  t t |  ƒ ƒ } d | _ t | d | j ƒ d  S(   Nt   xt   y(   R   R    R   Rj   R   (   t   buncht   bunch_from_pkl(    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_loads_dumps_bunchì   s    	c          C   s}   t  d d ƒ }  d |  j d <t t |  ƒ ƒ } t | j d ƒ t | d d ƒ d | _ t | j d ƒ t | d d ƒ d  S(   Nt   keyt   originals   set from __dict__t   changed(   R   t   __dict__R    R   R   Ro   (   Rl   Rm   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt8   test_bunch_pickle_generated_with_0_16_and_read_with_0_17ó   s    	c          C   s%   t  ƒ  }  d t |  ƒ k s! t ‚ d  S(   NR>   (   R   R'   R5   (   R>   (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   test_bunch_dir  s    	(>   R   R   R)   RO   RS   t   pickleR    R   t	   functoolsR   t   pytestt   sklearn.datasetsR   R   R   R   R   R   R	   R
   R   R   R   R   t   sklearn.datasets.baseR   t"   sklearn.datasets.tests.test_commonR   t   sklearn.externals.sixR   R   t   sklearn.externals._pilutilR   t   sklearn.utils.testingR   R   R   R   t   fixtureR$   R&   R1   R3   R6   R=   R?   RI   RL   RQ   RX   RY   R]   R_   Ra   Rd   Rf   Rg   Rh   Ri   Rn   Rs   Rt   (    (    (    s?   lib/python2.7/site-packages/sklearn/datasets/tests/test_base.pyt   <module>   sf   									
														