def test_vacuum(self, setup_eng, hbar, cutoff, bsize, pure, tol): """Test vacuum function matches Fock backends""" eng, prog = setup_eng(1) with prog.context as q: ops.Vac | q[0] state = eng.run(prog).state ket = utils.vacuum_state(basis="fock", fock_dim=cutoff, hbar=hbar) if not pure: expected = state.dm() ket = np.tile(np.outer(ket, ket.conj()), (bsize, 1, 1)) else: expected = state.ket() ket = np.tile(ket, (bsize, 1)) assert np.allclose(expected, ket, atol=tol, rtol=0)
def test_vacuum_state_fock(self, cutoff, hbar): """test vacuum state returns correct state vector""" state = utils.vacuum_state(basis="fock", hbar=hbar, fock_dim=cutoff) assert np.all(state == np.eye(1, cutoff, 0))
def test_vacuum_state_gaussian(self, hbar): """test vacuum state returns correct means and covariance""" means, cov = utils.vacuum_state(basis="gaussian", hbar=hbar) assert np.all(means == np.zeros(2)) assert np.all(cov == np.identity(2) * hbar / 2)