K iAlddlmZddlmZejjZd dZd dZd dZdZdZ y) )mp)libmpc |dkDr td||j}tj|\}}|dkDrtd|td|||z|j z}tj dtj tjz}d}t|D]-}t|D]} |t|| |f|| |fz z }/|||zz}d} t|D])}t|dz|D]} | t|| |fz } +|dkDr td|| |dkDr td |||ksJ| dk(sJy) Nzoriginal matrix (hessenberg): Q: zH: 皙?rzdifference (H):B: ) printrowsr hessenbergtranspose_conjexplogepsxrangeabs) AverbosenQHBrerr0xyerr1s ]/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/mpmath/tests/test_eigen.pyrun_hessenbergr sb{ /3 A == DAq{ fQ fQ A  ""A &&rvvbff~% &C D AY) )A C!A#1Q3( (D )) AEMD D AY Aq! A C!A#K D  { t,{ fa #:: 199c ||dkDr td||j}tj|\}}|dkDrtd|td|||z|j z}||j z}tj dtj tjz}d}t|D]-} t|D]} |t|| | f|| | fz z }/|||zz}d} t|D]9} t|D])} | | k(r|| | fxxdzcc<| t|| | fz } +;| ||zz} d} t|D])} t| dz|D]} | t|| | fz } +|dkDrtd|| | |dkDr td|||ksJ| |ksJ| dk(sJy) Nrzoriginal matrix (schur): rzR: rrzdifference (S):r ) r r rschurrrrrrr) rrrrRrCrrrrrerr2s r run_schurr&.s{ *A. A 88A;DAq{ fa fa A  ""A A   A &&rvvbff~% &C D AY) )A C!A#1Q3( (D )) AEMD D AY  AAv!A#!  C!A#K D   AEMD D AY Aq! A C!A#K D  { tT2{ fa #:: #:: 199r c|dkDr td||j}tj|dd\}}}|dkDr$td|td|td|tjdtj tj z}d }t|D]x}||dd|fz|||dd|fzz } t|tj| }||ddf|z||ddf||zz } t|tj| }z|||zz}|d kDr td |||ksJy) Nrzoriginal matrix (eig): T)leftrightzE: zEL: zER: rrzdifference (E):) r r reigrrrrmaxmnorm) rrrEELERrrirs rrun_eigr1\s2{ (!, Aqt4IAr2{ fa gr gr &&rvvbff~% &C D AY& 1Q3K!A$AaC. (4!% qsGaK"QqS'AaD. (4!% & AEMD{ & #::r c Zd}tdD]}dttjdzz}tjdkDrodtj||zdz }tjdkDr|dz}t|D]'}t|D]}t|||f|||f<)ndtj||zdz ddtj||zdz zz}tjdkDrt|dz}t|D]a}t|D]Q}ttj |||fdttj |||fzz|||f<Sct||t||t||y) Nrrg?r y?r) rintrrand randmatrixreimrr&r1)vr0rrrrs r test_eig_dynr<|s A AY  BGGIM" " 779s?BMM!Q''!+Awwy3R-A#AY-!$QqsV!A#--R]]1a((1,a"--1:M6MPQ6Q0RRAwwy3RNA#AYN!$RUU1QqS6]!3b3ruuQqsV};M6M!M!A#NN qA&!q!Q) r cd}g}tjgdgdgdg}|j||j|jtjgdgdgdg}|j||j|jtjgdgdgdg}|j||j|jtjgdgd gd g}|j||j|j|D])}t || t || t || +y) Nr)r rr)rr r)rrr )r rr)r rr)rr r)rrr)rrr)rrrr5)rmatrixappend transposerr&r1)r;ASrs rtest_eigrBs! A B 9 AIIaLIIakkm 9 AIIaLIIakkm 9 AIIaLIIakkm 9 AIIaLIIakkm  qA&!q!Q r N)r) mpmathrrbackendrrr&r1r<rBr rrFs8   "J,\@ 0 r