
    Ug+                     >    d dl Zd dlmZ d dlmZ d dlmZ d Z	d Z
dS )    N)assert_allclose)_sph_harm_allc                     d } d }d }d }| |||g}g d}g d}t          j        ddt           j        z            }t          j        dt           j                  }t          j        ||          \  }}t	          |||          D ]@\  }	}}t          t          j        ||||           |	||          d	d	d
| d| d           Ad S )Nc                 J    dt          j        dt           j        z            z  S )N      ?   )npsqrtpithetaphis     `/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/scipy/special/tests/test_sph_harm.pyY00z!test_first_harmonics.<locals>.Y00   s    271RU7####    c                     dt          j        ddt           j        z  z            z  t          j        d| z            z  t          j        |          z  S )Nr         y             r	   r
   r   expsinr   s     r   Yn11z"test_first_harmonics.<locals>.Yn11   sA    271ag;'''s5y(9(99"&++EEr   c                 t    dt          j        dt           j        z            z  t          j        |          z  S )Nr   r   )r	   r
   r   cosr   s     r   Y01z!test_first_harmonics.<locals>.Y01   s*    271RU7###BF3KK//r   c                     dt          j        ddt           j        z  z            z  t          j        d| z            z  t          j        |          z  S )Ng      r   r   y              ?r   r   s     r   Y11z!test_first_harmonics.<locals>.Y11   sA    BGAqwK(((5)9)99"&++EEr   )r   r   r   )r   r   r   r   r   r   gV瞯<zY^_z
 incorrect)rtolatolerr_msg)r	   linspacer   meshgridzipr   scsph_harm)
r   r   r   r   harmsmnr   r   harms
             r   test_first_harmonicsr,      s(   $ $ $F F F0 0 0F F F $S!EAAK1RU7##E
+a

CUC((JE3%A&& 8 8
aAq%55UC((" 6Q 6 6 6 6 6	8 	8 	8 	8 	88 8r   c                     d} t          j        ddt           j        z            }t          j        dt           j                  }t          d| z  | ||          }dD ]}dD ]}||k    rt	          j        ||||          }nd}t           j                            |||f         |d           ||k    rt	          j        | |||          }nd}t           j                            || |f         |d           d S )N2   r   r   )r   r   r      
      r.   gh㈵>)r    )r	   r#   r   r   r&   r'   testingr   )n_maxr   r   y_actualr*   r)   	y_desireds          r   test_all_harmonicsr6   )   s   EK1ru9%%E
+a

CQYuc::H% Q Q) 	Q 	QAQK1eS99			J&&x1~y&OOOQKAuc::			J&&xA	%&PPPP	QQ Qr   )numpyr	   numpy.testingr   scipy.specialspecialr&   scipy.special._basicr   r,   r6    r   r   <module>r=      su        ) ) ) ) ) )       . . . . . .8 8 8DQ Q Q Q Qr   