def test_grad_beamspitter():
"""Tests the value of the analytic gradient for the S2gate against finite differences"""
cutoff = 4
r = 1.0
theta = np.pi / 8
T = beamsplitter(r, theta, cutoff)
Dr, Dtheta = grad_beamsplitter(T, r, theta)
dr = 0.001
dtheta = 0.001
Drp = beamsplitter(r + dr, theta, cutoff)
Drm = beamsplitter(r - dr, theta, cutoff)
Dthetap = beamsplitter(r, theta + dtheta, cutoff)
Dthetam = beamsplitter(r, theta - dtheta, cutoff)
Drapprox = (Drp - Drm) / (2 * dr)
Dthetaapprox = (Dthetap - Dthetam) / (2 * dtheta)
assert np.allclose(Dr, Drapprox, atol=1e-4, rtol=0)
assert np.allclose(Dtheta, Dthetaapprox, atol=1e-4, rtol=0)
def test_BS_hong_ou_mandel_interference(tol):
r"""Tests Hong-Ou-Mandel interference for a 50:50 beamsplitter.
If one writes :math:`U` for the Fock representation of a 50-50 beamsplitter
then it must hold that :math:`\langle 1,1|U|1,1 \rangle = 0`.
"""
cutoff = 2
phi = 2 * np.pi * np.random.rand()
T = beamsplitter(np.pi / 4, phi, cutoff) # a 50-50 beamsplitter with phase phi
assert np.allclose(T[1, 1, 1, 1], 0.0, atol=tol, rtol=0)
def test_beamsplitter_values(tol):
r"""Test that the representation of an interferometer in the single
excitation manifold is precisely the unitary matrix that represents it
mode in space. This test in particular checks that the BS gate is
consistent with strawberryfields
"""
nmodes = 2
vec_list = np.identity(nmodes, dtype=int).tolist()
theta = 2 * np.pi * np.random.rand()
phi = 2 * np.pi * np.random.rand()
ct = np.cos(theta)
st = np.sin(theta) * np.exp(1j * phi)
U = np.array([[ct, -np.conj(st)], [st, ct]])
# Calculate the matrix \langle i | U | j \rangle = T[i+j]
T = beamsplitter(theta, phi, 3)
U_rec = np.empty([nmodes, nmodes], dtype=complex)
for i, vec_i in enumerate(vec_list):
for j, vec_j in enumerate(vec_list):
U_rec[i, j] = T[tuple(vec_i + vec_j)]
assert np.allclose(U, U_rec, atol=tol, rtol=0)
def test_BS_selection_rules(tol):
r"""Test the selection rules of a beamsplitter.
If one writes the beamsplitter gate of :math:`U` and its matrix elements as
:math:`\langle m, n |U|k,l \rangle` then these elements
are nonzero if and only if :math:`m+n = k+l`. This test checks
that this selection rule holds.
"""
cutoff = 4
T = beamsplitter(np.random.rand(), np.random.rand(), cutoff)
m = np.arange(cutoff).reshape(-1, 1, 1, 1)
n = np.arange(cutoff).reshape(1, -1, 1, 1)
k = np.arange(cutoff).reshape(1, 1, -1, 1)
l = np.arange(cutoff).reshape(1, 1, 1, -1)
# create a copy of T, but replace all elements where
# m+n != k+l with 0.
S = np.where(m + n != k + l, 0, T)
# check that S and T remain equal
assert np.allclose(S, T, atol=tol, rtol=0)
