class_docstr: Tests for integration with the PassthruQNode
docstr: Tests correct output shape and evaluation for a tape
  with prob and expval outputs
code: def test_ragged_differentiation(
      self, dev_name, diff_method, grad_on_execution, tol
  ):
      """Tests correct output shape and evaluation for a tape
      with prob and expval outputs"""
  
      if diff_method == "finite-diff":
          config.config.update("jax_enable_x64", True)
  
      dev = qml.device(dev_name, wires=2)
      x = jnp.array(0.543)
      y = jnp.array(-0.654)
  
      @qml.qnode(
          dev,
          diff_method=diff_method,
          grad_on_execution=grad_on_execution,
          interface="jax",
      )
      def circuit(x, y):
          qml.RX(x, wires=[0])
          qml.RY(y, wires=[1])
          qml.CNOT(wires=[0, 1])
          return qml.expval(qml.PauliZ(0)), qml.probs(wires=[1])
  
      res = circuit(x, y)
      expected = np.array(
          [
              np.cos(x),
              (1 + np.cos(x) * np.cos(y)) / 2,
              (1 - np.cos(x) * np.cos(y)) / 2,
          ]
      )
  
      assert np.allclose(
          qml.math.hstack(res), expected, atol=tol, rtol=0
      )
  
      res = jax.jacobian(circuit, (0, 1))(x, y)
  
      expected = np.array(
          [
              [
                  -np.sin(x),
                  -np.sin(x) * np.cos(y) / 2,
                  np.cos(y) * np.sin(x) / 2,
              ],
              [
                  0,
                  -np.cos(x) * np.sin(y) / 2,
                  np.cos(x) * np.sin(y) / 2,
              ],
          ]
      )
  
      assert np.allclose(
          qml.math.hstack([res[0][0], res[1][0]]),
          expected[0],
          atol=tol,
          rtol=0,
      )
      assert np.allclose(
          qml.math.hstack([res[0][1], res[1][1]]),
          expected[1],
          atol=tol,
          rtol=0,
      )
  
imports: from functools import partial
  import pytest
  import numpy
  import pennylane
  from pennylane import numpy
  from pennylane.devices.default_mixed import DefaultMixed
label_desc: Use image node_id 13 for calling the TestPassthruIntegration obj's underlying member method code with example usage: obj.test_ragged_differentiation(dev_name, diff_method, grad_on_execution, tol) and returns: qml,  qml