B +[*@sdddlZddlTddZddZddZd d Zd d Zd dZddZddZ ddZ ddZ dS)N)*c sftd}xtdD]}d|||f<qW|tdks6t|t|ksFttddjrZttdddgdddgddd gg}t|ttdd kstd |d <d |jksttdddgdddggtdd gddgd dgg}|tddgddggkst||||kr|ks"ntttfddd ddgdd dgdddgg}t|}| |kslt|t t |kstt|t d}|t t |kst|dt t |dkst|d d |kr|ksntj dkstjdkstd|_ d|_t|jdks(tdks>tttfddt||d ksjttddgddgddgddggtd dg}|tddddgksttd}t|ddtdd kstdtdddtdksttdd tdkstttfd dtd}t|} d!|d"<|| ksTtt|sbtdS)#N r)rr i'icsS)Nr)A2r9lib/python3.7/site-packages/mpmath/tests/test_matrices.pysz#test_matrix_basic..(<FPZ)Z force_typey?csS)Nrr)rA4rrr#sicsdS)Nrrr)A7rrr,si)rr)matrixrangeeyeAssertionError _matrix__datalistpytestraises ValueErrortolistevalreprfloatrowscolslensumoneszeros randmatrixZnstr) A1iZA3ZA5lZA6xZA8ZA9ZA10r)rrrrtest_matrix_basicsZ    & &  ""r6c s*tdddgdddgddd ggtdddddg}ddddfksNtdddftdgdgdggksvtdddftddd ggkstddddftddgdd ggkst|ddtddgkstttfd d td}|ddddf<|ddddftdddgdddgddd ggksLttd d dgg|dddf<|td d dgdddgddd ggksttdgdgdgg|dddf<|td d dgdddgdddggksttddgddgg|ddddf<|tdddgdddgdddggks*td |dd<|tdd d ddgksRttt"dddf|dddf<WdQRXtt&ddddf|dddf<WdQRXd |dddf<|tddd gddd gddd ggkstd|ddddf<x|D]}|dkstqWdS)Nrrrrrrr r r csddddfS)Nrrrr)Arrr?sz$test_matrix_slices..r rr rr)rr!r$r% IndexErrorr&)Vr2r5r)r7rtest_matrix_slices3sP ($,   $   & *  rBcCsptddgddgg}|d||ks(t|d|||ks@t|dt|ksTt|dt||ksltdS)Nrrrrr )rr!Zinverse)r7rrrtest_matrix_poweres rDcCstddgddgddgg}|j|krHtdddgdddggksNntt|dd|tddgddgddggks|tddg}t|dd|ddgkstttddddgtddddgddddgddddggkstdS)Nrrrrrrr)rTZ transposer!Zswap_rowextendr )r7r4rrrtest_matrix_transformls4 " rGcCstdtdgdtgg}|ttdddgdtddggksDt||jkrzttdddgdtddggksntdS)Nrrrr )rj conjugateZmpcr!Ztranspose_conjH)r7rrrtest_matrix_conjugatevs, rKcCstdddgtdddgdddgdddggks2ttdd}|jdkrP|jdksTtx|D]}|dksZtqZWtdd}|jdkr|jdkstx|D]}|dkstqWtdtdksttd}t dt||d|dg|d|d|dg|d|d|dggkstdS)Nrrrrr rr) Zdiagrr!r/r+r,r0r1ZmpfZhilbert)r2arZonerrrtest_matrix_creation|s2    rMcCstddgddgddgg}t|ddks,tt|tdks>tt|dtd ksTttdd ksdtddd d g}t|dd ksttt|dddksttt|dddkstt|tdkstdS)NrrCrr rrrFrrr ir gPz$,@g %^(@r)rZmnormr!infZsqrtZnormround)r7r5rrr test_normss rRcCstdddddg}|tdgdgdgdgdggks6t|ddksFtt|jdksXtt|ttdksptd|d<d|d<|ddkstt|t|jkrdksnt|j|tdggkstdS) Nrrrrrrir)rr!r-r"r#rrE)r5rrr test_vectors$$rTcCs4td}|}||kstd|d<||ks0tdS)Nrr)rr)r/copyr!)r7Brrrtest_matrix_copys  rWcCsZy ddl}Wntk r dSXddgddgddgg}||}t|t|ksVtdS)Nrrrrrrr)numpy ImportErrorZarrayrr!)rXr4rLrrrtest_matrix_numpys  rZ) r$Zmpmathr6rBrDrGrKrMrRrTrWrZrrrrs/2