ó
ÐEe]c           @` sŠ   d  d l  m Z m Z m Z d  d l Z d  d l m Z m Z m Z d  d l	 m
 Z
 d  d l m Z d  d l	 Z d e f d „  ƒ  YZ d S(   i    (   t   divisiont   absolute_importt   print_functionN(   t   assert_t   assert_array_equalt   assert_raises(   t   random(   t   longt   TestRegressionc           B` st   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 RS(   c         C` sq   xj t  j d d d ƒ D]S } t j j | d d ƒ } t t  j | t  j k ƒ oe t  j | t  j k ƒ ƒ q Wd  S(   Ng      Àg      @i   i   i2   (   t   npt   linspaceR   t   mtrandt   vonmisesR   t   allt   pi(   t   selft   mut   r(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_VonMises_range   s    c         C` sâ   t  t j t j j d d d d d ƒd k  ƒ ƒ t  t j t j j d d d d d ƒd k ƒ ƒ d d d f g } t j d k } | r± t j d k r± | j d d d f ƒ n  x* | D]" } t  t j j | Œ  d k ƒ q¸ Wd  S(   Ni   i   i   t   sizei
   i   i    i   i   i    t   win32i(   i   iþÿ i   iþÿ i   iþÿ I       I       Iþÿÿÿÿ   I       Iþÿÿÿÿ   I       Iþÿÿÿÿ   (	   R   R	   R   R   t   hypergeometrict   syst   maxsizet   platformt   append(   R   t   argst	   is_64bitst   arg(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_hypergeometric_range   s    11c         C` sª   d } t  j j d ƒ t  j j d d | ƒ} t  j | d k ƒ t | ƒ } d | } t | d k | ƒ t  j | d k ƒ t | ƒ } d	 | } t | d
 k  | ƒ d  S(   Niè  i    gš™™™™™é?R   i   s"   Frequency was %f, should be > 0.45gÍÌÌÌÌÌÜ?i   s"   Frequency was %f, should be < 0.23gq=
×£pÍ?(   R	   R   t   seedt	   logseriest   sumt   floatR   (   R   t   Nt   rvsnt   freqt   msg(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_logseries_convergence$   s    

c         C` s[   t  j j d ƒ t  j j d ƒ } t  j j d ƒ t  j j t d ƒ ƒ } t | | ƒ d  S(   NiÒ  i   (   R	   R   R   t   permutationR   R   (   R   t   at   b(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_permutation_longs4   s
    c         C` s£   xœ d d d d  g d d d d  g d d	 d
 d  g d d d d  g g D]X } t j j d ƒ t | ƒ } t j | ƒ t | | d | d | d | d g ƒ qC Wd  S(   Ni   i   i   i90  i    (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   i   i   (   t   NoneR	   R   R   t   listt   shuffleR   (   R   t   tt   shuffled(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_shuffle_mixed_dimension;   s    c         C` s£   t  j j ƒ  } t  j d d d d d d d d d d g
 ƒ } x` t d ƒ D]R } t  j j | ƒ | j d	 ƒ t | j d
 d d
 d t  j d
 ƒ d ƒ| ƒ qI Wd  S(   Ni    i   i   i   i   i	   i   i   iá  i
   R   t   pg      $@(	   R	   R   t   RandomStatet   arrayt   rangeR   R   t   choicet   ones(   R   t   mt   rest   i(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_call_within_randomstateF   s    -c         C` s|   t  j j d g d g g d d ƒt  j j d g d g g d t  j d ƒ ƒt  j j d g d g g d t  j d ƒ ƒd  S(   Ni    R   i   (   R	   R   t   multivariate_normalt   int_t   int64(   R   (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt#   test_multivariate_normal_size_typesP   s    "+c         C` sO   t  j j d ƒ t  j j d d d d ƒ} t t  j t  j | ƒ ƒ d ƒ d  S(   NiÒ–Ig-Cëâ6?R   id   s   Nans in np.random.beta(   R	   R   R   t   betaR   t   anyt   isnan(   R   t   x(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_beta_small_parametersX   s    c         C` s¼   t  j j d ƒ d d d g } d d d g } x‡ t  j t  j t  j f D]m } t  j | d | ƒt | ƒ } t  j j | d | ƒ} t	 | | k ƒ t
 t t  j j | d | d ƒqG Wd  S(	   NiÒ  i   i   i   i   t   dtypeR1   gÍÌÌÌÌÌì?(   R	   R   R   t   float16t   float32t   float64R3   R    R5   R   R   t
   ValueError(   R   R(   t   countst   dtt   probst   c(    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt"   test_choice_sum_of_probs_tolerance_   s    c         C` sj   t  j j d ƒ t  j d d d g ƒ } x$ t d ƒ D] } t  j j | ƒ q6 Wd d  l } | j ƒ  d  S(   NiÒ  R(   iè  id   i    (   R	   R   R   R3   R4   R-   t   gct   collect(   R   R(   t   _RN   (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt1   test_shuffle_of_array_of_different_length_stringsl   s    c         C` sx   t  j j d ƒ t  j t  j d ƒ t  j d ƒ g ƒ } x$ t d ƒ D] } t  j j | ƒ qD Wd d  l } | j ƒ  d  S(   NiÒ  i   i   iè  i    (	   R	   R   R   R3   t   arangeR4   R-   RN   RO   (   R   R(   RP   RN   (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt    test_shuffle_of_array_of_objects{   s    'c         C` s  d t  j f d „  ƒ  Y} t  j j d ƒ t  j d ƒ j | ƒ } t  j j | ƒ } t | t  j d d d g ƒ ƒ t | t  j d ƒ j | ƒ ƒ d t	 f d „  ƒ  Y} t  j j d ƒ | ƒ  } t  j j | ƒ } t | t  j d d d	 d d g ƒ ƒ t | j
 ƒ  t  j d
 ƒ ƒ d  S(   NR"   c           B` s   e  Z RS(    (   t   __name__t
   __module__(    (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyR"   Š   s   i   i   i    i   t   Mc           B` s    e  Z e j d  ƒ Z d „  Z RS(   i   c         S` s   |  j  S(   N(   R(   (   R   (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt	   __array__–   s    (   RT   RU   R	   RR   R(   RW   (    (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyRV   “   s   i   i   (   R	   t   ndarrayR   R   RR   t   viewR'   R   R3   t   objectRW   (   R   R"   t   origt   permRV   R7   (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   test_permutation_subclass‰   s    	%(   RT   RU   R   R   R&   R*   R0   R:   R>   RC   RM   RQ   RS   R]   (    (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyR      s   						
					(   t
   __future__R    R   R   R   t   numpy.testingR   R   R   t   numpyR   t   numpy.compatR   R	   RZ   R   (    (    (    sA   lib/python2.7/site-packages/numpy/random/tests/test_regression.pyt   <module>   s   