o "`@sddlZddlZddlTddZddZddZd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZejddZddZdd ZdS)!N)*c s<td}tdD]}d|||f<q|tdksJ|t|ks!Jtddjr+Jtgdgdgdg}t|ttddksEJd|d <d |jvsPJtgdgdgtd d gddgdd gg}|td dgddggkswJ||||kr|ksJJttfddgdgdgdg}t|}||ksJ|t t |ksJt |}|t t |ksJ|dt t |dksJ|dd|kr|ksJJj dksJj dksJd|_ d|_ t|jdksJdks Jttfddt||dks"Jtddgddgdd gddggtddg}|tgdksEJtd}t|ddtddksZJdtdddtdkskJtddtdksxJttfddtd}t|} d|d <|| ksJt|sJdS)!Nrrr r)rrr i'csSNr)A2rz#test_matrix_basic..)r)(r<)FPZy?csSrrr)rA4rrr$rrr r r )r#r#r#r#csdS)Nrrr)A7rrr-rirr)matrixrangeeye _matrix__datalistpytestraises ValueErrortolistevalreprfprowscolslensumoneszeros randmatrixZnstr) A1iZA3A5lZA6xZA8ZA9ZA10r)rr"r$rtest_matrix_basicsZ    $ $  ""r>cCsNtjdkr tdtgdgdg}tddgddgd d gg}td d S) z Test the PEP465 "@" matrix multiplication syntax. To avoid syntax errors when importing this file in Python 3.4 and below, we have to use exec() - sorry for that. iz9'@' (__matmul__) is only supported in Python 3.5 or newerrrr rrrrrzassert A4 @ A5 == A4 * A5N)sys hexversionr+skipr&exec)r"r;rrr test_matmul4s   rCcstgdgdgdgtgd}ddddfks!Jdddftdgdgdggks5JdddftgdgksFJdd dd ftdd gdd ggks\J|dd td d gksjJttfd dtd }|ddddf<|ddddftgdgdgdgksJtgdg|dddf<|tgdgdgdgksJtdgdgdgg|dddf<|tgdgdgdgksJtddgddgg|ddddf<|tgdgdgdgksJd|dd <|tgdksJttdddf|dddf<Wdn 1s-wYttddddf|dddf<Wdn 1sSwYd|dddf<|tgd gd!gd"gksrJd#|ddddf<|D] }|d#ksJq~dS)$Nrrr )rrrrr rrr r rr rrcsddddfS)Nrr rrArrrMsz$test_matrix_slices..)r rr )rrFrG)rr rH)r r rI)rJrKrG)rLrMrHr)rrrrr r)rJrKr)rLrMr)r r rr)r&r+r, IndexErrorr-)Vr9r=rrDrtest_matrix_slicesAsb (",     $    " rPcCsptddgddgg}|d||ksJ|d|||ks J|dt|ks*J|dt||ks6JdS)Nrrrrr)r&ZinverserDrrrtest_matrix_powerss rRcCstddgddgddgg}|j|kr#tgdgdgks&JJt|dd|tddgddgddggks=Jddg}t|d d|ddgksOJttdgd tgd gd gd gksgJdS)Nrrrrr r )rrr )rrr rr)rrrr)rrrr)rrrr)r&TZ transposeZswap_rowextendr()rEr<rrrtest_matrix_transformzs2 " 4rUcCstdtdgdtgg}|ttdddgdtddggks"J||jkr>ttdddgdtddggksAJJdS)Nrrrr)r&j conjugateZmpcZtranspose_conjHrDrrrtest_matrix_conjugates ,  rYcCstgdtgdgdgdgksJtdd}|jdkr$|jdks&J|D]}|dks0Jq(tdd}|jdkr@|jdksBJ|D]}|dksLJqDtd td ksWJtd}tdt||d|dg|d|d|d g|d|d |d ggksJdS) Nr)rrr)rrr)rrrrrrrrrr ) Zdiagr&r6r2r3r7r8mpfZhilbert)r9arZonerrrtest_matrix_creations*  r\cCstddgddgddgg}t|ddksJt|tdksJt|dtd ks*Jtdd ks2Jgd }t|dd ks?Jtt|dd dksKJtt|d d dksWJt|tdks`JdS)NrrQrrrr rFrr)rrQr rrgPz$,@g %^(@r)r&ZmnorminfZsqrtZnormround)rEr=rrr test_normssracCstgd}|tdgdgdgdgdggksJ|ddks Jt|jdks)Jt|ttdks5Jd|d<d|d<|ddksEJt|t|jkrUdksXJJ|j|td ggkseJdS) N)rrrrrrrrrrr r#r)r&r4r)r*r'rS)r=rrr test_vectors $&rccCsZtd}|}| }||ksJ||ksJd|d<||ks!Jd|d<||ks+JdS)Nr rr%*)r6copy)rEBCrrrtest_matrix_copys   rhcCsZzddl}Wn tyYdSwddgddgddgg}||}t|t|ks+JdS)Nrrrrrr r )numpy ImportErrorZarrayr&)rir<r[rrrtest_matrix_numpys   rkc Cstdd}||d}td}||d}d}t|}||d}t|}td}||||||||fD]} | |t| gksDJ|| t| gksPJq6dS)z/Multiplication of iv.matrix and any scalar typerry@rdgX9v?y@N)mpirZr1convertivr6r&) r[bcdefghMr=rrr test_interval_matrix_scalar_mults       rwcCs|td}td}td}|||fD](}||t|ks J|||ks(J||t|ks3J|||ks;JqdS)z2Multiplication of iv.matrix and other matrix typesrN)r6r1rnr&)rErfrvXrrr test_interval_matrix_matrix_mults  rycCshtdtdtdfD]$}t|}td}t|dt|dks'J||ks1Jq dS)Nrr%)r(r1rnr&typer.)Zother_type_eyerErfrrrtest_matrix_conversion_to_ivs   r{cCs"td}t|gg}t|}t|gg}||ksJ||}||}||ks*J|djdks3J|djdksrCrPrRrUrYr\rarcrhrkrwZmarkZxfailryr{r|rrrrs&/ 2