Remove coherent state tests

Signed-off-by: Thien Nguyen <[email protected]>

Author: 1tnguyen

python/tests/operator/integrators/test_evolve_dynamics_torch_integrators.py

@@ -377,120 +377,6 @@ def test_analytical_models(integrator, model, runner):
                                    model.expected_state(model.tlist[i]),
                                    atol=1e-3)
 
-
-@pytest.mark.parametrize('integrator', all_integrator_classes)
-def test_squeezing(integrator):
-
-    def n_thermal(w: float, w_th: float):
-        """
-        Return the number of average photons in thermal equilibrium for a
-            an oscillator with the given frequency and temperature.
-        """
-        if (w_th > 0) and np.exp(w / w_th) != 1.0:
-            return 1.0 / (np.exp(w / w_th) - 1.0)
-        else:
-            return 0.0
-
-    # Problem parameters
-    w0 = 1.0 * 2 * np.pi
-    gamma0 = 0.05
-    # the temperature of the environment
-    w_th = 0.0 * 2 * np.pi
-    # the number of average excitations in the environment mode w0 at temperature w_th
-    Nth = n_thermal(w0, w_th)
-    # squeezing parameter for the environment
-    r = 1.0
-    theta = 0.1 * np.pi
-    N = Nth * (np.cosh(r)**2 + np.sinh(r)**2) + np.sinh(r)**2
-    sz_ss_analytical = -1 / (2 * N + 1)
-    # System Hamiltonian
-    hamiltonian = -0.5 * w0 * spin.z(0)
-    # Collapse operators
-    c_ops = [
-        np.sqrt(gamma0) * (spin.minus(0) * np.cosh(r) +
-                           spin.plus(0) * np.sinh(r) * np.exp(1j * theta))
-    ]
-
-    # System dimension
-    dimensions = {0: 2}
-    # Start in an arbitrary superposition state
-    psi0_ = cp.array([2j, 1.0], dtype=cp.complex128)
-    psi0_ = psi0_ / cp.linalg.norm(psi0_)
-    psi0 = cudaq.State.from_data(psi0_)
-    # Simulation time points
-    steps = np.linspace(0, 50, 1001)
-    schedule = Schedule(steps, ["t"])
-
-    # Run the simulation
-    evolution_result = evolve(hamiltonian,
-                              dimensions,
-                              schedule,
-                              psi0,
-                              observables=[spin.x(0),
-                                           spin.y(0),
-                                           spin.z(0)],
-                              collapse_operators=c_ops,
-                              store_intermediate_results=True,
-                              integrator=integrator())
-
-    exp_val_x = [
-        exp_vals[0].expectation()
-        for exp_vals in evolution_result.expectation_values()
-    ]
-    exp_val_y = [
-        exp_vals[1].expectation()
-        for exp_vals in evolution_result.expectation_values()
-    ]
-    exp_val_z = [
-        exp_vals[2].expectation()
-        for exp_vals in evolution_result.expectation_values()
-    ]
-
-    np.testing.assert_allclose(exp_val_z[-1], sz_ss_analytical, atol=1e-3)
-
-
-@pytest.mark.parametrize('integrator', all_integrator_classes)
-def test_cat_state(integrator):
-    # Number of Fock levels
-    N = 15
-    # Kerr non-linearity
-    chi = 1 * 2 * np.pi
-
-    dimensions = {0: N}
-
-    a = operators.annihilate(0)
-    a_dag = operators.create(0)
-
-    # Defining the Hamiltonian for the system (non-linear Kerr effect)
-    hamiltonian = 0.5 * chi * a_dag * a_dag * a * a
-
-    # we start with a coherent state with alpha=2.0
-    # This will evolve into a cat state.
-    rho0 = cudaq.State.from_data(coherent_state(N, 2.0))
-
-    # Choose the end time at which the state evolves to the exact cat state.
-    steps = np.linspace(0, 0.5 * chi / (2 * np.pi), 51)
-    schedule = Schedule(steps, ["t"])
-
-    # Run the simulation: observe the photon count, position and momentum.
-    evolution_result = evolve(hamiltonian,
-                              dimensions,
-                              schedule,
-                              rho0,
-                              observables=[],
-                              collapse_operators=[],
-                              store_intermediate_results=False,
-                              integrator=integrator())
-
-    # The expected cat state: superposition of `|alpla>` and `|-alpha>` coherent states.
-    expected_state = np.exp(1j * np.pi / 4) * coherent_state(N, -2.0j) + np.exp(
-        -1j * np.pi / 4) * coherent_state(N, 2.0j)
-    expected_state = expected_state / cp.linalg.norm(expected_state)
-    final_state = evolution_result.final_state()
-    overlap = final_state.overlap(expected_state)
-    np.testing.assert_allclose(overlap, 1.0, atol=1e-3)
-
-
 @pytest.mark.parametrize('integrator', all_integrator_classes)
 def test_floquet_steady_state(integrator):
     # two-level system coupled with the external heat bath (fixed temperature)

