ó î&]\c@`sAdZddlmZmZmZddlmZddlmZddl Z ddl Z ddl m Z m Z ddlmZmZddlZddljZddlmZee d ƒZd efd „ƒYZd efd „ƒYZd„Zd„Zd„Zd„Zd„Z d„Z!d„Z"d„Z#d„Z$dS(s6Test functions for the sparse.linalg.interface module i(tdivisiontprint_functiontabsolute_import(tpartial(tproductN(traisestwarns(tassert_t assert_equal(t interfacetmatmultTestLinearOperatorcB`s#eZd„Zd„Zd„ZRS(cC`stjdddgdddggƒ|_tjddgddgddggƒ|_tjddgddggƒ|_dS(Niiiiii(tnptarraytAtBtC(tself((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt setup_methods c `s‡ d„}xá||jƒD]Ð}tj|‰tˆjdkƒtˆjtjdddgƒƒddgƒtˆjtjdgdgdggƒƒdgdggƒtˆtjdddgƒddgƒtˆtjdgdgdggƒdgdggƒtˆj tjdddgƒƒddgƒtˆj tjdgdgdggƒƒdgdggƒtˆjtj dgdgdggƒƒdgdggƒtˆtj dgdgdggƒdgdggƒtˆj tj dgdgdggƒƒdgdggƒtdˆdddgddgƒtdˆj ddgƒd dd gƒtdˆj jddgƒd dd gƒtdˆdgdgdggdgdggƒtdˆj dgdgdggƒdgdggƒtˆddddgddgƒtˆddgdgdggdgdggƒtd ˆdddgd d gƒtˆˆdddgddgƒtˆˆj ddgƒd dd gƒtˆˆj jddgƒd dd gƒtˆˆdgdgdggdgdggƒtˆˆj dgdgdggƒdgdggƒtˆ dddgddgƒtˆ dgdgdggdgdggƒtˆˆdddgddgƒtˆˆdgdgdggdgdggƒˆˆ}tt|jƒdkoZ|jdˆkoZ|jdˆkƒdˆ}tt|jƒdko£|jdˆko£|jddkƒttˆjdddgƒtjƒƒttˆjtjdgdgdggƒƒtjƒƒttˆtjdddgƒtjƒƒttˆtjdgdgdggƒtjƒƒttˆj tjdddgƒƒtjƒƒttˆj tjdgdgdggƒƒtjƒƒttˆjtj dgdgdggƒƒtjƒƒttˆtj dgdgdggƒtjƒƒttˆj tj dgdgdggƒƒtjƒƒttdˆtjƒƒttd ˆtjƒƒttˆˆtjƒƒttˆ tjƒƒttˆˆtjƒƒtd ˆjtjkƒttˆjtjddgƒƒttˆjtjddddgƒƒttˆjtjdgdggƒƒttˆjtjdgdgdgdggƒƒtt‡fd†ƒtt‡fd†ƒqWx®t||jƒ||jƒƒD]‹\}}tj|‰tj|‰tˆˆddgddgƒtˆˆdgdggdgdggƒtˆˆj dgdggƒdgdggƒtˆˆj ddgƒddgƒtˆˆj jddgƒddgƒttˆˆtjƒƒtt‡‡fd†ƒtt‡fd†ƒˆˆ}tt|jƒdko– |jdˆko– |jdˆkƒq Wxâ||jƒD]Ñ}tj|}t|dddgddgƒt|dj ddgƒddgƒt|dj jddgƒddgƒt|dj dgdggƒdgdggƒtt|dtjƒƒq® WdS(Nc`sniˆjd6‡fd†d6‡fd†d6iˆjd6‡fd†d6‡fd†d6‡fd†d 6gS( Ntshapec`s tjˆ|ƒjˆjdƒS(Ni(R tdottreshapeR(tx(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt%stmatvecc`s)tjˆjjƒ|ƒjˆjdƒS(Ni(R RtTtconjRR(R(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR&strmatvecc`stjˆ|ƒS(N(R R(R(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR+sc`stjˆjjƒ|ƒS(N(R RRR(R(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR,sc`stjˆ|ƒS(N(R R(R(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR-stmatmat(R(R((RsGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt get_matvecs"s  iiiii i ii iy@y(@y>@iúÿÿÿiñÿÿÿiic`sˆˆS(N(((R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRosc`sˆdS(Ni(((R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRpsi2iqiGi\c`sˆˆS(N(((RR(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR€sc`sˆdS(Ni(((R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRsii%i((RR tLinearOperatorRtargsRRR R RtmatrixRtHRtlent isinstancetndarrayt_ScaledLinearOperatort_SumLinearOperatortdtypetcomplex_t assert_raisest ValueErrorRRt_ProductLinearOperatorRt_PowerLinearOperator(RRtmatvecstztmatvecsAtmatvecsBtmatvecsCR((RRsGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt test_matvec!s˜ .=)8.==8=$),38$3$$),38!0$3 ? ?(:,51::5:"((4!-2&) C!&)2c`ststjdƒniˆjjd6‡fd†d6‡fd†d6‡fd†d6}tj|}tjd d d gd d dgdddggƒ}|d}t t j ||ƒ||ƒt t j ||ƒ||ƒt t t j |d ƒt t t j d |ƒdS(Ns$matmul is only tested in Python 3.5+Rc`s&tjˆj|ƒjˆjjdƒS(Ni(R RRRR(R(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR•sRc`s/tjˆjjjƒ|ƒjˆjjdƒS(Ni(R RRRRRR(R(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR–sRc`stjˆj|ƒS(N(R RR(R(R(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR˜sRiiiiiiiii i(t TEST_MATMULtpytesttskipRRR RR R RtoperatorR R)R*(RtDRRtb((RsGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt test_matmuls  (t__name__t __module__RR2R9(((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR s otTestAsLinearOperatorcB`s#eZd„Zd„Zd„ZRS(c`s:gˆ_‡fd†}|dƒ|dƒ|dƒdS(Nc`sAˆjjtjdddgdddggd|ƒƒˆjjtjdddgdddggd|ƒƒˆjjtjdddgdddggd|ƒƒd„‰d „‰d tjf‡fd †ƒY}d |f‡fd †ƒY}d|f‡‡fd†ƒY}ˆjj||ƒƒˆjj||ƒƒdS(NiiiiiiR'cS`s‰tjd|dd|dd|dd|dd|dd|dgd|ƒ}t|jƒdkr…|jd dƒ}n|S( NiiiiiiiR'iÿÿÿÿ(R R R"RR(RR'ty((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pytmv°s )2cS`sXtjd|dd|dd|dd|dd|dd|dgd|ƒS( NiiiiiiiR'(R R (RR'((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pytrmv·st BaseMatlikec`s eZd„Z‡fd†ZRS(cS`stj|ƒ|_d|_dS(Nii(ii(R R'R(RR'((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt__init__½sc`sˆ||jƒS(N(R'(RR(R>(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt_matvecÁs(R:R;RARB((R>(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR@¼s t HasRmatvecc`seZ‡fd†ZRS(c`sˆ||jƒS(N(R'(RR(R?(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt_rmatvecÅs(R:R;RD((R?(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRCÄst HasAdjointc`seZ‡‡fd†ZRS(c `si|jd|jdf}tˆd|jƒ}tˆd|jƒ}tjd|d|d|jd|ƒS(NiiR'RRR(RRR'R R(RRRR(R>R?(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt_adjointÉs (R:R;RF((R>R?(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyREÈs( tcasestappendR R R tsparset csr_matrixR R(R'R@RCRE(R(R>R?sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt make_cases©s777   tint32tfloat32tfloat64(RG(RRK((RsGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR¦s  ,  cC`s–x|jD]„}tj|ƒ}|j\}}t|jtjdddgƒƒddgƒt|jtjdgdgdggƒƒdgdggƒt|tjdddgƒddgƒt|tjdgdgdggƒdgdggƒt|jtjddgƒƒdddgƒt|jtjdgdggƒƒdgdgdggƒt|j jtjddgƒƒdddgƒt|j jtjdgdggƒƒdgdgdggƒt|j tjdd gdd gdd ggƒƒddgdd ggƒt|tjdd gdd gdd ggƒddgdd ggƒt |d ƒr t|j |j ƒq q WdS(Niiiii i i iiiiiMR'( RGR taslinearoperatorRRRR R RR!RthasattrR'(RtMRtN((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt test_basicÙs".=)8.=1@0GcC`síxæ|jD]Û}tj|ƒ}|j\}}t|jtjdddgƒƒddgƒt|jtjdgdgdggƒƒdgdggƒt|jtjddgddgddggƒƒddgdd ggƒq WdS( Niiiii iiiiM(RGR RORRRR R (RRQRRR((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyttest_dotós.=0(R:R;RRSRT(((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR<¥s 3 cC`s>tjdddd„ƒ}t|ƒ}td|k|ƒdS(NRiRcS`sdS(Ni((R((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRssunspecified dtype(ii(R RtreprR(Rtrepr_A((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt test_reprs cC`stjdƒ}t|dddgdddgƒt|jtjdƒjddƒƒjƒtjdƒƒtt |j ddddgƒdS(Niiii i(ii( R tIdentityOperatorRRR tarangeRtravelR)R*R(tident((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt test_identitys#:cC`s»tjtjdƒjddƒƒ}d„}tjdd d|ƒ}xo|||||j|||||dgD]?}tt|dƒƒtt|dƒƒtt|dƒƒqtWdS( NiicS`s>tj|ƒ}t|jdkp-|jdkƒtjdƒS(Niii(i(ii(R tasarrayRRtones(R((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pytalways_four_oness"RiRR'RB(ii( R ROR RYRRR!RRP(RR_Rtop((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyttest_attributess$ 5cC`s tjdƒS(s> Needed for test_pickle as local functions are not pickleable i(R tzeros(R((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRscC`s—ddl}x„t|jdƒD]o}tjdtƒ}|j|d|ƒ}|j|ƒ}x0|jD]%}t t ||ƒt ||ƒƒqfWq WdS(Niiitprotocol(ii( tpickletrangetHIGHEST_PROTOCOLR RRtdumpstloadst__dict__Rtgetattr(RdRcRtsRtk((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt test_pickle"s c`sdtjfd„ƒY}ttddƒtt|ƒWdQXdtjf‡fd†ƒY‰ˆdƒ}t|jdd dgƒdd dgƒtt|j d d d gƒd tjf‡fd†ƒY‰ˆt j j d dƒƒ}t|jt j j dƒƒj dƒdS(NtEmptycB`seZRS((R:R;(((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRn.stmatchsshould implement at leasttIdentityc`s eZ‡fd†Zd„ZRS(c`s)tˆ|ƒjddd||fƒdS(NR'R(tsuperRAtNone(Rtn(Rp(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRA5scS`s|S(N((RR((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRB8s(R:R;RARB((Rp(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRp4siiiiiit MatmatOnlyc`s eZ‡fd†Zd„ZRS(c`s,tˆ|ƒj|j|jƒ||_dS(N(RqRAR'RR(RR(Rt(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRA@scS`s|jj|ƒS(N(RR(RR((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyt_matmatDs(R:R;RARu((Rt(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyRt?s(i(R RRtRuntimeWarningR)t TypeErrorRRtNotImplementedErrorRR trandomtrandnR(Rntid3tmm((RpRtsGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyttest_inheritance-s (cC`stjjddƒdtjjddƒ}tjjddƒ}tj|ƒ}tj|ƒ}||}||}t|jtjƒt|jtjƒdS(Niyð?( R RytrandR RORR'RNt complex128(t mat_complextmat_realtcomplex_operatort real_operatort sum_complextsum_real((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyttest_dtypes_of_operator_sumJs,  c`sBdg‰‡fd†}tjdd|ƒ}tˆddƒdS(Nic`sˆdcd7<|S(Nii((tv(t call_count(sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyR\siRi(ii(R RR(RR((RˆsGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyttest_no_double_initYs cC`s˜tjdggƒ}tj|ƒ}d|}d|}tjdgƒ}t|j|ƒ|j|ƒƒt|jj|ƒ|jjƒj|ƒƒdS(Nyð?i( R R R RORRR!RR(tXRRtYR‡((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyttest_adjoint_conjugatees  (%t__doc__t __future__RRRt functoolsRt itertoolsRR6R4RR)Rt numpy.testingRRtnumpyR t scipy.sparseRItscipy.sparse.linalgR RPR3tobjectR R<RWR\RaRRmR}R†R‰RŒ(((sGlib/python2.7/site-packages/scipy/sparse/linalg/tests/test_interface.pyts,   Ž\