catch PLDW

Author: JerryChen97

tests/test_device.py

@@ -18,6 +18,7 @@
 import requests
 
 import pennylane as qml
+from pennylane.exceptions import PennyLaneDeprecationWarning
 import numpy as np
 
 import pennylane_aqt.device
@@ -425,7 +426,8 @@ class TestAQTDeviceIntegration:
     def test_load_from_device_function(self, num_wires, shots):
         """Tests that the AQTDevice can be loaded from PennyLane `device` function."""
 
-        dev = qml.device("aqt.sim", wires=num_wires, shots=shots, api_key=SOME_API_KEY)
+        with pytest.warns(PennyLaneDeprecationWarning, match="shots on device is deprecated"):
+            dev = qml.device("aqt.sim", wires=num_wires, shots=shots, api_key=SOME_API_KEY)
 
         assert dev.num_wires == num_wires
         assert dev.shots.total_shots == shots
@@ -464,7 +466,8 @@ def test_device_gets_local_config(self, monkeypatch, tmpdir):
             "pennylane.default_config", qml.Configuration("config.toml")
         )  # force loading of config
 
-        dev = qml.device("aqt.sim", wires=2)
+        with pytest.warns(PennyLaneDeprecationWarning, match="shots on device is deprecated"):
+            dev = qml.device("aqt.sim", wires=2)
 
         assert dev.shots.total_shots == 99
         assert API_HEADER_KEY in dev.header.keys()
@@ -487,7 +490,8 @@ def test_device_gets_api_key_default_config_directory(self, monkeypatch, tmpdir)
         c = qml.Configuration("config.toml")
         monkeypatch.setattr("pennylane.default_config", c)  # force loading of config
 
-        dev = qml.device("aqt.sim", wires=2)
+        with pytest.warns(PennyLaneDeprecationWarning, match="shots on device is deprecated"):
+            dev = qml.device("aqt.sim", wires=2)
 
         assert API_HEADER_KEY in dev.header.keys()
         assert dev.header[API_HEADER_KEY] == SOME_API_KEY
@@ -506,7 +510,8 @@ def test_device_gets_api_key_pennylane_conf_env_var(self, monkeypatch, tmpdir):
             "pennylane.default_config", qml.Configuration("config.toml")
         )  # force loading of config
 
-        dev = qml.device("aqt.sim", wires=2)
+        with pytest.warns(PennyLaneDeprecationWarning, match="shots on device is deprecated"):
+            dev = qml.device("aqt.sim", wires=2)
 
         assert API_HEADER_KEY in dev.header.keys()
         assert dev.header[API_HEADER_KEY] == SOME_API_KEY
@@ -519,7 +524,8 @@ def test_device_gets_api_key_aqt_token_env_var(self, monkeypatch):
         monkeypatch.setenv("PENNYLANE_CONF", "")
         monkeypatch.setenv("AQT_TOKEN", SOME_API_KEY + "XYZ987")
 
-        dev = qml.device("aqt.sim", wires=2)
+        with pytest.warns(PennyLaneDeprecationWarning, match="shots on device is deprecated"):
+            dev = qml.device("aqt.sim", wires=2)
 
         assert API_HEADER_KEY in dev.header.keys()
         assert dev.header[API_HEADER_KEY] == NEW_API_KEY
@@ -528,8 +534,10 @@ def test_executes_with_online_api(self, monkeypatch):
         """Tests that a PennyLane QNode successfully executes with a
         mocked out online API."""
 
-        dev = qml.device("aqt.sim", wires=2, shots=10, api_key=SOME_API_KEY)
+        with pytest.warns(PennyLaneDeprecationWarning, match="shots on device is deprecated"):
+            dev = qml.device("aqt.sim", wires=2, api_key=SOME_API_KEY)
 
+        @qml.set_shots(10)
         @qml.qnode(dev)
         def circuit(x, y):
             qml.RX(x, wires=0)