def test_init_with_int():
target = 0
instr = Instruction(Gate.H(), target)
assert instr.target == QubitSet(0)
def test_qubits_setter(h):
h.qubits = QubitSet(1)
def test_init_with_qubits():
target = QubitSet([0, 1])
instr = Instruction(Gate.CNot(), target)
assert instr.target == target
),
(
Circuit().h(0).variance(observable=Observable.H(),
target=1).state_vector().amplitude(["01"]),
2,
),
],
)
def test_qubit_count(circuit, expected_qubit_count):
assert circuit.qubit_count == expected_qubit_count
@pytest.mark.parametrize(
"circuit,expected_qubits",
[
(Circuit().h(0).h(1).h(2), QubitSet([0, 1, 2])),
(
Circuit().h(0).expectation(
observable=Observable.H() @ Observable.X(),
target=[0, 1]).sample(
observable=Observable.H() @ Observable.X(), target=[0, 1]),
QubitSet([0, 1]),
),
(
Circuit().h(0).probability([1, 2]).state_vector(),
QubitSet([0]),
),
(
Circuit().h(0).variance(observable=Observable.H(),
target=1).state_vector().amplitude(["01"]),
QubitSet([0, 1]),