ó 2ÄÈ[c@`s]dZddlmZmZmZddlZddlZddlZddlZ ddl m Z m Z m Z mZmZmZmZd„Zd„Zd„Zd„Zd „Zd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z!d„Z"d„Z#d„Z$de%fd„ƒYZ&d„Z'd„Z(dS(sƒTests of interaction of matrix with other parts of numpy. Note that tests with MaskedArray and linalg are done in separate files. i(tdivisiontabsolute_importtprint_functionN(tassert_t assert_equalt assert_raisestassert_raises_regextassert_array_equaltassert_almost_equaltassert_array_almost_equalcC`s±tjddgddggƒ}tt|dddgdd…ftjƒƒtjtjdƒjddƒƒ}t|dd…tjd ƒf|dd…d fj ƒdS( Niiiiii2ii iÿÿÿÿ( tnptmatrixRt isinstancetasmatrixtarangetreshapeRtarraytT(tmtx((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_fancy_indexings!/$cC`sbddg}ddg}tj||gƒ}tjjj|||ƒ}tt|tjƒƒdS(Niiii(R R t polynomialt polyutilst mapdomainRR (tdom1tdom2Rtres((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_polynomial_mapdomain!s   cC`sutjdddggƒ}tj|ddƒ}tjdddggƒ}t||ƒtt|ƒtjkƒdS(Niiitaxis(R R tsorttNoneRRttype(tatactualtexpected((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_sort_matrix_none+s  cC`sxtjdddggƒ}tj|dddƒ}tjdddggƒ}t||ƒtt|ƒtjkƒdS(NiiiR(R R t partitionRRRR(R R!R"((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_partition_matrix_none4s  cC`sotjddgdtƒ}tjddggdtƒ}ttj|dƒ|ƒttjd|ƒ|ƒdS(Niitdtypeii(R R tobjectRtdot(tarrtdesired((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyt%test_dot_scalar_and_matrix_of_objects>scC`sÄx½tjdtjddD]Ÿ}tjdd|ƒd }tjddgddggd|ƒ}tjdd gd d ggd|ƒ}ttj||ƒ|ƒttj||ƒ|ƒqWdS( Nt AllIntegertAllFloatt?iR&iiiii i ((R t typecodesRR Rtinner(tdttscaR)R*((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_inner_scalar_and_matrixGs #''cC`sotjddgdtƒ}tjddggdtƒ}ttj|dƒ|ƒttjd|ƒ|ƒdS(NiiR&ii(R R R'RR0(R)R*((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyt'test_inner_scalar_and_matrix_of_objectsQsc C`s¯tjddgddggƒ}tjdƒjddƒj}tj||dggdgdgddggƒ}tt|j dƒtjkƒtt|j dƒtj k ƒt |j dj d ƒtjdƒjdddƒ}t ttj||dggdgdgddggƒtj||dggdgdgdddggƒ}tt|j dƒtj kƒtt|j dƒtjk ƒt |j dj d ƒdS( Niiiitreadonlyt writeonlytallocatet no_subtype(ii(iii(R R RRRtnditerRRRtoperandstndarrayRtshapeRt RuntimeError(R tbti((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyt!test_iter_allocate_output_subtypeZs !     cC`s˜tjddgddggƒ}xptjtjtjfD]V}||ƒ}tt|ƒtjkƒ||dtƒ}tt|ƒtjk ƒq:WdS(Niiiitsubok(R R t zeros_liket ones_liket empty_likeRRtFalse(R t like_functionR>tc((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyRFus ! cC`såtjdddgdddggddƒ}|jdd td tƒ}t||kƒ|jd d tƒ}t||ƒtt|ƒtjƒ|jdd td tƒ}t||ƒt||k ƒtt|ƒtjk ƒdS( NiiiiiiR&tf4RAtcopyti4(R R tastypetTrueRERRR(R R>((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_array_astype€s-  cC`sAtjddgddggƒ}ttdtj||gƒdS(Niiiisshape too large to be a matrix(R R Rt ValueErrortstack(R((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyt test_stack“s! cC`sotjddgdtƒ}tjddggdtƒ}ttj|dƒ|ƒttjd|ƒ|ƒdS(NiiR&ii(R R R'Rtmultiply(R)R*((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_object_scalar_multiply›sc C`s­tjtjdƒƒ}x¯tjtjgD]›}||ddƒ}tt|tjƒƒt|jd kƒ||ddƒ}tt|tjƒƒt|jd kƒ||ƒ}ttj|ƒƒq+Wtj |d(sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyt4s(R RRRR RRRN(tselftab_rtab_c((R R>sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyt test_matrix's  %.cC`sXtjdddgdf}tt|ƒtjƒttj|ƒdddggƒdS(NRwiii(R RRRR R(RRw((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_matrix_scalar6scC`s tjdgƒ}tjdgƒ}tjdgƒ}tjdgƒ}tjd}tj||g||ggƒ}t||ƒtt|ƒt|ƒƒdS(Niiiis a, b; c, d(R RRtbmatRR(RR R>RGtdR!R"((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_matrix_builder;s ! (t__name__t __module__R’R“R–(((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyR‹%s  cC`s´y/ttjddgƒtjddgƒƒWnrtk r£}t|ƒ}|jddƒ}tjdƒ}yt||ƒWq°tk rŸt||ƒq°Xn Xtdƒ‚dS(Niisshapes (2L,), (1L, 2L)sshapes (2,), (1, 2)s| Arrays are not equal (shapes (2,), (1, 2) mismatch) x: array([1, 2]) y: matrix([[1, 2]])s Did not raise( RR RR Rtstrtreplacettextwraptdedent(tetmsgtmsg2t msg_reference((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyt%test_array_equal_error_message_matrixGs/   cC`stjddggƒ}tjdtjggƒ}tjdtj ggƒ}tjtjtjggƒ}tjddgtjtjggƒ}xjttfD]\}xS|||||fD]<}|||ƒtj|ƒ}|||ƒ|||ƒq½Wq¡WdS(Ngð?g@(R R RYtinfR RR(tm1tm2tm3tm4tm5t assert_funcRR ((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyttest_array_almost_equal_matrix]s'  ()t__doc__t __future__RRRtpytestR›RZtnumpyR t numpy.testingRRRRRRR RRR#R%R+R3R4R@RFRMRPRRRfRpRxR~R‚R‡RŠR'R‹R¡R©(((sE/tmp/pip-build-fiC0ax/numpy/numpy/matrixlib/tests/test_interaction.pyts8    4     )    "