ó ʽ÷Xc@`s%dZddlmZmZmZddlmZddlmZddl Z ddl Z ddl m Z m Z mZmZddlZddljZddlmZee dƒZd e fd „ƒYZd e fd „ƒYZd „Zd„Zd„Zd„Zd„Zd„Zd„Z dS(s6Test functions for the sparse.linalg.interface module i(tdivisiontprint_functiontabsolute_import(tpartial(tproductN(tTestCasetassert_t assert_equalt assert_raises(t interfacetmatmultTestLinearOperatorcB`s#eZd„Zd„Zd„ZRS(cC`stjdddgdddggƒ|_tjddgddgddggƒ|_tjddgddggƒ|_dS(Niiiiii(tnptarraytAtBtC(tself((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pytsetUps 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(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt&stmatvecc`s)tjˆjjƒ|ƒjˆjdƒS(Ni(R RtTtconjRR(R(R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR'strmatvecc`stjˆ|ƒS(N(R R(R(R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR,sc`stjˆjjƒ|ƒS(N(R RRR(R(R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR-sc`stjˆ|ƒS(N(R R(R(R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR.stmatmat(R(R((RsG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt get_matvecs#s  iiiii i ii iy@y(@y>@iúÿÿÿiñÿÿÿiic`sˆˆS(N(((R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRpsc`sˆdS(Ni(((R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRqsi2iqiGi\c`sˆˆS(N(((RR(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRsc`sˆdS(Ni(((R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR‚sii%i((RR tLinearOperatorRtargsRRR R RtmatrixRtHRtlent isinstancetndarrayt_ScaledLinearOperatort_SumLinearOperatortdtypetcomplex_Rt ValueErrorRRt_ProductLinearOperatorRt_PowerLinearOperator(RRtmatvecstztmatvecsAtmatvecsBtmatvecsCR((RRsG/tmp/pip-build-7oUkmx/scipy/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(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR–sRc`s/tjˆjjjƒ|ƒjˆjjdƒS(Ni(R RRRRRR(R(R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR—sRc`stjˆj|ƒS(N(R RR(R(R(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR™sRiiiiiiiii i(t TEST_MATMULtnosetSkipTestRRR RR R RtoperatorR RR)(RtDRRtb((RsG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt test_matmul‘s  (t__name__t __module__RR1R8(((sG/tmp/pip-build-7oUkmx/scipy/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((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pytmv±s )2cS`sXtjd|dd|dd|dd|dd|dd|dgd|ƒS( NiiiiiiiR'(R R (RR'((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pytrmv¸st BaseMatlikec`s eZd„Z‡fd†ZRS(cS`stj|ƒ|_d|_dS(Nii(ii(R R'R(RR'((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt__init__¾sc`sˆ||jƒS(N(R'(RR(R=(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt_matvecÂs(R9R:R@RA((R=(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR?½s t HasRmatvecc`seZ‡fd†ZRS(c`sˆ||jƒS(N(R'(RR(R>(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt_rmatvecÆs(R9R:RC((R>(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRBÅ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>(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt_adjointÊs (R9R:RE((R=R>(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRDÉs( tcasestappendR R R tsparset csr_matrixR R(R'R?RBRD(R(R=R>sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt make_casesªs777   tint32tfloat32tfloat64(RF(RRJ((RsG/tmp/pip-build-7oUkmx/scipy/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'( RFR taslinearoperatorRRRR R RR!RthasattrR'(RtMRtN((sG/tmp/pip-build-7oUkmx/scipy/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(RFR RNRRRR R (RRPRRQ((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyttest_dotôs.=0(R9R:RRRRS(((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR;¦s 3 cC`s>tjdddd„ƒ}t|ƒ}td|k|ƒdS(NRiRcS`sdS(Ni((R((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRssunspecified dtype(ii(R RtreprR(Rtrepr_A((sG/tmp/pip-build-7oUkmx/scipy/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 tarangeRtravelRR)R(tident((sG/tmp/pip-build-7oUkmx/scipy/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((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pytalways_four_oness"RiRR'RA(ii( R RNR RXRRR!RRO(RR^Rtop((sG/tmp/pip-build-7oUkmx/scipy/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((sG/tmp/pip-build-7oUkmx/scipy/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(RcRbRtsRtk((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt test_pickle#s c`sødtjfd„ƒY}tt|ƒdtjf‡fd†ƒY‰ˆdƒ}t|jdddgƒdddgƒtt|jdd d gƒd tjf‡fd †ƒY‰ˆtj j d dƒƒ}t|jtj j dƒƒj d ƒdS(NtEmptycB`seZRS((R9R:(((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRm/stIdentityc`s eZ‡fd†Zd„ZRS(c`s)tˆ|ƒjddd||fƒdS(NR'R(tsuperR@tNone(Rtn(Rn(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR@5scS`s|S(N((RR((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRA8s(R9R:R@RA((Rn(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRn4siiiiiit MatmatOnlyc`s eZ‡fd†Zd„ZRS(c`s,tˆ|ƒj|j|jƒ||_dS(N(RoR@R'RR(RR(Rr(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyR@@scS`s|jj|ƒS(N(RR(RR((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyt_matmatDs(R9R:R@Rs((Rr(sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyRr?s(i( R RRt TypeErrorRRtNotImplementedErrorRR trandomtrandnR(Rmtid3tmm((RnRrsG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyttest_inheritance.s  (cC`stjjddƒdtjjddƒ}tjjddƒ}tj|ƒ}tj|ƒ}||}||}t|jtjƒt|jtjƒdS(Niyð?( R RvtrandR RNRR'RMt complex128(t mat_complextmat_realtcomplex_operatort real_operatort sum_complextsum_real((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyttest_dtypes_of_operator_sumJs,  (!t__doc__t __future__RRRt functoolsRt itertoolsRR5R3t numpy.testingRRRRtnumpyR t scipy.sparseRHtscipy.sparse.linalgR ROR2R R;RVR[R`RRlRzRƒ(((sG/tmp/pip-build-7oUkmx/scipy/scipy/sparse/linalg/tests/test_interface.pyts&  " Ž\