Update test_braket_device.py

test/unit_tests/test_braket_device.py

@@ -1627,6 +1627,22 @@ def test_supported_ops_set(monkeypatch):
         assert dev.operations == test_ops
 
 
+def test_simplification():
+    """Test that the Projector observable is correctly supported."""
+    wires = 5
+    dev = BraketLocalQubitDevice(wires=wires)
+
+    obs = qml.ops.LinearCombination([1.0, 2.0], [qml.X(0) @ qml.I(1), qml.Y(0) @ qml.X(1)])
+
+    @qml.qnode(dev)
+    def circuit(x):
+        qml.RX(x, 0)
+        return qml.expval(obs)
+
+    phi = np.array(1.5, requires_grad=True)
+    assert np.isclose(circuit(phi), 0)
+
+
 def test_projection():
     """Test that the Projector observable is correctly supported."""
     wires = 2