a .$. : r1cCsTd}t|dD]>}t|D]0}tt||tdtdtdtfdksJqqdS)Nrrr)ranger rr,r)Zn_maxr/r0rrr test_norm4s r3cCstd}td}td}tddd|||t| ttks@Jtddd|||tjt| d|t|tt |dttksJtd dd|||dtd t|tt|t|dtttdd td td |d |ddks JdS) Nr,phithetarrrr rr!r%) rrrr rr-r rr )r,r4r5rrr test_psi_nlm;s0(& "   r8cCs ttttd dtdks$Jttddtdks@Jtddtdd dks`Jtddtdd dksJtdtj dksJtdtj dksJtdtj dksJtdtj d ksJtd tj d ksJttd d dS)Nrr6r/r#rr di NcSstdSNr)rrrrrSz(test_hydrogen_energies..)rr/r+rZOner ValueErrorrrrrtest_hydrogen_energiesFs$  rAcCstddddddtddks$Jttddddddtddtdtddd dkshJttdddddd tdd td tdd ddksJttdddddddtdtddtdddksJttdddddddtdtddtdddks6JtddD]^}t|D]N}tt||t|ddsnJ|dkrLtt||dt|ddsLJqLq@d}tddD]h}t|D]X}tt|||dt||ddsJ|dkrtt||d|t||ddsJqqd}tddD]h}t|D]X}tt|||dt||ddsZJ|dkr2tt||d|t||dds2Jq2q&ttd d!ttd"d!ttd#d!dS)$Nrrr)r+cr#rr%r;r 6r!l iKidiiB)i9iQIRiY<'i /ixNr&gh㈵>F)r+g-C6?gMbP?cSs tddSr=rrrrrr>|r?z.test_hydrogen_energies_relat..cSs tddS)Nrr6rIrrrrr>}r?cSs tdddS)NrrFrIrrrrr>~r?)rr r r2rrrr@)r/r0r+rrrtest_hydrogen_energies_relatVsj$   ( $ , $ ,rJN)rr)$Zsympy.core.numbersrrrrZsympy.core.singletonrZsympy.core.symbolrZ&sympy.functions.elementary.exponentialrZ(sympy.functions.elementary.miscellaneousr Z(sympy.functions.elementary.trigonometricr r Zsympy.integrals.integralsr Zsympy.simplify.simplifyr Zsympy.physics.hydrogenrrrrZsympy.testing.pytestrr/r,r+rr1r3r8rArJrrrrs