U 0e0@sddlmZmZmZmZmZddlmZmZm Z m Z m Z m Z ddl mZddlmZedZddZdd Zd d Zd d ZddZddZddZddZdS))labeller render_labelMzCreateOneQubitGate CreateCGate)CNOTHSWAPCGateST) import_module)skipZ matplotlibcCs*tddksttdddidks&tdS)NZq0z$\left|q0\right\rangle$0z,$\left|q0\right\rangle=\left|0\right\rangle$)rAssertionErrorrrP/tmp/pip-unpacked-wheel-_6tpq7m6/sympy/physics/quantum/tests/test_circuitplot.pytest_render_label srcCsttddkstdS)NrzMz(0))strrrrrrrtest_Mz srcCs td}t|ddkstdS)NQrzQ(0))rrrZQgaterrr test_create1srcCs$td}t|dgddks tdS)Nrrz C((1),Q(0)))rrrrrrr test_createcsrcCs0tdddgksttdddddgks,td S) zTest the labeller utilityq_1q_0jj_2j_1j_0N)rrrrrr test_labellersr#cCststdn ddlm}|tdddtdd}|jdks@t|jdksNt|j ddgks`t|tddd}|jdks~t|jdkst|j gkstd 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&rrngatesrnqubitsr(r&crrr test_cnots  r/cCsltstdn ddlm}|tddtddtdd}|jdksHt|j dksVt|j ddgkshtdS) Nr$rr%rrr'rr) r)rr*r&rrrr+rr,r(r-rrrtest_ex10s   r0cCststdn ddlm}|tddtdtdtdtdtdtdtdtdtddt dd d }|j d kst |j dkst |j d d dgkst dS)Nr$rr%r)rr)rrrr'r r!r")r)rr*r&r rr r r rr+rr,r(r-rrrtest_ex4;s  6 r2N)r*rrrrrZsympy.physics.quantum.gaterrr r r r Zsympy.externalr Zsympy.testing.pytestrr)rrrrr#r/r0r2rrrrs