
    bi                     \    d dl Zd dlmZmZmZ d dlmZ d dl	m
Z
mZ  G d d      Zd Zd Zy)	    N)assert_array_equalassert_array_almost_equal_nulpassert_allclose)raises)gammalnmultigammalnc                       e Zd Zd Zd Zd Zy)TestMultiGammaLnc                     t         j                  j                  d       t        j                  t         j                  j	                               }t        t        |d      t        |             y )N     )nprandomseedabsrandnr   r   r   )selfas     _/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scipy/special/tests/test_spfun_stats.pytest1zTestMultiGammaLn.test1   sB     			tFF299??$%<1-wqz:    c                 
   t        j                  ddg      }t        |d      }t        j                  t        j                  t         j
                              t        |      z   t        |dz
        z   }t        ||dd       y )Ng      @      $@         ?gv!>r   )atolrtol)r   arrayr   logsqrtpir   r   )r   r   resultexpecteds       r   test2zTestMultiGammaLn.test2   sa     HHc4[!a#66"''"%%.)GAJ6S9IIvA>r   c                 0    t        t        t        dd       y )Nr   g333333?)assert_raises
ValueErrorr   )r   s    r   test_barargzTestMultiGammaLn.test_bararg   s    j,S9r   N)__name__
__module____qualname__r   r$   r(    r   r   r
   r
   	   s    ;?:r   r
   c                    t        | |      }t        | j                  |j                         | j                         }|j                         }t	        | j
                        D ]  }t        ||   t        ||   |               y )N)r   r   shaperavelrangesizer   )r   dr"   a1result1is         r    _check_multigammaln_array_resultr6      sj     !QFqww-	
BllnG166] K&wqz<1q3IJKr   c                     t         j                  j                  d       t        j                  t         j                  j	                  dd            dz   dft        j                  t         j                  j	                  dd            dz   dft        j
                  dd      j                  ddd      dft        j                  dg      dft        j                  d      dfg} | D ]  \  }}t        ||        y )	Nr      r      r   r   g      2@g       @)
r   r   r   r   r   arangereshaper   float64r6   )casesr   r2   s      r   test_multigammaln_array_argr>   *   s     IINN4 
		1%	&	*A.			1%	&	*A.	4		&	&q!Q	/3	3%!	C!E  /1(A./r   )numpyr   numpy.testingr   r   r   pytestr   r&   scipy.specialr   r   r
   r6   r>   r,   r   r   <module>rC      s-    , , * /: :*	K/r   