File: //opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyo
�
� 9Zc @` sq d Z d d l m Z m Z m Z d d l Z d d l j j Z
d d l m Z d d l
m Z m Z m Z m Z m Z m Z e j d g � d Z e j d d g � d Z e j d d d g � d Z e j d
d d d g � d
Z e j d
d d d d g � d
Z e j d d d d d d g � d Z e j d d d d d d d g � d Z 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& d/ e f d0 � � YZ' e( d1 k rme � n d S(2 s Tests for laguerre module.
i ( t divisiont absolute_importt print_functionN( t polyval( t TestCaset assert_almost_equalt
assert_raisest assert_equalt assert_t run_module_suitei i����i i����i i��i i i����iH i�ix i����iX i8���i i� i �i i����i� i���c C` s t j | d d �S( Nt tolg���ư>( t lagt lagtrim( t x( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.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 ( R R t lagdomain( t self( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagdomain s c C` s t t j d g � d S( Ni ( R R t lagzero( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagzero! s c C` s t t j d g � d S( Ni ( R R t lagone( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagone$ s c C` s t t j d d g � d S( Ni i����( R R t lagx( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagx' s ( t __name__t
__module__R R R R ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.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 lagaddR R ( R t it jt msgt tgtt res( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagadd. 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 lagsubR R ( R R" R# R$ R% R& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagsub8 s .c C` s� t t j d g � d g � t t j d g � d d g � xm t d d � D]\ } d g | d g } d g | d | d | d | d g } t t j | � | � qK Wd S( Ni i i����i i ( R R t lagmulxR R ( R R" t serR% ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagmulxB 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 lagvalR
t lagmulR t lenR (
R R" t pol1t val1R# R$ t pol2t val2t pol3t val3( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagmulJ 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 lagdivR. R R (
R R" R# R$ t cit cjR% t quot remR& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagdivX s (
R R R t linspaceR
R' R) R, R6 R< ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.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 �?g @g @c C` sN t t j g d g � j d � t j d d � } g t D] } t | | � ^ q8 } x[ t d � D]M } d | } | | } 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 R= t LlistR R R R t shape( R R
t ct yR" R$ R% R& t dims( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagvaln 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
RC R t
ValueErrorR t lagval2dt c2dR R t onesR RA (
R t x1t x2t x3t y1t y2t y3R% R& t z( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt
test_lagval2d� 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
RC R RF R t lagval3dt c3dR R RI R RA (
R RJ RK RL RM RN RO R% R& RP ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt
test_lagval3d� 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
RC R t einsumR t laggrid2dRH R RI R RA (
R RJ RK RL RM RN RO R% R& RP ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_laggrid2d� 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
RC R RU R t laggrid3dRS R RI R RA (
R RJ RK RL RM RN RO R% R& RP ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_laggrid3d� s
( i i ( R R R t arrayt c1dRU RH RS t randomR
R RC RE RQ RT RW RY ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.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 | � 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 �} qHWt j | d | d t
t | � � d d �} t t | � t | � � qWqWx� t d � D]� } x� t d d � D]� } d g | d g } | } x8 t | � D]* } t j | d d d | g d
d �} qWt j | d | d t
t | � � d
d �} t t | � t | � � q�Wq�Wd S( Ni g �?i����i i i t mt kt lbndt scl( R RF R t lagintR R t poly2lagt lag2polyR R- t list(
R R" R_ R& Ra t polR% t lagpolRb R# ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagint� 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 R\ t vstackt TR Rb R ( R RH RB R% R& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagint_axis s 1
+
1( R R Rh Rl ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyR] � 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 Ra ( R RF R t lagderR R R Rb R ( R R" R% R& R# ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagder' 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 Ri i i ( i i ( R R\ Rj Rk R Rn R ( R RH RB R% R& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagder_axis@ s 1
+( R R Ro Rp ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyRm % 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 lagvanderR RA R R R- RZ ( R R
t vR" t coef( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagvanderQ 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 R\ R t lagvander2dRG t dott flatR R RA ( R RJ RK RL RB t vanR% R& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagvander2db 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 R\ R t lagvander3dRR Rx Ry R R RA ( R RJ RK RL RB Rz R% R& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagvander3do s !
*( i i ( R R R R\ R
Rv R{ R} ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyRq M 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 t t j d g d g d g � t t t j d g d g d d d g � t t t j d g d g g � t j d d � } | | � } t j | | d � } t t | � d � t t j
| | � | � t j | | d d d d g � } t t | � d � t t j
| | � | � t j | | d � } t t | � d
� t t j
| | � | � t j | | d d d d d g � } t t | � d
� t t j
| | � | � t j | t j | | g � j d � } t | t j | | g � j � t j | t j | | g � j d d d d g � } t | t j | | g � j � t j
| � } | j � } d | d d d � <d | d d d � <t j | | d d | �} t | | � t j | | d d d d g d | �} t | | � t j | t j | | g � j d d | �}
t |
t j | | g � j � t j | t j | | g � j d d d d g d | �}
t |
t j | | g � j � d d d d g } t t j | | d � d d g � t t j | | d d g � d d g � d S(
Nc S` s | | d | d S( Ni i ( ( R
( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt f� s i i����i i t wi i i i y �?y �( R RF R t lagfitt TypeErrorR R= R R/ R R- RZ Rk t
zeros_liket copy( R R R
RC t coef3t coef4t coef2dR� t ywt wcoef3t wcoef2d( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagfit sZ " %""++"( !$' 3
'
- 9 "( R R R� ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyR~ } s t
TestCompanionc B` s# e Z d � Z d � Z d � Z RS( c C` s- t t t j g � t t t j d g � d S( Ni ( R RF R t lagcompanion( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_raises� s c C` sT xM t d d � D]< } d g | d g } t t j | � j | | f k � q Wd S( Ni i i ( R R R R� RA ( R R" Ru ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_dimensions� s c C` s' t t j d d g � d d k � d S( Ni i i g �?( i i ( R R R� ( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_linear_root� s ( R R R� R� R� ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyR� � 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 � � d } t | j � | � d S( Nid ic i g �?( R t laggaussRs R Rx Rk t sqrtt diagonalt NoneR t eyet sum( R R
R� Rt t vvt vdR% ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_100� s ( R R R� ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.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 lagfromrootsR R R R t cosR= t piR- R R/ Rd ( R R& R" t rootsRf R% ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagfromroots� 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 i ( R R t lagrootsR R R= R� R ( R R" R% R& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt
test_lagroots� 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����( R RF R R R ( R Ru ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagtrim� s
c C` s# t t j d d � d d g � d S( Ni i i i����( R R t lagline( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_lagline� 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 Rd R@ ( R R" ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt
test_lag2poly 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 Rc R@ ( R R" ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt
test_poly2lag s c C` sE t j d d d � } t j | � } t j | � } t | | � d S( Ni i
i ( R R= t expR t lagweightR ( R R
R% R& ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt test_weight s ( R R R� R� R� R� R� R� R� ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyR� � s t __main__() t __doc__t
__future__R R R t numpyR t numpy.polynomial.laguerret
polynomialt laguerreR t numpy.polynomial.polynomialR t
numpy.testingR R R R R R RZ t L0t L1t L2t L3t L4t L5t L6R@ R R R R> R] Rm Rq R~ R� R� R� R ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_laguerre.pyt <module> s4 . "%( 9^c(0>3