
&]\c           @` sJ  d  d l  m Z m Z m 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 m Z m Z m Z d  d l m Z m Z m Z m Z m Z m Z d Z d e f d	     YZ d
 e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ  d S(   i    (   t   divisiont   print_functiont   absolute_importN(   t   assert_(   t   FuncData(   t
   kolmogorovt   kolmogit   smirnovt   smirnovi(   t   _kolmogct	   _kolmogcit   _kolmogpt	   _smirnovct
   _smirnovcit	   _smirnovpg|=t   TestSmirnovc           B` sY   e  Z d    Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 RS(	   c         C` s#   t  t j t d t j    d  S(   Ni   (   R   t   npt   isnanR   t   nan(   t   self(    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_nan   s    c         C` s   d d d d d f d d d f g } t  j |  } t t | d d d	 t j d
 t t t g  d | d  d   d f | d  d   d f <t t | d d d	 t j d
 t t t g  d  S(   Ni   g?g?g      ?g      ?i   i   i    t   rtolt   dtypesi(   i   g?g?(   i   g      ?g      ?g      ?g      ?g      `?(   i    i   (   i    i   (	   R   t   asarrayR   R   t   _rtolt   checkt   intt   floatR   (   R   t   dataset(    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt
   test_basic   s    .*c         C` s   g  t  j t d d  t d d   D] } | d d f ^ q% } t j |  } t t | d
 d d t j d t	 t
 t
 g  d | d  d   d	 f | d  d   d	 f <t t | d d d t j d t	 t
 t
 g  d  S(   Ni   i   i  i  i    i   R   R   i(   i    i   (   i    i   (   t	   itertoolst   chaint   rangeR   R   R   R   R   R   R   R   R   (   R   t   nR   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_x_equals_0    s
    @.*c         C` s   g  t  j t d d  t d d   D] } | d d f ^ q% } t j |  } t t | d
 d d t j d t	 t
 t
 g  d | d  d   d	 f | d  d   d	 f <t t | d d d t j d t	 t
 t
 g  d  S(   Ni   i   i  i  i   i    R   R   i(   i    i   (   i    i   (   R   R   R    R   R   R   R   R   R   R   R   R   (   R   R!   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_x_equals_1'   s
    @.*c         C` s   d d d d d d d d  d! d" g
 } t  j |  } t t | d# d d t j d t t t g  d | d  d   d f | d  d   d f <t t | d$ d d t j d t t t g  d  S(%   Ni   g      ?i   g      ?i   g?UUUU?i   g      ?i   gy&1?i   gEʠ?i   g>?i   g!?i	   g+ˎ{?i
   g4o?i    R   R   i(   i   g      ?g      ?(   i   g      ?g      ?(   i   g      ?g?UUUU?(   i   g      ?g      ?(   i   g      ?gy&1?(   i   g      ?gEʠ?(   i   g      ?g>?(   i   g      ?g!?(   i	   g      ?g+ˎ{?(   i
   g      ?g4o?(   i    i   (   i    i   (	   R   R   R   R   R   R   R   R   R   (   R   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_x_equals_0point5.   s    	.*c         C` s   t  j d d d d t } t  j d g t |  | d | g  } t t | d	 d d t j d t	 t
 t
 g  d | d  d   d f | d  d   d f <t t | d
 d d t j d t	 t
 t
 g  d  S(   Ni    i   ie   t   endpointi   R   R   i(   i    i   (   i    i   (   R   t   linspacet   Truet   column_stackt   lenR   R   R   R   R   R   R   (   R   t   xR   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_n_equals_1?   s
    ).*c         C` s   t  j d d d d t } t  j d | d  } t  j d g t |   } t  j | | | g  } t t | d
 d d t	 j
 d t t t g  d | d  d   d	 f | d  d   d	 f <t t | d d d t	 j
 d t t t g  d  S(   Ng      ?i   ie   R%   i   i    R   R   i(   i    i   (   i    i   (   R   R&   R'   t   powert   arrayR)   R(   R   R   R   R   R   R   R   (   R   R*   t   pR!   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_n_equals_2F   s    .*c         C` s   t  j d d d d t } t  j d | d  } t  j d g t |   } t  j | | | g  } t t | d d d t	 j
 d	 t t t g  d | d  d   d
 f | d  d   d
 f <t t | d d d t	 j
 d	 t t t g  d  S(   Ngffffff?i   i   R%   i   i    i   R   R   i(   i    i   (   i    i   (   R   R&   R'   R,   R-   R)   R(   R   R   R   R   R   R   R   (   R   R*   R.   R!   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_n_equals_3O   s    .*c         C` ss   d } t  j g  t d d d  D] } t | |  ^ q  } t  j |  } t t  j | d k  d d | d  S(   Ng?i  iL  i   i    t   msgs   Not all diffs negative %s(   R   R-   R    R   t   diffR   t   all(   R   R*   R!   t   pvalst   dfs(    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_n_largeX   s    7(   t   __name__t
   __module__R   R   R"   R#   R$   R+   R/   R0   R6   (    (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR      s   										t   TestSmirnovic           B` sY   e  Z d    Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 RS(	   c         C` s#   t  t j t d t j    d  S(   Ni   (   R   R   R   R   R   (   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR   b   s    c         C` s   d d d d d d d f d	 d d f d
 d d d d f g } t  j |  } t t | d d d t j d t t t g  d | d  d   d f | d  d   d f <t t | d d d t j d t t t g  d  S(   Ni   g?g333333?gGz?g{Gz?i   g      ?g      ?i   i
   g      ?i   i    R   R   (   i   g?g333333?(   i   g333333?g?(   i   gGz?g{Gz?(   i   g{Gz?gGz?g      ?g      ?g      `?I       g      ?(   i    i   (   i    i   (	   R   R   R   R   R   R   R   R   R   (   R   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR   e   s    .*c         C` s   g  t  j t d d  t d d   D] } | d d f ^ q% } t j |  } t t | d	 d d t j d t	 t
 t
 g  d | d  d   d f | d  d   d f <t t | d
 d d t j d t	 t
 t
 g  d  S(   Ni   i   i  i  i    i   R   R   (   i    i   (   i    i   (   R   R   R    R   R   R   R   R   R   R   R   R   (   R   R!   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR"   s   s
    @.*c         C` s   g  t  j t d d  t d d   D] } | d d f ^ q% } t j |  } t t | d	 d d t j d t	 t
 t
 g  d | d  d   d f | d  d   d f <t t | d
 d d t j d t	 t
 t
 g  d  S(   Ni   i   i  i  i   i    R   R   (   i    i   (   i    i   (   R   R   R    R   R   R   R   R   R   R   R   R   (   R   R!   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR#   z   s
    @.*c         C` s   t  j d d d d t } t  j d g t |  | d | g  } t t | d d d t j d t	 t
 t
 g  d | d  d   d f | d  d   d f <t t | d	 d d t j d t	 t
 t
 g  d  S(
   Ni    i   ie   R%   i   R   R   (   i    i   (   i    i   (   R   R&   R'   R(   R)   R   R   R   R   R   R   R   (   R   t   ppR   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR+      s
    ).*c         C` s   t  j d d d d t } t  j d | d  } t  j d g t |   } t  j | | | g  } t t | d	 d d t	 j
 d t t t g  d | d  d   d f | d  d   d f <t t | d
 d d t	 j
 d t t t g  d  S(   Ng      ?i   ie   R%   i   i    R   R   (   i    i   (   i    i   (   R   R&   R'   R,   R-   R)   R(   R   R   R   R   R   R   R   (   R   R*   R.   R!   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR/      s    .*c         C` s   t  j d d d d t } t  j d | d  } t  j d g t |   } t  j | | | g  } t t | d
 d d t	 j
 d	 t t t g  d | d  d   d f | d  d   d f <t t | d d d t	 j
 d	 t t t g  d  S(   Ngffffff?i   i   R%   i   i    i   R   R   (   i    i   (   i    i   (   R   R&   R'   R,   R-   R)   R(   R   R   R   R   R   R   R   (   R   R*   R.   R!   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR0      s    .*c      	   C` s   d   } d   } d d d d d d d g } t  j |  } t | | d d d t j d t t t g  t | | d d d t j d t t t g  d  S(   Nc         S` s   t  |  t |  |   S(   N(   R   R   (   R!   R.   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   _sm_smi   s    c         S` s   t  |  t |  |   S(   N(   R   R   (   R!   R.   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt	   _smc_smci   s    i   g?g333333?i   g      ?i   g      ?i
   g+?g-C6?i    R   R   (   i   g?g?(   i   g333333?g333333?(   i   g      ?g      ?(   i   g      ?g      ?(   i   g      ?g      ?(   i
   g+?g+?(   i
   g-C6?g-C6?(   i    i   (   i    i   (   R   R   R   R   R   R   R   (   R   R;   R<   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_round_trip   s    			.c         C` s   d d d d d d d d  d! d" d# g } t  j |  } t t | d$ d d t j d t t t g  d | d  d   d f | d  d   d f <t t | d% d d t j d t t t g  d  S(&   Ni   g      ?i   gwzl?g      ?i   g<{?i   gFwrZ?i   g0 ?i   g0@?i   g A?i   g^CKf?i	   g޵&?i
   gL4J?i    R   R   (   i   g      ?g      ?(   i   g      ?gwzl?(   i   g      ?g      ?(   i   g      ?g<{?(   i   g      ?gFwrZ?(   i   g      ?g0 ?(   i   g      ?g0@?(   i   g      ?g A?(   i   g      ?g^CKf?(   i	   g      ?g޵&?(   i
   g      ?gL4J?(   i    i   (   i    i   (	   R   R   R   R   R   R   R   R   R   (   R   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR$      s    	.*(   R7   R8   R   R   R"   R#   R+   R/   R0   R=   R$   (    (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR9   a   s   										t   TestSmirnovpc           B` s_   e  Z d    Z d   Z d   Z d   Z d   Z e j j	 e
 j d
 k d d d	    Z RS(   c         C` s#   t  t j t d t j    d  S(   Ni   (   R   R   R   R   R   (   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR      s    c         C` s   t  j d d  } t  j | t  j | d  t  j | d  g  } t t | d	 d d t j d t t	 t	 g  t  j d d  } t  j | t  j | d  t  j | d  g  } t t | d
 d d t j d t t	 t	 g  d  S(   Ni   i
   i    ii   R   R   g      ?(   i    i   (   i    i   (
   R   t   arangeR(   t	   full_likeR   R   R   R   R   R   (   R   t   n1_10t   dataset0t   n2_10t   dataset1(    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR      s    0.0c         C` s   t  j d d  } d | } d d | } | | | d } | d t  j | d  d d 8} t  j | | | g  } t t | d	 d d t j d t t	 t	 g  d  S(
   Ni   i   g      ?i   g      ?i    R   R   (   i    i   (
   R   R?   t   signR(   R   R   R   R   R   R   (   R   R!   R*   t   xm1t   pp1RD   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_oneminusoneovern   s    
#c         C` s   t  j d d  } d | } | | d d | | d } t  j | | | g  } t t | d	 d d t j d t t t g  d  S(
   Ni   i   g      ?i   i    R   R   g      ?(   i    i   (	   R   R?   R(   R   R   R   R   R   R   (   R   R!   R*   R:   RB   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_oneovertwon   s
    
c         C` s   d t  j d d  } d | } | | d d | | d d } t  j | | | g  } t t | d	 d d t j d t t t g  d  S(
   Ni   i   i
   g      ?g      ?i    R   R   (   i    i   (	   R   R?   R(   R   R   R   R   R   R   (   R   R!   R*   R:   RB   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_oneovern   s
    
#i   i    t   reasons   requires 64-bit platformc         C` s   t  j d d  } d | d t  j t  j } | | d d | | d } t  j | | | g  } t t | d	 d d t j	 d t
 t t g  d | d t  j t  j } | | d d | | d d } t  j | | | g  } t t | d
 d d t j	 d t
 t t g  d  S(   Ni   i   g      ?i   i   i    R   R   (   i    i   (   i    i   (   R   R?   t   finfoR   t   epsR(   R   R   R   R   R   (   R   R!   R*   R:   RB   RD   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_oneovernclose   s    .#I       (   R7   R8   R   R   RH   RI   RJ   t   pytestt   markt   xfailt   syst   maxsizeRN   (    (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR>      s   		
	
		t   TestKolmogorovc           B` s>   e  Z d    Z d   Z d   Z d   Z d   Z d   Z RS(   c         C` s    t  t j t t j    d  S(   N(   R   R   R   R   R   (   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR      s    c         C` sG   d d d d d g } t  j |  } t t | d d d
 t j   d  S(   Ni    g      ?g      ?g˜?gBgR{{?i   g	=G?i   g !E?R   (   i    g      ?(   g      ?g˜?(   gBgR{{?g      ?(   i   g	=G?(   i   g !E?(   i    (   R   R   R   R   R   R   (   R   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR      s    	c         C` s  t  j d d d  } d d d d d d d	 d
 d d d d d d d d d d d d d g } d d d d d d d d d  d! d" d# d$ d% d& d' d( d) d* d+ d, g } t  j | | g  } t t | d/ d- d. t j   t  j | | g  } t t | d0 d- d. t j   d  S(1   Ni    g       @i   g      ?g;?gCg{?g?g˜?g<[4?gUOp?g|Xi?g\"?g
=G?gT	@x?g1qVd?grl}n?gS.Q?grI;N?glPz?ghMy?gn㨫D!i?g9=W?gK!E?g        glAcR5gkC a=gz>gbk g?gQ3u?g._?gTe`{?g{N,?gQn?g|)\?g=!P?g1csh?grB0?gz$?gl%I?gLZ?gt.d?gWT?gc"?g?i   R   (   i    (   i    (   R   R&   R(   R   R   R   R   R	   (   R   R*   R   t	   dataset_c(    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_linspace  s&    												c         C` s'  t  j d d d d t } t  j d d d d d	 d
 d d d d d d d d d d d d d d g } d d d d d d d d d d d d d d d d	 d d d d t  j g } t  j | d | d g  } t t | d" d d  t j   t  j | d!  | d!  g  } t t	 | d# d d  t j   d  S($   Ni    g      ?i   R%   gź?g3?gn	o5?g	)?g(!N?gU#?g r?gl?g2|?gBgR{{?gK)|?g?g2 ?gˍ?gu?gN?gk<?gf'G?g&?g        g&?gk?gU#?go	o5?g4?gź?i   R   i(   i    (   i    (
   R   R&   R'   t   infR(   R   R   R   R   R
   (   R   R.   R   RU   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_linspacei  s&    											c         C` s   d t  j d d  } t  j d d d d d d	 d
 d d d d d d g  } t  j | d | g  } t t | d d d t j   d  S(   Ng?i   i   gZ'G?gp=9?g=$7?g,^5?gȤ@?g1l4?gQGד?g%5?g	By?gZbqf-?gK?g	N?g"?i    R   (   i    (   R   R?   R-   R(   R   R   R   R   (   R   t   epsilonR*   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_smallx6  s    c         C` s   d   } d   } t  j d d d d t } | | d k | d k B} t  j | | g  } t | | d d	 d
 t j   t  j | | g  } t | | d d	 d
 t j   d  S(   Nc         S` s   t  t |    S(   N(   R   R   (   t   _x(    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   _ki_kA  s    c         S` s   t  t |    S(   N(   R
   R	   (   R[   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   _kci_kcD  s    g        g       @i   R%   i    gzG?i   R   (   i    (   i    (   R   R&   R'   R(   R   R   R   (   R   R\   R]   R*   t   x02t	   dataset02R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR=   @  s    		(   R7   R8   R   R   RV   RX   RZ   R=   (    (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyRT      s   		
			
t   TestKolmogic           B` s5   e  Z d    Z d   Z d   Z d   Z d   Z RS(   c         C` s    t  t j t t j    d  S(   N(   R   R   R   R   R   (   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR   Q  s    c         C` sJ   d d d d d d g } t  j |  } t t | d d	 d t j   d  S(   Ng      ?i    g˜?g      ?g?gZ'G?gBgR{{?g	=G?i   g !E?i   R   (   g      ?i    (   g˜?g      ?(   g?gZ'G?(   g      ?gBgR{{?(   g	=G?i   (   g !E?i   (   i    (   R   R   R   R   R   R   (   R   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR   T  s    	c         C` s   d t  j d d d  } t  j d d d d d	 d
 d d d d d d d d d d d d g  } t  j d | | g  } t t | d d d t j   t  j | | g  } t t | d d d t j   d  S(   Ng      ?i   i7   i   gBgR{{?gza<?gN(?gB9|?gZ"h?g:A?gi.^$?gڎ4?gMtxA?g~Iy?g^?gcw?guz)t?g<?gw(uR?gj_#K7?gvn@?g2n.?i    R   (   i    (   i    (	   R   R?   R-   R(   R   R   R   R   R
   (   R   RY   R*   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_smallpcdf_  s    				c         C` s   d t  j d d d  } t  j d d d d d	 d
 d d d d d d d d d d d d g  } t  j | | g  } t t | d d d t j   t  j d | | g  } t t | d d d t j   d  S(   Ng      ?i   i7   i   gBgR{{?gI\%	?gI?gX~=?gA^12@gk@g	@gd1.@g[{U@gNo[\	@gHaI
@gi@g/a<@gAї
(@g~=@g[5;$@gM @gH$@i    R   (   i    (   i    (	   R   R?   R-   R(   R   R   R   R   R
   (   R   RY   R*   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   test_smallpsfq  s    				c         C` s\   d   } t  j d d d d t } t  j | | g  } t | | d	 d d t j   d  S(
   Nc         S` s   t  t |    S(   N(   R   R   (   t   _p(    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   _k_ki  s    g?g      ?i
   R%   i    i   R   (   i    (   R   R&   R'   R(   R   R   R   (   R   Rd   R.   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR=     s    	(   R7   R8   R   R   Ra   Rb   R=   (    (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR`   P  s
   				t   TestKolmogpc           B` s   e  Z d    Z d   Z RS(   c         C` s    t  t j t t j    d  S(   N(   R   R   R   R   R   (   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR     s    c         C` sY   d d d d d d d  d! d" d# d$ g } t  j |  } t t | d% d d t j   d  S(&   Ng        g       g?g=g?gt鹿g333333?g^ڛ/g?g!$Jg      ?g2&g333333?guk~;gffffff?ggKk|q̿g?gg?g?𝖿g       @gz!ui    i   R   (   g        g       (   g?g=(   g?gt鹿(   g333333?g^ڛ/(   g?g!$J(   g      ?g2&(   g333333?guk~;(   gffffff?ggKk|q̿(   g?g(   g?g?𝖿(   g       @gz!u(   i    (   R   R   R   R   R   R   (   R   R   (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyR     s    	(   R7   R8   R   R   (    (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyRe     s   	(!   t
   __future__R    R   R   R   RR   RO   t   numpyR   t   numpy.testingR   t   scipy.special._testutilsR   t   scipy.specialR   R   R   R   t   scipy.special._ufuncsR	   R
   R   R   R   R   R   t   objectR   R9   R>   RT   R`   Re   (    (    (    sB   lib/python2.7/site-packages/scipy/special/tests/test_kolmogorov.pyt   <module>   s   ".P`9V<