o *ήcØã@s<ddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl m Zddl mZddlmZdd lmZmZdd lmZdd lmZdd lmZed dd\ZZZZedƒZeddegdedggƒZ ededgdedggƒZ!dd„Z"dd„Z#dd„Z$dd„Z%dd„Z&dd „Z'd!d"„Z(d#d$„Z)d%S)&é)ÚI)Úsymbols)Ú unchanged)ÚMatrixÚ SparseMatrix)Ú Commutator)Ú TensorProduct)Útensor_product_simp)ÚDagger)ÚQubitÚQubitBra)Ú OuterProduct)ÚDensity)ÚTrzA,B,C,DF©Ú commutativeÚxéééécCs t dd¡}tt||ƒsJ‚dS)Nrr)rÚdiagrr)Úspm©rúU/tmp/pip-target-vg8gfxp4/lib/python/sympy/physics/quantum/tests/test_tensorproduct.pyÚtest_sparse_matricess rcCsVtttttƒƒt tttƒttƒƒksJ‚ttttƒƒtttƒttƒƒks)J‚dS)N)r rrÚAÚBÚmat1Úmat2rrrrÚtest_tensor_product_daggers ÿ  ÿr cCsŒtttdtƒtdtttƒksJ‚tttƒtttƒks J‚tttƒjdus*J‚ttttƒtƒs4J‚tttƒ tt¡tttƒksDJ‚dS)NrF)ÚTPrrrÚis_commutativeÚ isinstanceÚsubsÚCrrrrÚtest_tensor_product_abstract#s ($r&cCsJtttttƒjddtttƒtttƒtttƒtttƒks#J‚dS©NT)Ú tensorproduct)r!rrr%ÚexpandrrrrÚtest_tensor_product_expand,s& ÿr*cCs†ttttƒtƒ ¡jddttttƒttttƒksJ‚ttttƒtttƒƒ ¡tttƒtttƒtttƒtttƒksAJ‚dSr')r!ÚCommrrr%Údoitr)rrrrÚtest_tensor_product_commutator1s ÿ& ÿr-cCsttttƒtttƒƒtttttƒksJ‚ttttƒtƒtttttƒks+J‚tttttƒdƒtttdtdƒksCJ‚tttttƒdtttƒƒtttdttdtƒksdJ‚ttttƒtttƒtƒtttƒtttttƒks‚J‚dS)Nr)r r!rrr%rÚDrrrrÚtest_tensor_product_simp8s .(0B@r/cCs8tdtdƒtdƒjƒtdttdƒtdƒƒƒksJ‚dS)NrÚ1)rr Údualr r rrrrÚtest_issue_5923As ÿr2c Cs:tddd\}}}}}}t||ƒ}t|dgƒ}t|ƒ}| ¡dt|t|ƒƒt|t|ƒƒks3J‚t|||ƒ}t|dgƒ}t|dgƒ}| ¡d|t|ƒt|t|ƒƒ|t|ƒksaJ‚t|ddgƒ}| ¡dt|t|ƒƒ|t|ƒt|t|ƒƒks†J‚t||ƒ} t||ƒ} t| dg| dgƒ}t|ƒ}| ¡dt|t|ƒƒt|t|ƒƒdt|t|ƒƒt|t|ƒƒksÇJ‚t|dgƒ}| ¡dt|t|ƒƒ|t|ƒdt|t|ƒƒ|t|ƒksóJ‚t| | dgƒ}t|ƒ}| ¡dt|t|ƒƒt|t|ƒƒdt|t|ƒƒt|t|ƒƒdt|t|ƒƒt|t|ƒƒdt|t|ƒƒt|t|ƒƒksNJ‚t|dgƒ}| ¡d|t|ƒt|t|ƒƒd|t|ƒt|t|ƒƒd|t|ƒt|t|ƒƒd|t|ƒt|t|ƒƒks›J‚dS) Nz A B C D E FFrgð?rrrgà?)rrrrr,r ) rrr%r.ÚEÚFÚtÚdÚtrÚt2Út3rrrÚtest_eval_traceIsN  0   8<  (" ÿ $ ÿ("ÿ"þ" ý $ÿþýr:N)*Úsympy.core.numbersrÚsympy.core.symbolrÚsympy.core.exprrÚsympy.matricesrrÚ sympy.physics.quantum.commutatorrr+Ú#sympy.physics.quantum.tensorproductrr!r Úsympy.physics.quantum.daggerr Úsympy.physics.quantum.qubitr r Úsympy.physics.quantum.operatorr Úsympy.physics.quantum.densityrÚsympy.physics.quantum.tracerrrr%r.rrrrr r&r*r-r/r2r:rrrrÚs2