90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
|
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():
|
||
|
|
# weaker test condition (str) since we don't have access to the actual class
|
||
|
|
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))'
|