|
|
from sympy.physics.quantum.qasm import Qasm, flip_index, trim,\ |
|
|
get_index, nonblank, fullsplit, fixcommand, stripquotes, read_qasm |
|
|
from sympy.physics.quantum.gate import X, Z, H, S, T |
|
|
from sympy.physics.quantum.gate import CNOT, SWAP, CPHASE, CGate, CGateS |
|
|
from sympy.physics.quantum.circuitplot import Mz |
|
|
|
|
|
def test_qasm_readqasm(): |
|
|
qasm_lines = """\ |
|
|
qubit q_0 |
|
|
qubit q_1 |
|
|
h q_0 |
|
|
cnot q_0,q_1 |
|
|
""" |
|
|
q = read_qasm(qasm_lines) |
|
|
assert q.get_circuit() == CNOT(1,0)*H(1) |
|
|
|
|
|
def test_qasm_ex1(): |
|
|
q = Qasm('qubit q0', 'qubit q1', 'h q0', 'cnot q0,q1') |
|
|
assert q.get_circuit() == CNOT(1,0)*H(1) |
|
|
|
|
|
def test_qasm_ex1_methodcalls(): |
|
|
q = Qasm() |
|
|
q.qubit('q_0') |
|
|
q.qubit('q_1') |
|
|
q.h('q_0') |
|
|
q.cnot('q_0', 'q_1') |
|
|
assert q.get_circuit() == CNOT(1,0)*H(1) |
|
|
|
|
|
def test_qasm_swap(): |
|
|
q = Qasm('qubit q0', 'qubit q1', 'cnot q0,q1', 'cnot q1,q0', 'cnot q0,q1') |
|
|
assert q.get_circuit() == CNOT(1,0)*CNOT(0,1)*CNOT(1,0) |
|
|
|
|
|
|
|
|
def test_qasm_ex2(): |
|
|
q = Qasm('qubit q_0', 'qubit q_1', 'qubit q_2', 'h q_1', |
|
|
'cnot q_1,q_2', 'cnot q_0,q_1', 'h q_0', |
|
|
'measure q_1', 'measure q_0', |
|
|
'c-x q_1,q_2', 'c-z q_0,q_2') |
|
|
assert q.get_circuit() == CGate(2,Z(0))*CGate(1,X(0))*Mz(2)*Mz(1)*H(2)*CNOT(2,1)*CNOT(1,0)*H(1) |
|
|
|
|
|
def test_qasm_1q(): |
|
|
for symbol, gate in [('x', X), ('z', Z), ('h', H), ('s', S), ('t', T), ('measure', Mz)]: |
|
|
q = Qasm('qubit q_0', '%s q_0' % symbol) |
|
|
assert q.get_circuit() == gate(0) |
|
|
|
|
|
def test_qasm_2q(): |
|
|
for symbol, gate in [('cnot', CNOT), ('swap', SWAP), ('cphase', CPHASE)]: |
|
|
q = Qasm('qubit q_0', 'qubit q_1', '%s q_0,q_1' % symbol) |
|
|
assert q.get_circuit() == gate(1,0) |
|
|
|
|
|
def test_qasm_3q(): |
|
|
q = Qasm('qubit q0', 'qubit q1', 'qubit q2', 'toffoli q2,q1,q0') |
|
|
assert q.get_circuit() == CGateS((0,1),X(2)) |
|
|
|
|
|
def test_qasm_flip_index(): |
|
|
assert flip_index(0, 2) == 1 |
|
|
assert flip_index(1, 2) == 0 |
|
|
|
|
|
def test_qasm_trim(): |
|
|
assert trim('nothing happens here') == 'nothing happens here' |
|
|
assert trim("Something #happens here") == "Something " |
|
|
|
|
|
def test_qasm_get_index(): |
|
|
assert get_index('q0', ['q0', 'q1']) == 1 |
|
|
assert get_index('q1', ['q0', 'q1']) == 0 |
|
|
|
|
|
def test_qasm_nonblank(): |
|
|
assert list(nonblank('abcd')) == list('abcd') |
|
|
assert list(nonblank('abc ')) == list('abc') |
|
|
|
|
|
def test_qasm_fullsplit(): |
|
|
assert fullsplit('g q0,q1,q2, q3') == ('g', ['q0', 'q1', 'q2', 'q3']) |
|
|
|
|
|
def test_qasm_fixcommand(): |
|
|
assert fixcommand('foo') == 'foo' |
|
|
assert fixcommand('def') == 'qdef' |
|
|
|
|
|
def test_qasm_stripquotes(): |
|
|
assert stripquotes("'S'") == 'S' |
|
|
assert stripquotes('"S"') == 'S' |
|
|
assert stripquotes('S') == 'S' |
|
|
|
|
|
def test_qasm_qdef(): |
|
|
|
|
|
q = Qasm("def Q,0,Q",'qubit q0','Q q0') |
|
|
assert str(q.get_circuit()) == 'Q(0)' |
|
|
|
|
|
q = Qasm("def CQ,1,Q", 'qubit q0', 'qubit q1', 'CQ q0,q1') |
|
|
assert str(q.get_circuit()) == 'C((1),Q(0))' |
|
|
|
