o GZh0@sddlmZmZmZmZmZddlmZmZm Z m Z m Z m Z ddl mZddlmZedZddZdd Zd d Zd d ZddZddZddZddZdS))labeller render_labelMzCreateOneQubitGate CreateCGate)CNOTHSWAPCGateST) import_module)skipZ matplotlibcCs*tddksJtdddidksJdS)Nq0z$\left|q0\right\rangle$0z,$\left|q0\right\rangle=\left|0\right\rangle$)rrr[/var/www/auris/lib/python3.10/site-packages/sympy/physics/quantum/tests/test_circuitplot.pytest_render_label srcCsttddks JdS)NrzMz(0))strrrrrrtest_Mz srcCs td}t|ddksJdS)NQrzQ(0))rrZQgaterrr test_create1srcCs$td}t|dgddksJdS)Nrrz C((1),Q(0)))rrrrrr test_createcsrcCs.tdddgks JtddgdksJdS)zTest the labeller utilityq_1q_0jZj_2Zj_1Zj_0N)rrrrr test_labellersr!cCststdnddlm}|tdddtdd}|jdks J|jdks'J|jddgks0J|tddd}|jdks?J|jdksFJ|jgksMJd S) zTest a simple cnot circuit. Right now this only makes sure the code doesn't raise an exception, and some simple properties matplotlib not installedr CircuitPlotrrlabelsrrN) mplr!sympy.physics.quantum.circuitplotr$rrngatesnqubitsr&r$crrr test_cnots  r-cCsltstdnddlm}|tddtddtdd}|jdks$J|jdks+J|j ddgks4JdS) Nr"rr#rrr%rr) r'rr(r$rrrr)r*r&r+rrrtest_ex10s   r.cCststdnddlm}|tddtdtdtdtdtdtdtdtdtddt dd d }|j d ksBJ|j dksIJ|j gd ksRJdS) Nr"rr#r)rr)rrrr%r ) r'rr(r$r rr r r rr)r*r&r+rrrtest_ex4;s  6  r0N)r(rrrrrZsympy.physics.quantum.gaterrr r r r Zsympy.externalr Zsympy.testing.pytestrr'rrrrr!r-r.r0rrrrs