ó
Ê½÷Xc           @` s®   d  d l  m Z m Z m Z d  d l Z d  d l m Z m Z m	 Z	 m
 Z
 m Z m Z d  d l m Z m Z d e f d „  ƒ  YZ d „  Z d „  Z e d	 k rª e	 ƒ  n  d S(
   i    (   t   divisiont   print_functiont   absolute_importN(   t   assert_array_equalt   TestCaset   run_module_suitet   assert_array_almost_equal_nulpt   assert_raisest   assert_almost_equal(   t   gammalnt   multigammalnt   TestMultiGammaLnc           B` s#   e  Z d  „  Z d „  Z d „  Z RS(   c         C` sH   t  j j d ƒ t  j t  j j ƒ  ƒ } t t | d ƒ t | ƒ ƒ d  S(   NiÒ  i   (   t   npt   randomt   seedt   abst   randnR   R
   R	   (   t   selft   a(    (    sC   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_spfun_stats.pyt   test1   s    c         C` sh   t  j d d g ƒ } t | d ƒ } t  j t  j t  j ƒ ƒ t | ƒ t | d ƒ } t | | ƒ d  S(   Ng      @g      $@i   g      à?(   R   t   arrayR
   t   logt   sqrtt   piR	   R   (   R   R   t   resultt   expected(    (    sC   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_spfun_stats.pyt   test2   s    3c         C` s   t  t t d d ƒ d  S(   Ng      à?g333333ó?(   R   t
   ValueErrorR
   (   R   (    (    sC   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_spfun_stats.pyt   test_bararg   s    (   t   __name__t
   __module__R   R   R   (    (    (    sC   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_spfun_stats.pyR   
   s   		c         C` sv   t  |  | ƒ } t |  j | j ƒ |  j ƒ  } | j ƒ  } x5 t |  j ƒ D]$ } t | | t  | | | ƒ ƒ qJ Wd  S(   N(   R
   R   t   shapet   ravelt   ranget   sizeR   (   R   t   dR   t   a1t   result1t   i(    (    sC   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_spfun_stats.pyt    _check_multigammaln_array_result   s    c          c` sÔ   t  j j d ƒ t  j t  j j d d ƒ ƒ d d f t  j t  j j d d ƒ ƒ d d f t  j d d ƒ j d d d ƒ d f t  j d g ƒ d f t  j d ƒ d f g }  x" |  D] \ } } t	 | | f Vq² Wd  S(	   NiÒ  i   i   i   i   g      $@g      2@g       @(
   R   R   R   R   R   t   aranget   reshapeR   t   float64R'   (   t   casesR   R#   (    (    sC   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_spfun_stats.pyt   test_multigammaln_array_arg+   s    %%$t   __main__(   t
   __future__R    R   R   t   numpyR   t   numpy.testingR   R   R   R   R   R   t   scipy.specialR	   R
   R   R'   R,   R   (    (    (    sC   /tmp/pip-build-7oUkmx/scipy/scipy/special/tests/test_spfun_stats.pyt   <module>   s   .		