python/tests/operator/test_evolve_dynamics.py

@@ -377,120 +377,6 @@ def test_analytical_models(integrator, model, runner):
                                    model.expected_state(model.tlist[i]),
                                    atol=1e-3)
 
-
-@pytest.mark.parametrize('integrator', all_integrator_classes)
-def test_squeezing(integrator):
-
-    def n_thermal(w: float, w_th: float):
-        """
-        Return the number of average photons in thermal equilibrium for a
-            an oscillator with the given frequency and temperature.
-        """
-        if (w_th > 0) and np.exp(w / w_th) != 1.0:
-            return 1.0 / (np.exp(w / w_th) - 1.0)
-        else:
-            return 0.0
-
-    # Problem parameters
-    w0 = 1.0 * 2 * np.pi
-    gamma0 = 0.05
-    # the temperature of the environment
-    w_th = 0.0 * 2 * np.pi
-    # the number of average excitations in the environment mode w0 at temperature w_th
-    Nth = n_thermal(w0, w_th)
-    # squeezing parameter for the environment
-    r = 1.0
-    theta = 0.1 * np.pi
-    N = Nth * (np.cosh(r)**2 + np.sinh(r)**2) + np.sinh(r)**2
-    sz_ss_analytical = -1 / (2 * N + 1)
-    # System Hamiltonian
-    hamiltonian = -0.5 * w0 * spin.z(0)
-    # Collapse operators
-    c_ops = [
-        np.sqrt(gamma0) * (spin.minus(0) * np.cosh(r) +
-                           spin.plus(0) * np.sinh(r) * np.exp(1j * theta))
-    ]
-
-    # System dimension
-    dimensions = {0: 2}
-    # Start in an arbitrary superposition state
-    psi0_ = cp.array([2j, 1.0], dtype=cp.complex128)
-    psi0_ = psi0_ / cp.linalg.norm(psi0_)
-    psi0 = cudaq.State.from_data(psi0_)
-    # Simulation time points
-    steps = np.linspace(0, 50, 1001)
-    schedule = Schedule(steps, ["t"])
-
-    # Run the simulation
-    evolution_result = evolve(hamiltonian,
-                              dimensions,
-                              schedule,
-                              psi0,
-                              observables=[spin.x(0),
-                                           spin.y(0),
-                                           spin.z(0)],
-                              collapse_operators=c_ops,
-                              store_intermediate_results=True,
-                              integrator=integrator())
-
-    exp_val_x = [
-        exp_vals[0].expectation()
-        for exp_vals in evolution_result.expectation_values()
-    ]
-    exp_val_y = [
-        exp_vals[1].expectation()
-        for exp_vals in evolution_result.expectation_values()
-    ]
-    exp_val_z = [
-        exp_vals[2].expectation()
-        for exp_vals in evolution_result.expectation_values()
-    ]
-
-    np.testing.assert_allclose(exp_val_z[-1], sz_ss_analytical, atol=1e-3)
-
-
-@pytest.mark.parametrize('integrator', all_integrator_classes)
-def test_cat_state(integrator):
-    # Number of Fock levels
-    N = 15
-    # Kerr non-linearity
-    chi = 1 * 2 * np.pi
-
-    dimensions = {0: N}
-
-    a = operators.annihilate(0)
-    a_dag = operators.create(0)
-
-    # Defining the Hamiltonian for the system (non-linear Kerr effect)
-    hamiltonian = 0.5 * chi * a_dag * a_dag * a * a
-
-    # we start with a coherent state with alpha=2.0
-    # This will evolve into a cat state.
-    rho0 = cudaq.State.from_data(coherent_state(N, 2.0))
-
-    # Choose the end time at which the state evolves to the exact cat state.
-    steps = np.linspace(0, 0.5 * chi / (2 * np.pi), 51)
-    schedule = Schedule(steps, ["t"])
-
-    # Run the simulation: observe the photon count, position and momentum.
-    evolution_result = evolve(hamiltonian,
-                              dimensions,
-                              schedule,
-                              rho0,
-                              observables=[],
-                              collapse_operators=[],
-                              store_intermediate_results=False,
-                              integrator=integrator())
-
-    # The expected cat state: superposition of `|alpla>` and `|-alpha>` coherent states.
-    expected_state = np.exp(1j * np.pi / 4) * coherent_state(N, -2.0j) + np.exp(
-        -1j * np.pi / 4) * coherent_state(N, 2.0j)
-    expected_state = expected_state / cp.linalg.norm(expected_state)
-    final_state = evolution_result.final_state()
-    overlap = final_state.overlap(expected_state)
-    np.testing.assert_allclose(overlap, 1.0, atol=1e-3)
-
-
 @pytest.mark.parametrize('integrator', all_integrator_classes)
 def test_floquet_steady_state(integrator):
     # two-level system coupled with the external heat bath (fixed temperature)