def test_gate_control_0():
q = Qubits(2)
q.gate(X, target=0, control_0=1)
assert q.state[0, 0] == 0
assert q.state[0, 1] == 0
assert q.state[1, 0] == 1
assert q.state[1, 1] == 0
q = Qubits(2)
q.gate(X, target=1)
q.gate(X, target=0, control_0=1)
assert q.state[0, 0] == 0
assert q.state[0, 1] == 1
assert q.state[1, 0] == 0
assert q.state[1, 1] == 0
def test_expect_batched():
q = Qubits(2, n_batch_axes=3)
res = q.expect({'YI': 2.5, 'II': 2, 'IX': 1.5, 'IZ': 1})
assert res.ndim == 3
assert _allclose(res, xp.array([[[3], [3], [3]]]))
q = Qubits(2, n_batch_axes=3)
q.gate([I, X], target=1, batch_index=1)
res = q.expect({'IZ': 2})
assert res.ndim == 3
assert _allclose(res, xp.array([[[2], [-2]]]))
def test_init():
q = Qubits(1)
assert q.state[0] == 1
assert q.state[1] == 0
assert _allclose(q.state.flatten(), xp.array([1, 0]))
q = Qubits(2)
assert q.state[0, 0] == 1
assert q.state[0, 1] == 0
assert q.state[1, 0] == 0
assert q.state[1, 1] == 0
assert _allclose(q.state.flatten(), xp.array([1, 0, 0, 0]))
q = Qubits(2, n_batch_axes=3)
assert q.state.shape == (1, 1, 1, 2, 2)
assert q.state[0, 0, 0, 0, 0] == 1
assert q.state[0, 0, 0, 0, 1] == 0
assert q.state[0, 0, 0, 1, 0] == 0
assert q.state[0, 0, 0, 1, 1] == 0
assert _allclose(q.state.flatten(), xp.array([1, 0, 0, 0]))
def test_gate_batched():
q = Qubits(3, n_batch_axes=3)
q.gate(batch_rx([0.1, 0.2, 0.3]), target=1, control=2, batch_index=1)
psi1 = q.state
psi2 = []
for theta in [0.1, 0.2, 0.3]:
q = Qubits(3, n_batch_axes=2)
q.gate(rx(theta), target=1, control=2)
psi2.append(q.state)
psi2 = xp.stack(psi2, axis=1)
print(psi1)
print(psi2)
assert _allclose(psi1, psi2)
def test_gate_single_target():
q = Qubits(2)
q.gate(X, target=0)
assert q.state[0, 0] == 0
assert q.state[0, 1] == 0
assert q.state[1, 0] == 1
assert q.state[1, 1] == 0
assert _allclose(q.state.flatten(), xp.array([0, 0, 1, 0]))
q = Qubits(2)
q.gate(X, target=1)
assert q.state[0, 0] == 0
assert q.state[0, 1] == 1
assert q.state[1, 0] == 0
assert q.state[1, 1] == 0
assert _allclose(q.state.flatten(), xp.array([0, 1, 0, 0]))
def test_gate_single_qubit():
q = Qubits(1)
q.gate(Y, target=0)
assert _allclose(q.state, xp.array([0, 1j]))
q = Qubits(1)
q.gate(ry(0.1), target=0)
q.gate(rz(0.1), target=0)
psi1 = q.state
q = Qubits(1)
q.gate(xp.dot(rz(0.1), ry(0.1)), target=0)
psi2 = q.state
q = Qubits(1)
q.gate(xp.dot(ry(0.1), rz(0.1)), target=0)
psi3 = q.state
assert _allclose(psi1, psi2)
assert not _allclose(psi1, psi3)
def test_get_state():
q = Qubits(2)
q.gate(X, target=0)
q.gate(X, target=1)
assert _allclose(q.get_state(), xp.array([0, 0, 0, 1]))
q = Qubits(2)
q.gate(X, target=1)
assert _allclose(q.get_state(flatten=False), xp.array([[0, 1], [0, 0]]))
q = Qubits(2, n_batch_axes=3)
q.gate([X, Y, I], target=1, batch_index=1)
assert _allclose(
q.get_state(),
xp.array([[
[[0, 1, 0, 0]],
[[0, 1j, 0, 0]],
[[1, 0, 0, 0]],
]]))
q.gate([X, Y, I], target=1, batch_index=1)
assert _allclose(
q.get_state(),
xp.array([[
[[1, 0, 0, 0]],
[[1, 0, 0, 0]],
[[1, 0, 0, 0]],
]]))
def test_expect():
q = Qubits(2)
res = q.expect({'YI': 2.5, 'II': 2, 'IX': 1.5, 'IZ': 1})
assert res == 3
q = Qubits(2)
q.gate(X, target=1)
res = q.expect({'IZ': 2})
assert res == -2
q = Qubits(2)
q.gate(X, target=0)
res = q.expect({'IZ': 2})
assert res == 2
q = Qubits(2)
q.gate(X, target=1)
res = q.expect({'ZI': 2})
assert res == 2
q = Qubits(2)
q.gate(X, target=0)
q.gate(X, target=1)
res = q.expect({'ZZ': 2})
assert res == 2
def test_project_batched():
q = Qubits(2, n_batch_axes=3)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert _allclose(res0, xp.array([0, 0, 0]))
assert _allclose(res1, xp.array([0, 0, 0]))
q = Qubits(2, n_batch_axes=3)
q.gate(Y, target=0)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert _allclose(res0, xp.array([1, 1, 1]))
assert _allclose(res1, xp.array([0, 0, 0]))
q = Qubits(2, n_batch_axes=3)
q.gate(Y, target=1)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert _allclose(res0, xp.array([0, 0, 0]))
assert _allclose(res1, xp.array([1, 1, 1]))
q = Qubits(2, n_batch_axes=3)
q.gate(Y, target=0)
q.gate(Y, target=1)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert _allclose(res0, xp.array([1, 1, 1]))
assert _allclose(res1, xp.array([1, 1, 1]))
def test_project():
q = Qubits(2)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert res0 == 0
assert res1 == 0
q = Qubits(2)
q.gate(Y, target=0)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert res0 == 1
assert res1 == 0
q = Qubits(2)
q.gate(Y, target=1)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert res0 == 0
assert res1 == 1
q = Qubits(2)
q.gate(Y, target=0)
q.gate(Y, target=1)
res0 = q.project(target=0)
res1 = q.project(target=1)
assert res0 == 1
assert res1 == 1
def test_gate_multi_targets():
q = Qubits(2)
q.gate(X, target=0)
assert q.state[0, 0] == 0
assert q.state[0, 1] == 0
assert q.state[1, 0] == 1
assert q.state[1, 1] == 0
q.gate(swap, target=(0, 1))
assert q.state[0, 0] == 0
assert q.state[0, 1] == 1
assert q.state[1, 0] == 0
assert q.state[1, 1] == 0
q = Qubits(2)
q.gate(X, target=0)
q.gate(swap, target=(1, 0))
assert q.state[0, 0] == 0
assert q.state[0, 1] == 1
assert q.state[1, 0] == 0
assert q.state[1, 1] == 0