Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyo
�
� 9Zc
@` sF 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 m Z m
Z
m Z m Z m Z d � Z d g Z d d g Z d d d g Z d d d d
g Z d d d d d g Z d d
d d d d g Z d d d d d d d g Z d d d d d d d d g Z d d d d d d d d d g Z 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 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% d0 k rBe � n d S(1 s Tests for polynomial module.
i ( t divisiont absolute_importt print_functionN( t TestCaset assert_almost_equalt
assert_raisest assert_equalt assert_t run_module_suitec C` s t j | d d �S( Nt tolg���ư>( t polyt polytrim( t x( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt trim
s i i����i i����i i����i i i��i i i���i i����i8 i����i@ i��i� i ���i� i i����i� i����i 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
polydomain( t self( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polydomain s c C` s t t j d g � d S( Ni ( R R
t polyzero( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt
test_polyzero# s c C` s t t j d g � d S( Ni ( R R
t polyone( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyone& s c C` s t t j d d g � d S( Ni i ( R R
t polyx( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt
test_polyx) s ( t __name__t
__module__R R R R ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyR s t TestArithmeticc B` s5 e Z d � Z d � Z d � Z d � Z d � Z RS( 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 polyaddR R
( R t it jt msgt tgtt res( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyadd/ 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 polysubR R
( R R! R" R# R$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polysub9 s .c C` s� t t j d g � d g � t t j d g � d d g � xY t d d � D]H } d g | d g } d g | d d g } t t j | � | � qK Wd S( Ni i i ( R R
t polymulxR ( R R! t serR$ ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt
test_polymulxC s c C` s� x� t d � D]� } x� t d � D]� } d | | f } t j | | d � } | | | 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 R ( R R R R
t polymulR R
( R R! R" R# R$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polymulK s .c
C` sL t t t j d g d g � t j d g d g � \ } } t | | f d � t j d d g d g � \ } } t | | f d � x� t d � D]� } x� t d � D]� } d | | f } d g | d d g } d g | d d g } t j | | � } t j | | � \ } } t j t j | | � | � } t | | d | �q� Wq� Wd S(
Ni i i i s
At i=%d, j=%dR ( i i ( i i ( ( i i i ( R t ZeroDivisionErrorR
t polydivR R R R, (
R t quot remR! R" R# t cit cjR$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polydivT s ! ( R R R& R( R+ R- R4 ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.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
j 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 c C` s| t t j g d g � j d � t j d d � } g t d � D] } | | ^ q> } xK t d � D]= } | | } t j | d g | d g � } t | | � qa W| | d d } t j | d d d d g � } t | | � x� 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 i i (
R R
t polyvalt sizeR t linspaceR R R t shape( R R R! t yR$ R% t dims( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyvalt 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: R t
ValueErrorR
t polyval2dt c2dR R t onesR R9 (
R t x1t x2t x3t y1t y2t y3R$ R% t z( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyval2d� 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: R R= R
t polyval3dt c3dR R R@ R R9 (
R RA RB RC RD RE RF R$ R% RG ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyval3d� 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
polygrid2dR? R R@ R R9 (
R RA RB RC RD RE RF R$ R% RG ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polygrid2d� 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 RL R
t
polygrid3dRJ R R@ R R9 (
R RA RB RC RD RE RF R$ R% RG ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polygrid3d� s
( i i ( R R R t arrayt c1dRL R? RJ t randomR R
R6 R: R< RH RK RN RP ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyR5 j s t TestIntegralc B` s e Z d � Z d � Z RS( c C` sJ 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]z } | d } d g | d g } | g d g | d | g } t j | d d d | g �} t t | � t | � � q� Wxo t d � D]a } | d } d g | d g } t j | d d d | g d d �} t t j d | � | � qOWx� t d � D]� } | d } d g | d g } | g d g | d | g } t j | d d d | g d
d �} 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 | � � qhWqRWx� 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 | � � 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 t | � � d d �} t t | � t | � � q�Wq�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 t | � � d
d �} t t | � t | � � q�Wq�Wd S( Ni g �?i����i i i t mt kt lbndt scl( R R= R
t polyintR R R
R6 t list( R R! RV R% RX t polR$ R" ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyint� sp "
$
$ !"'!(-!(-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 RV ( i i ( R RS t vstackt TR
RY R ( R R? t cR$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyint_axis s 1
+
1( R R R\ Ra ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyRT � s Kt 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 RU i RX ( R R= R
t polyderR R R
RY R ( R R! R$ R% R" ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyder* 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 R] i i ( i i ( R RS R^ R_ R
Rc R ( R R? R` R$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyder_axisC s 1
+( R R Rd Re ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyRb ( 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
polyvanderR R9 R R R6 RQ ( R R t vR! t coef( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyvanderT 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 RS R
t polyvander2dR> t dott flatR R R9 ( R RA RB RC R` t vanR$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyvander2de 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 RS R
t polyvander3dRI Rm Rn R R R9 ( R RA RB RC R` Ro R$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyvander3dr s !
*( i i ( R R R RS R Rk Rp Rr ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyRf P 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 R= R
t
polycompanion( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.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
Rt R9 ( R R! Rj ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.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
Rt ( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_linear_root� s ( R R Ru Rv Rw ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyRs � s t TestMiscc B` s5 e 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 | } t j | � d | d } t t | � t | � � q5 Wd S( Ni i i i (
R
t
polyfromrootsR R
R R t cosR8 t pit Tlist( R R% R! t rootsR$ ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyfromroots� s 7
c 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 g �i i����( R R
t polyrootsR R R8 Ry R
( R R! R$ R% ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyroots� s c C` s� d � } 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 � t j d d � } | | � } t j | | d
� } t t j
| | � | � t j | | d d d
g � }
t t j
| |
� | � t | |
� d S( Nc S` s | | d | d S( Ni i ( ( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt f� s c S` s | d | d d S( Ni i i ( ( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt f2� s i i����i i t wi i i i y �?y �( R R= R
t polyfitt TypeErrorR R8 R t lenR R6 RQ R_ t
zeros_liket copy( R R� R� R R: t coef3t coef4t coef2dR� t ywt wcoef3t wcoef2dt coef1t coef2( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt test_polyfit� sj " %""++"( !$' 3
'
- 9 "( 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 R= R
R R ( R Rj ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt
test_polytrim� s
c C` s# t t j d d � d d g � d S( Ni i ( R R
t polyline( R ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt
test_polyline� s ( R R R~ R� R� R� R� ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyRx � s
F t __main__(& t __doc__t
__future__R R R t numpyR t numpy.polynomial.polynomialt
polynomialR
t
numpy.testingR R R R R R R
t T0t T1t T2t T3t T4t T5t T6t T7t T8t T9R| R R R5 RT Rb Rf Rs Rx R ( ( ( sY /opt/alt/python27/lib64/python2.7/site-packages/numpy/polynomial/tests/test_polynomial.pyt <module> s4 . !$$=`^(0h
Mini Shell