class_docstr: Test that using the QNode with JAX integrates with the PennyLane
  stack
class_docstr_tok: ["Test","that","using","the","QNode","with","JAX","integrates","with","the","PennyLane","stack"]
docstr: Test classical processing within the quantum tape
docstr_tok: ["Test","classical","processing","within","the","quantum","tape"]
code: def test_classical_processing(
      self, dev, diff_method, grad_on_execution, device_vjp, interface
  ):
      """Test classical processing within the quantum tape"""
      a = jax.numpy.array(0.1)
      b = jax.numpy.array(0.2)
      c = jax.numpy.array(0.3)
  
      @qnode(
          dev,
          diff_method=diff_method,
          interface=interface,
          grad_on_execution=grad_on_execution,
          device_vjp=device_vjp,
      )
      def circuit(a, b, c):
          qml.RY(a * c, wires=0)
          qml.RZ(b, wires=0)
          qml.RX(c + c**2 + jax.numpy.sin(a), wires=0)
          return qml.expval(qml.PauliZ(0))
  
      res = jax.grad(circuit, argnums=[0, 2])(a, b, c)
  
      if diff_method == "finite-diff":
          assert circuit.qtape.trainable_params == [0, 2]
  
      assert len(res) == 2
  
imports: from itertools import product
  import numpy
  import pytest
  import pennylane
  from pennylane import qnode
  from pennylane.devices import DefaultQubit
label_desc: Use image node_id 3 for calling the TestQNode obj's underlying member method code with example usage: obj.test_classical_processing(dev, diff_method, grad_on_execution, device_vjp, interface) and returns: qml