Skip to content

Commit

Permalink
Remove deprecated use of Tensor from tests
Browse files Browse the repository at this point in the history
  • Loading branch information
astralcai committed Oct 7, 2024
1 parent 9753059 commit 784280f
Showing 1 changed file with 12 additions and 13 deletions.
25 changes: 12 additions & 13 deletions tests/test_qpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@
import pytest
from conftest import BaseTest
from flaky import flaky
from pennylane.operation import Tensor
from pennylane.wires import Wires
from pyquil.experiment import SymmetrizationLevel

Expand Down Expand Up @@ -72,9 +71,9 @@ def test_qpu_args(self):
@pytest.mark.parametrize(
"obs", [qml.PauliX(0), qml.PauliZ(0), qml.PauliY(0), qml.Hadamard(0), qml.Identity(0)]
)
def test_tensor_expval_parametric_compilation(self, obs):
"""Test the QPU expval method for Tensor observables made up of a single observable when parametric compilation is
turned on.
def test_prod_expval_parametric_compilation(self, obs):
"""Test the QPU expval method for Prod observables made up of a single observable when
parametric compilation is turned on.
As the results coming from the qvm are stochastic, a constraint of 3 out of 10 runs was added.
"""
Expand All @@ -101,16 +100,16 @@ def template(param):
qml.CNOT(wires=[2, 3])

@qml.qnode(dev)
def circuit_tensor(param):
def circuit_prod(param):
template(param)
return qml.expval(Tensor(obs))
return qml.expval(qml.prod(obs))

@qml.qnode(dev_1)
def circuit_obs(param):
template(param)
return qml.expval(obs)

res = circuit_tensor(p)
res = circuit_prod(p)
exp = circuit_obs(p)

assert np.allclose(res, exp, atol=2e-2)
Expand All @@ -119,9 +118,9 @@ def circuit_obs(param):
@pytest.mark.parametrize(
"obs", [qml.PauliX(0), qml.PauliZ(0), qml.PauliY(0), qml.Hadamard(0), qml.Identity(0)]
)
def test_tensor_expval_operator_estimation(self, obs, shots):
"""Test the QPU expval method for Tensor observables made up of a single observable when parametric compilation is
turned off allowing operator estimation.
def test_prod_expval_operator_estimation(self, obs, shots):
"""Test the QPU expval method for Prod observables made up of a single observable when
parametric compilation is turned off allowing operator estimation.
As the results coming from the qvm are stochastic, a constraint of 3 out of 10 runs was added.
"""
Expand Down Expand Up @@ -150,16 +149,16 @@ def template(param):
qml.CNOT(wires=[2, 3])

@qml.qnode(dev)
def circuit_tensor(param):
def circuit_prod(param):
template(param)
return qml.expval(Tensor(obs))
return qml.expval(qml.prod(obs))

@qml.qnode(dev_1)
def circuit_obs(param):
template(param)
return qml.expval(obs)

res = circuit_tensor(p)
res = circuit_prod(p)
exp = circuit_obs(p)

assert np.allclose(res, exp, atol=2e-2)
Expand Down

0 comments on commit 784280f

Please sign in to comment.