def test_tensor_gates_two_qubit():
prog = Program().inst([CNOTgate(0, 1)])
test_unitary = tensor_gates(gate_matrix, {}, prog.instructions[0], 4).toarray()
true_unitary = apply_gate(gate_matrix['CNOT'], [0, 1], 4).toarray()
assert np.allclose(test_unitary, true_unitary)
prog = Program().inst([CNOTgate(1, 0)])
test_unitary = tensor_gates(gate_matrix, {}, prog.instructions[0], 4).toarray()
true_unitary = apply_gate(gate_matrix['CNOT'], [1, 0], 4).toarray()
assert np.allclose(test_unitary, true_unitary)
prog = Program().inst([CNOTgate(1, 3)])
test_unitary = tensor_gates(gate_matrix, {}, prog.instructions[0], 4).toarray()
true_unitary = apply_gate(gate_matrix['CNOT'], [1, 3], 4).toarray()
assert np.allclose(test_unitary, true_unitary)
def test_single_qubit_gates():
test_unitary = apply_gate(gate_matrix['H'], 0, 4).toarray()
true_unitary = np.kron(np.eye(8), gate_matrix['H'])
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 1, 4).toarray()
true_unitary = np.kron(np.eye(4), np.kron(gate_matrix['H'], np.eye(2)))
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 2, 4).toarray()
true_unitary = np.kron(np.eye(2), np.kron(gate_matrix['H'], np.eye(4)))
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 3, 4).toarray()
true_unitary = np.kron(gate_matrix['H'], np.eye(8))
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 0, 5).toarray()
true_unitary = np.kron(np.eye(2**4), gate_matrix['H'])
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 1, 5).toarray()
true_unitary = np.kron(np.eye(2**3), np.kron(gate_matrix['H'], np.eye(2)))
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 2, 5).toarray()
true_unitary = np.kron(np.eye(2**2), np.kron(gate_matrix['H'], np.eye(2**2)))
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 3, 5).toarray()
true_unitary = np.kron(np.eye(2**1), np.kron(gate_matrix['H'], np.eye(2**3)))
assert np.allclose(test_unitary, true_unitary)
test_unitary = apply_gate(gate_matrix['H'], 4, 5).toarray()
true_unitary = np.kron(np.eye(2**0), np.kron(gate_matrix['H'], np.eye(2**4)))
assert np.allclose(test_unitary, true_unitary)
def test_two_qubit_gates():
unitary_test = apply_gate(gate_matrix['CNOT'], [1, 0], 2).toarray()
unitary_true = np.kron(utility_gates['P0'], np.eye(2)) + \
np.kron(utility_gates['P1'], gate_matrix['X'])
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['CNOT'], [0, 1], 2).toarray()
unitary_true = np.kron(np.eye(2), utility_gates['P0']) + \
np.kron(gate_matrix['X'], utility_gates['P1'])
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['CNOT'], [2, 1], 3).toarray()
unitary_true = np.kron(gate_matrix['CNOT'], np.eye(2 ** 1))
assert np.allclose(unitary_test, unitary_true)
with pytest.raises(ValueError):
apply_gate(gate_matrix['CNOT'], [2, 1], 2)
unitary_test = apply_gate(gate_matrix['ISWAP'], [0, 1], 3).toarray()
unitary_true = np.kron(np.eye(2), gate_matrix['ISWAP'])
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [1, 0], 3).toarray()
unitary_true = np.kron(np.eye(2), gate_matrix['ISWAP'])
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [1, 2], 4).toarray()
unitary_true = np.kron(np.eye(2), np.kron(gate_matrix['ISWAP'], np.eye(2)))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [3, 2], 4).toarray()
unitary_true = np.kron(gate_matrix['ISWAP'], np.eye(4))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [2, 3], 4).toarray()
unitary_true = np.kron(gate_matrix['ISWAP'], np.eye(4))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [0, 3], 4).toarray()
swap_01 = np.kron(np.eye(4), gate_matrix['SWAP'])
swap_12 = np.kron(np.eye(2), np.kron(gate_matrix['SWAP'], np.eye(2)))
swapper = swap_12.dot(swap_01)
V = np.kron(gate_matrix['ISWAP'], np.eye(4))
unitary_true = np.dot(np.conj(swapper.T), np.dot(V, swapper))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [3, 0], 4).toarray()
swap_01 = np.kron(np.eye(4), gate_matrix['SWAP'])
swap_12 = np.kron(np.eye(2), np.kron(gate_matrix['SWAP'], np.eye(2)))
swapper = swap_12.dot(swap_01)
V = np.kron(gate_matrix['ISWAP'], np.eye(4))
unitary_true = np.dot(np.conj(swapper.T), np.dot(V, swapper))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [1, 3], 4).toarray()
swap_12 = np.kron(np.eye(2), np.kron(gate_matrix['SWAP'], np.eye(2)))
swapper = swap_12
V = np.kron(gate_matrix['ISWAP'], np.eye(4))
unitary_true = np.dot(np.conj(swapper.T), np.dot(V, swapper))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['ISWAP'], [3, 1], 4).toarray()
swap_12 = np.kron(np.eye(2), np.kron(gate_matrix['SWAP'], np.eye(2)))
swapper = swap_12
V = np.kron(gate_matrix['ISWAP'], np.eye(4))
unitary_true = np.dot(np.conj(swapper.T), np.dot(V, swapper))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['CNOT'], [3, 1], 4).toarray()
swap_12 = np.kron(np.eye(2), np.kron(gate_matrix['SWAP'], np.eye(2)))
swapper = swap_12
V = np.kron(gate_matrix['CNOT'], np.eye(4))
unitary_true = np.dot(np.conj(swapper.T), np.dot(V, swapper))
assert np.allclose(unitary_test, unitary_true)
unitary_test = apply_gate(gate_matrix['SWAP'], [3, 1], 4).toarray()
swap_12 = np.kron(np.eye(2), np.kron(gate_matrix['SWAP'], np.eye(2)))
swapper = swap_12
V = np.kron(gate_matrix['SWAP'], np.eye(4))
unitary_true = np.dot(np.conj(swapper.T), np.dot(V, swapper))
assert np.allclose(unitary_test, unitary_true)