K i{ddlmZmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZmZddlmZddlmZdd lmZmZmZmZdd lmZed \ZZZdd Zd ZdZ dZ!dZ"dZ#y))IRationaloopi)S)symbols)exp)sqrt)cossin) integrate)simplify)R_nlE_nl E_nl_diracPsi_nlm)raiseszn r Zct|}t|}t||z |kryt|t|kDrt||z |z }||kSt||z |z }||kS)NT)floatabs)abmax_relative_errormax_absolute_errorrelative_errors g/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sympy/physics/tests/test_hydrogen.pyfeqrsu aA aA 1q5z&& 1vAa!eQY / //a!eQY / //cdtz }dtd|dzz ztt |z ztdd|dzzz tt d|zz zdtd|zz z ztj tdd|dzzz ztt d|zz ztz|z t ddtdd|dzzz ztt d|zz zddtzd|zz z t ddt|z dzzzzt ddtdd|dzzz ztt d|zz zdtd|zz z ztz|z t ddtdd|dzzz ztt d|zz zt|z dzzt ddtd|dzz ztt d|zz zddtzd|zz z t dd t|z dzzzt dd t|z dzzz zt dd td d|dzzz ztt d|zz zdtd|zz z t dd t|z dzzzzt|z zt ddtdd |dzzz ztt d|zz zdtd|zz z zt|z dzzt ddtdd|dzzz ztt d|zz zt|z dzzd }|D]1\}}tt||tt|||fz dk(r1Jy)NQP@ i#) )r r)r!r)r!r )r"r)r"r )r"r!)r%r)r%r )r%r!)r%r"r) Zr r rrHalfrrr)rRnls rtest_wavefunctionr6sT !A$qAv,aRT*Q!Q$Z 3r1Q3x=0A1Q3K@a1a4j))CAaCM9A=a?Aa1a4j!11CAaCMA QqS!A#Y!RAaC!83 35B$q!AqD&z"22S!QqS]B Q!WB$q"QT'{"33c1"ac(mCqsQhNAa1f-QB!H = QqS!A#Y!Q1Q3(2 2Xa5E1q5P PRB$q!AqD&z"22S!QqS]B Q!Wx2!A#1 1367c;B$q!AqD&z"22S!QqS]B Q1XA#"C 42ad7 #44sA2qs8}D!axO A";1Q1a(1aV945:::;rc d}t|dzD]J}t|D]:}tt||tdztdzztdtfdk(r:JLy)Nr!r r)ranger rr1r)n_maxr4r5s r test_normr:4sg E 519 Gq GAT!Q]A-14q!RjAQF FF GGrc td}td}td}tddd|||t| ttz k(sJtddd|||tj t| dz z|zt |tt |zzdttzz zk(sJtd dd|||dtd t |ztt|zzt|zdttzz tdzd z td ztd |zd z z|dzdzzk(sJy) Nr1phithetar rr!r%r"r&r*) rrr r rr2r rr )r1r<r=s r test_psi_nlmr@;sSfA %C G*E Aq!QU +sA2wb/A AB A Aq"ae ,#qbAh-1G!1K u:QBH %T"X 628 88 8 Aq!QUA .47(SZ2G q3w<3e*3%()DH 3689!3=?A3B z 3 a1.3/23a%A3> >? >rcttttdz dtdzzz k(sJttddtdzzz k(sJtddtddz dz k(sJtddtddz dz k(sJtdtj dz k(sJtdtj dz k(sJtdtj dz k(sJtdtj d z k(sJtd tj d z k(sJt t d y) Nr!r>r /r(r"r% di NctdSNr)rrrz(test_hydrogen_energies..Ss tAwr)rr4r0rOner ValueErrorrHrrtest_hydrogen_energiesrLFs# 1:!Q$!Q$ '' ' 7b!AqD&k !! ! 2;1R5!8)V, ,, , 2;1R5!8)V, ,, , 7quuffo %% % 7quuffo %% % 7quuffo %% % 7quuffo %% % 9) )) ) :'rc tdddddtdz dz k(sJttdddddtdzdztdtdzdzz d z z dk(sJttddddd tdzd ztd tdzd zz dz z dk(sJttddddddtdzztddtdzzz dz z dk(sJttddddddtdzztddtdzzz dz z dk(sJtddD]_}t|D]O}t t||t |ddsJ|dkDs,t t||dt |ddrOJad}tddD]d}t|D]T}t t|||t ||ddsJ|dkDs/t t||d|t ||ddrTJfd}tddD]d}t|D]T}t t|||t ||ddsJ|dkDs/t t||d|t ||ddrTJft td t td!t td"y)#Nr!rr )r0cr(r"r*rDr%6r&l iKidiiB)i9iQIRiY<'i /ixNr+gh㈵>F)r0g-C6?gMbP?ctddSrGrrHrrrIz.test_hydrogen_energies_relat..|s z!Q/rctddS)Nr r>rVrHrrrIz.test_hydrogen_energies_relat..}s z!R0rctdddS)Nr rFrVrHrrrIz.test_hydrogen_energies_relat..~sz!Q6r)rr rr8rrrrK)r4r5r0s rtest_hydrogen_energies_relatrYVsR aa1 %47Q 66 6 Jq!qA.Ad1gINr$q'zB'4(*+4,- .12 33 3 Jq!qA.BtAwJOs47{S()4*,-4./ 034 55 5 Jq!qC0i( T C79x$t*'<<=6>@E6FG HKL MM M Jq!rS1y7 U D8IU (;;<7=?D7EF GJK LL L 1a[Iq IAz!Q'a$= ==1u:aE2DGT4HHH II A 1a[Oq OAz!Q!,d1aj$E EE1u:aE15tAqz4NNN OO A 1a[Oq OAz!Q!,d1aj$E EE1u:aE15tAqz4NNN OO :/0 :01 :67rN)-q=rZ)$sympy.core.numbersrrrrsympy.core.singletonrsympy.core.symbolr&sympy.functions.elementary.exponentialr (sympy.functions.elementary.miscellaneousr (sympy.functions.elementary.trigonometricr r sympy.integrals.integralsr sympy.simplify.simplifyrsympy.physics.hydrogenrrrrsympy.testing.pytestrr4r1r0rr6r:r@rLrYrHrrresX44"%69?/,BB' ' 1a 0;.G ? ( (8r