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]),