update device tests

PennyLaneAI · Apr 16, 2024 · 1f4e1d7 · 1f4e1d7
1 parent e6c59f6
commit 1f4e1d7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_device.py b/tests/test_device.py
@@ -306,6 +306,7 @@ def test_check_validity_on_valid_queue(self, mock_device_supporting_paulis):
  # Raises an error if queue or observables are invalid
  dev.check_validity(queue, observables)
 
+ @pytest.mark.usefixtures("use_new_opmath")
  def test_check_validity_containing_prod(self, mock_device_supporting_prod):
  """Tests that the function Device.check_validity works with Prod"""
 
@@ -323,6 +324,7 @@ def test_check_validity_containing_prod(self, mock_device_supporting_prod):
 
  dev.check_validity(queue, observables)
 
+ @pytest.mark.usefixtures("use_new_opmath")
  def test_prod_containing_unsupported_nested_observables(self, mock_device_supporting_prod):
  """Tests that the observables nested within Prod are checked for validity"""