| Current File : //usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyo |
�
E�`Qc @ s� d Z d d l m Z d d l Z d d l j j Z d d l m
Z
d d l Te j d g � Z
e j d d g � Z e j d d d g � Z e j d d d d g � Z e j d d d
d d g � Z e j d d d d d d g � Z e j d
d d d d
d d g � Z e j d d d d d d d d g � Z e j d d d d d d d d d g � Z e j d d d d d d d d d d g
� Z e
e e e e e e e e e g
Z d � Z d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d! � � YZ d" e f d# � � YZ d$ e f d% � � YZ d& e f d' � � YZ d( e f d) � � YZ! d* e f d+ � � YZ" e# d, k r�e$ � n d S(- s Tests for hermite_e module.
i����( t divisionN( t polyval( t *i i i����i i����i i����i�i- i����ii i��i\���i� i��i� i���iz i���c C s t j | d d �S( Nt tolg���ư>( t hermet hermetrim( t x( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt trim s t
TestConstantsc B s, e Z d � Z d � Z d � Z d � Z RS( c C s t t j d d g � d S( Ni����i ( t assert_equalR t hermedomain( t self( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermedomain s c C s t t j d g � d S( Ni ( R R t hermezero( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermezero! s c C s t t j d g � d S( Ni ( R R t hermeone( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermeone$ s c C s t t j d d g � d S( Ni i ( R R t hermex( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermex' s ( t __name__t
__module__R R R R ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyR s t TestArithmeticc B sJ e Z e j d d d � Z d � Z d � Z d � Z d � Z d � Z RS( i����i id c C s� x� t d � D]� } x� t d � D]� } d | | f } t j t | | � d � } | | c d 7<| | c d 7<t j d g | d g d g | d g � } t t | � t | � d | �q Wq
Wd S( Ni s
At i=%d, j=%di i t err_msg( t ranget npt zerost maxR t hermeaddR R ( R t it jt msgt tgtt res( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermeadd. s .c C s� x� t d � D]� } x� t d � D]� } d | | f } t j t | | � d � } | | c d 7<| | c d 8<t j d g | d g d g | d g � } t t | � t | � d | �q Wq
Wd S( Ni s
At i=%d, j=%di i R ( R R R R R t hermesubR R ( R R R R R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermesub8 s .c C s� t t j d g � d g � t t j d g � d d g � x_ t d d � D]N } d g | d g } d g | d | d d g } t t j | � | � qK Wd S( Ni i i ( R R t hermemulxR ( R R t serR ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermemulxB s c
C s� x� t d � D]� } d g | d g } t j | j | � } x� t d � D]� } d | | f } d g | d g } t j | j | � } t j | | � } t j | j | � } t t | � | | d k | � t | | | d | �qI Wq
Wd S( Ni i i s
At i=%d, j=%dR ( R R t hermevalR t hermemult assert_t lent assert_almost_equal(
R R t pol1t val1R R t pol2t val2t pol3t val3( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermemulJ s !c
C s� x� t d � D]� } x� t d � D]� } d | | f } d g | d g } d g | d g } t j | | � } t j | | � \ } } t j t j | | � | � } t t | � t | � d | �q Wq
Wd S( Ni s
At i=%d, j=%di i R ( R R R t hermedivR( R R (
R R R R t cit cjR t quot remR ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermedivX s (
R R R t linspaceR R! R# R& R2 R8 ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyR + s
t TestEvaluationc B s� e Z e j d d d g � Z e j d e e � Z e j d e e e � Z e j j d � d d Z e
e d d d g � Z d
� Z d � Z
d � Z d
� Z d � Z RS( g @g @g @s i,j->ijs
i,j,k->ijki i i i g �?c
C sW t t j g d g � j d � t j d d � } g t D] } t | | � ^ q8 } xd t d � D]V } d | } t j } | | } t j | d g | d g � } t
| | d | �q` Wx� t d � D]� } d g | } t j | � } t t j | d g � j | � t t j | d d g � j | � t t j | d d d g � j | � q� Wd S( Ni i i����i
s At i=%dR i i ( R R R' t sizeR R9 t HelistR R R R+ t shape(
R R t ct yR R R% R R t dims( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermevalo s "
"c
C s� | j \ } } } | j \ } } } t t t j | | d | j � | | } t j | | | j � } t | | � t j d � } t j | | | j � } t
| j d k � d S( Ni i ( i i ( i i ( R R? t
assert_raisest
ValueErrorR t
hermeval2dt c2dR+ R t onesR) R= (
R t x1t x2t x3t y1t y2t y3R R t z( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermeval2d� s
c
C s� | j \ } } } | j \ } } } t t t j | | | d | j � | | | } t j | | | | j � } t | | � t j d � } t j | | | | j � } t
| j d k � d S( Ni i ( i i ( i i ( R R? RB RC R t
hermeval3dt c3dR+ R RF R) R= (
R RG RH RI RJ RK RL R R RM ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermeval3d� s #
c
C s� | j \ } } } | j \ } } } t j d | | � } t j | | | j � } t | | � t j d � } t j | | | j � } t | j
d k � d S( Ns i,j->iji i ( i i ( i i ( i i i i ( R R? R t einsumR t hermegrid2dRE R+ RF R) R= (
R RG RH RI RJ RK RL R R RM ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermegrid2d� s
c
C s� | j \ } } } | j \ } } } t j d | | | � } t j | | | | j � } t | | � t j d � } t j | | | | j � } t | j
d k � d S( Ns
i,j,k->ijki i ( i i ( i i ( i i i i i i ( R R? R RR R t hermegrid3dRP R+ RF R) R= (
R RG RH RI RJ RK RL R R RM ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermegrid3d� s
( i i ( R R R t arrayt c1dRR RE RP t randomR R R? RA RN RQ RT RV ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyR: d s t TestIntegralc B s e Z d � Z d � Z RS( c
C s� t t t j d g d � t t t j d g d � t t t j d g d d d g � x` t d d � D]O } d g | d d g } t j d g d | d | �} t | d d g � qd Wx� t d � D]� } | d } d g | d g } | g d g | d | g } t j | � } t j | d d d | g �} t j | � } t t | � t | � � q� Wx~ t d � D]p } | d } d g | d g } t j | � } t j | d d d | g d d �} t t j d | � | � qmWx� t d � D]� } | d } d g | d g } | g d g | d | g } t j | � } t j | d d d | g d
d �} t j | � } t t | � t | � � q�Wx� t d � D]� } x� t d d � D]{ } d g | d g } | } x) t | � D] } t j | d d �} q�Wt j | d | �} t t | � t | � � q�Wq�Wx� t d � D]� } x� t d d � D]� } d g | d g } | } x2 t | � D]$ } t j | d d d | g �} q�Wt j | d | d t | � �} t t | � t | � � qYWqCWx� t d � D]� } x� t d d � D]� } d g | d g } | } x8 t | � D]* } t j | d d d | g d d �} qBWt j | d | d t | � d d �} t t | � t | � � qWq�Wx� t d � D]� } x� t d d � D]� } d g | d g } | } x8 t | � D]* } t j | d d d | g d
d �} q Wt j | d | d t | � d
d �} t t | � t | � � q�Wq�Wd S( Ni g �?i����i i i t mt kt lbndt scl(
RB RC R t hermeintR R+ t
poly2hermet
herme2polyR R' (
R R R\ R R^ t polR t hermepolR_ R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermeint� sz "
$
$ !"!!('!('c C s t j j d � } t j g | j D] } t j | � ^ q" � j } t j | d d �} t | | � t j g | D] } t j | � ^ qr � } t j | d d �} t | | � t j g | D] } t j | d d �^ q� � } t j | d d d d �} t | | � d S( Ni i t axisi i R\ ( i i ( R RY t vstackt TR R_ R+ ( R RE R> R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermeint_axis s 1
+
1( R R Rd Rh ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyRZ � s Pt TestDerivativec B s e Z d � Z d � Z RS( c C s� t t t j d g d � t t t j d g d � xV t d � D]H } d g | d g } t j | d d �} t t | � t | � � q? Wx t d � D]q } xh t d d � D]W } d g | d g } t j t j | d | �d | �} t t | � t | � � q� Wq� Wx� t d � D]} } xt t d d � D]c } d g | d g } t j t j | d | d d �d | d d �} t t | � t | � � q0WqWd S( Ni g �?i����i i R[ i R^ ( RB RC R t hermederR R R R_ R+ ( R R R R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermeder) s $!0c C s� t j j d � } t j g | j D] } t j | � ^ q" � j } t j | d d �} t | | � t j g | D] } t j | � ^ qr � } t j | d d �} t | | � d S( Ni i Re i i ( i i ( R RY Rf Rg R Rj R+ ( R RE R> R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermeder_axisB s 1
+( R R Rk Rl ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyRi ' s t
TestVanderc B s= e Z e j j d � d d Z d � Z d � Z d � Z RS( i i i i c C s# t j d � } t j | d � } t | j d k � xK t d � D]= } d g | d g } t | d | f t j | | � � qA Wt j d d g d d g d d g g � } t j | d � } t | j d
k � xK t d � D]= } d g | d g } t | d | f t j | | � � q� Wd S( Ni i i i .i i i ( i i ( i i i (
R t arangeR t hermevanderR) R= R R+ R' RW ( R R t vR t coef( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermevanderT s '*c C s� | j \ } } } t j j d � } t j | | d d g � } t j | | | � } t j | | j � } t | | � t j | g | g d d g � } t | j
d k � d S( Ni i i i i ( i i ( i i i ( R R RY R t
hermevander2dRD t dott flatR+ R) R= ( R RG RH RI R> t vanR R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermevander2de s
!c C s� | j \ } } } t j j d � } t j | | | d d d g � } t j | | | | � } t j | | j � } t | | � t j | g | g | g d d d g � } t | j
d k � d S( Ni i i i i i ( i i i ( i i i ( R R RY R t
hermevander3dRO Rt Ru R+ R) R= ( R RG RH RI R> Rv R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermevander3ds s !
*( i i ( R R R RY R Rr Rw Ry ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyRm O s t TestFittingc B s e Z d � Z RS( c C s� d � } t t t j d g d g d � t t t j d g g d g d � t t t j g d g d � t t t j d g d g g g d � t t t j d d g d g d � t t t j d g d d g d � t t t j d g d g d d d g g �t t t j d g d g d d d d g �t j d d � } | | � } t j | | d � } t t | � d � t t j
| | � | � t j | | d � } t t | � d � t t j
| | � | � t j | t j | | g � j d � } t | t j | | g � j � t j
| � } | j � } d | d d d � <d | d d d � <t j | | d d | �} t | | � t j | t j | | g � j d d | �}
t |
t j | | g � j � d d
d d g } t t j | | d � d d g � d S( Nc S s | | d | d S( Ni i ( ( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt f� s i i����i i t wi i i y �?y �( RB RC R t hermefitt TypeErrorR R9 R R* R+ R' RW Rg t
zeros_liket copy( R R{ R R? t coef3t coef4t coef2dR| t ywt wcoef3t wcoef2d( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt
test_hermefit� s: " %""++ '
- ( R R R� ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyRz � s t TestGaussc B s e Z d � Z RS( c C s� t j d � \ } } t j | d � } t j | j | | � } d t j | j � � } | d d � d f | | } t | t j
d � � t j d t j � } t | j � | � d S( Nid ic i i (
R t
hermegaussRo R Rt Rg t sqrtt diagonalt NoneR+ t eyet pit sum( R R R| Rp t vvt vdR ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_100� s ( R R R� ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyR� � s t TestMiscc B sG e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C s� t j g � } t t | � d g � x� t d d � D]� } t j t j t j d d | d � d d d � � } t j | � } t j | | � } d } t
t | � | d k � t t j | � d d � t | | � q5 Wd S( Ni i i i i����(
R t hermefromrootsR+ R R R t cosR9 R� R' R) R* Ra ( R R R t rootsRb R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermefromroots� s 7c C s� t t j d g � g � t t j d d g � d g � x] t d d � D]L } t j d d | � } t j t j | � � } t t | � t | � � qH Wd S( Ni i����i i ( R+ R t
hermerootsR R R9 R� R ( R R R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermeroots� s c C s d d d d g } t t t j | d � t t j | � | d � t t j | d � | d � t t j | d � d g � d S( Ni i����i i i����( RB RC R R R ( R Rq ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermetrim� s
c C s# t t j d d � d d g � d S( Ni i ( R R t hermeline( R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_hermeline� s c C sC x< t d � D]. } t t j d g | d g � t | � q
Wd S( Ni
i i ( R R+ R Ra R< ( R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_herme2poly� s c C sC x< t d � D]. } t t j t | � d g | d g � q
Wd S( Ni
i i ( R R+ R R` R< ( R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_poly2herme� s c C sL t j d d d � } t j d | d � } t j | � } t | | � d S( Ni����i i g �i ( R R9 t expR t hermeweightR+ ( R R R R ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt test_weight� s ( R R R� R� R� R� R� R� R� ( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyR� � s t __main__(% t __doc__t
__future__R t numpyR t numpy.polynomial.hermite_et
polynomialt hermite_eR t numpy.polynomial.polynomialR t
numpy.testingRW t He0t He1t He2t He3t He4t He5t He6t He7t He8t He9R< R t TestCaseR R R: RZ Ri Rm Rz R� R� R t run_module_suite( ( ( sK /usr/lib64/python2.7/site-packages/numpy/polynomial/tests/test_hermite_e.pyt <module> s8
!$'*-$ 9`c(2-3