def test_drawer_measurement():
drawer = CircuitDrawer(default_measure=0)
eng = MainEngine(drawer, [])
qubit = eng.allocate_qubit()
Measure | qubit
assert int(qubit) == 0
drawer = CircuitDrawer(default_measure=1)
eng = MainEngine(drawer, [])
qubit = eng.allocate_qubit()
Measure | qubit
assert int(qubit) == 1
drawer = CircuitDrawer(accept_input=True)
eng = MainEngine(drawer, [])
qubit = eng.allocate_qubit()
old_input = _drawer.input
_drawer.input = lambda x: '1'
Measure | qubit
assert int(qubit) == 1
_drawer.input = old_input
qb1 = WeakQubitRef(engine=eng, idx=1)
qb2 = WeakQubitRef(engine=eng, idx=2)
with pytest.raises(ValueError):
eng.backend._print_cmd(
Command(engine=eng, gate=Measure, qubits=([qb1], ),
controls=[qb2]))
def test_drawer_getlatex(ordered):
old_latex = _drawer.to_latex
_drawer.to_latex = lambda x, drawing_order, draw_gates_in_parallel: x
drawer = CircuitDrawer()
drawer.set_qubit_locations({0: 1, 1: 0})
drawer2 = CircuitDrawer()
eng = MainEngine(drawer, [drawer2])
qureg = eng.allocate_qureg(2)
H | qureg[1]
H | qureg[0]
X | qureg[0]
CNOT | (qureg[0], qureg[1])
lines = drawer2.get_latex(ordered=ordered)
assert len(lines) == 2
assert len(lines[0]) == 4
assert len(lines[1]) == 3
# check if it was sent on correctly:
lines = drawer.get_latex(ordered=ordered)
assert len(lines) == 2
assert len(lines[0]) == 3
assert len(lines[1]) == 4
_drawer.to_latex = old_latex
def test_drawer_isavailable():
drawer = CircuitDrawer()
drawer.is_last_engine = True
assert drawer.is_available(None)
assert drawer.is_available("Everything")
mock_engine = MockEngine()
mock_engine.called = False
drawer.is_last_engine = False
drawer.next_engine = mock_engine
assert not drawer.is_available(None)
assert mock_engine.called
assert mock_engine.cmd is None
def test_drawer_measurement():
drawer = CircuitDrawer(default_measure=0)
eng = MainEngine(drawer, [])
qubit = eng.allocate_qubit()
Measure | qubit
assert int(qubit) == 0
drawer = CircuitDrawer(default_measure=1)
eng = MainEngine(drawer, [])
qubit = eng.allocate_qubit()
Measure | qubit
assert int(qubit) == 1
drawer = CircuitDrawer(accept_input=True)
eng = MainEngine(drawer, [])
qubit = eng.allocate_qubit()
old_input = _drawer.input
_drawer.input = lambda x: '1'
Measure | qubit
assert int(qubit) == 1
_drawer.input = old_input
def test_drawer_qubitmapping():
drawer = CircuitDrawer()
# mapping should still work (no gate has been applied yet)
valid_mappings = [{0: 1, 1: 0}, {2: 1, 1: 2}]
for valid_mapping in valid_mappings:
drawer.set_qubit_locations(valid_mapping)
drawer = CircuitDrawer()
# invalid mapping should raise an error:
invalid_mappings = [{3: 1, 0: 2}, {0: 1, 2: 1}]
for invalid_mapping in invalid_mappings:
drawer = CircuitDrawer()
with pytest.raises(RuntimeError):
drawer.set_qubit_locations(invalid_mapping)
eng = MainEngine(drawer, [])
qubit = eng.allocate_qubit()
# mapping has begun --> can't assign it anymore
with pytest.raises(RuntimeError):
drawer.set_qubit_locations({0: 1, 1: 0})