
    bi                     @    d dl Zd dlZd dlmZ d dlmZ  G d d      Zy)    N)assert_allclosec                       e Zd Zej                  j
                  d        Zej                  j                  dg d      ej                  j                  dg d      d               Zy)TestSphHarmc                    d}d}t        j                  dt         j                        }t        j                  ddt         j                  z        }t        j                  ||      \  }}t	        j
                  ||||d      \  }}}t	        j                  |||d      \  }}	}
t        j                  t        j                  |dz         t        j                  | d      g      }t        j                  |dt        t        d|j                  dz               z         }t        ||t        j                  d	|z  |z        z         t        |d
   |	t        j                  d	|z  |z        z         t        |d   d	|z  |z  t        j                  d	|z  |z        z         t        |d   |
t        j                  d	|z  |z        z         t        |d   d	|z  |	z  t        j                  d	|z  |z        z         t        |d   |d          t        |d   | |z  |z  t        j                  d	|z  |z        z         y )N   
   r      )diff_n   r   axisy              ?).r   ).r   ).r   r   ).r   r   ).r   r   ).r   r   )nplinspacepimeshgridscsph_harm_y_allsph_legendre_p_allconcatenatearangeexpand_dimstuplerangendimr   exp)selfm_maxn_maxthetaphiyy_jacy_hesspp_jacp_hessms               \/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/scipy/special/tests/test_sph_harm.pytest_pzTestSphHarm.test_p   s   Aruu%kk!QruuW%[[,
s,,UE5#aP5&00uQO5&NNBIIeai0"))UFA2FGHNN14eAejjl.C(D#DE1rvvb1fsl334furvvb1fsl/C'CDfrAvzBFF26C<4H'HIy)6BFF26C<4H+HIy)26E>BFF26C<<P+PQy)6)+<=y)A26A:rAv|8L+LM    r   )   r   2   r   )r         	      c                    t        j                  dt         j                        }t        j                  ddt         j                  z        }t        j                  |dz         }t        j                  |t        t        d|j                  dz                     }t        j                  t        j                  |dz         t        j                  | d      g      }t        j                  |dt        t        d|j                  dz               z         }t        j                  ||||      }t        j                  ||||      }t         j                  j                  ||d       y )Nr   r	   r   r   r   gh㈵>)rtol)r   r   r   r   r   r   r   r   r   r   r   
sph_harm_ytestingr   )	r   r   r   r    r!   nr(   y_actual	y_desireds	            r)   test_allzTestSphHarm.test_all!   s     Aruu%kk!QY'IIeai NN15qA)>#?@NNBIIeai0"))UFA2FGHNN14eAejjl.C(D#DE$$UE5#>MM!Qs3	


""8YU"Cr+   N)	__name__
__module____qualname__pytestmarkslowr*   parametrizer9    r+   r)   r   r      sa    [[N N0 [[Wk2[[W&67D 8 3Dr+   r   )	numpyr   r=   numpy.testingr   scipy.specialspecialr   r   rA   r+   r)   <module>rF      s      ) )D )Dr+   