class_docstr: Test that using vector-valued QNodes with JAX integrate with the
  PennyLane stack
class_docstr_tok: ["Test","that","using","vector-valued","QNodes","with","JAX","integrate","with","the","PennyLane","stack"]
docstr: Tests correct output shape and evaluation for a tape
  with a single prob output
docstr_tok: ["Tests","correct","output","shape","and","evaluation","for","a","tape","with","a","single","prob","output"]
code: def test_diff_single_probs(
      self, dev_name, diff_method, grad_on_execution, interface, tol
  ):
      """Tests correct output shape and evaluation for a tape
      with a single prob output"""
      kwargs = dict(
          diff_method=diff_method,
          interface=interface,
          grad_on_execution=grad_on_execution,
      )
      if diff_method == "adjoint":
          pytest.skip("Adjoint does not support probs")
      elif diff_method == "spsa":
          kwargs["sampler_rng"] = np.random.default_rng(SEED_FOR_SPSA)
          tol = TOL_FOR_SPSA
  
      num_wires = 2
  
      if diff_method == "hadamard":
          num_wires = 3
  
      dev = qml.device(dev_name, wires=num_wires)
      x = jax.numpy.array(0.543)
      y = jax.numpy.array(-0.654)
  
      @qnode(dev, **kwargs)
      def circuit(x, y):
          qml.RX(x, wires=[0])
          qml.RY(y, wires=[1])
          qml.CNOT(wires=[0, 1])
          return qml.probs(wires=[1])
  
      res = jax.jacobian(circuit, argnums=[0, 1])(x, y)
  
      expected = np.array(
          [
              [-np.sin(x) * np.cos(y) / 2, -np.cos(x) * np.sin(y) / 2],
              [np.cos(y) * np.sin(x) / 2, np.cos(x) * np.sin(y) / 2],
          ]
      )
  
      assert isinstance(res, tuple)
      assert len(res) == 2
  
      assert isinstance(res[0], jax.numpy.ndarray)
      assert res[0].shape == (2,)
  
      assert isinstance(res[1], jax.numpy.ndarray)
      assert res[1].shape == (2,)
  
      assert np.allclose(res[0], expected.T[0], atol=tol, rtol=0)
      assert np.allclose(res[1], expected.T[1], atol=tol, rtol=0)
  
imports: import pytest
  import pennylane
  from pennylane import numpy
  from pennylane import qnode
  from pennylane.tape import QuantumScript
label_desc: Use image node_id 3 for calling the TestVectorValuedQNode obj's underlying member method code with example usage: obj.test_diff_single_probs(dev_name, diff_method, grad_on_execution, interface, tol) and returns: qml