class_docstr: Class to test Jax integration with classical Jacobian.
class_docstr_tok: ["Class","to","test","Jax","integration","with","classical","Jacobian","."]
docstr: Test ``classical_jacobian`` with ``argnum=<int>`` and JAX
  with ``trainable_only=False`` .
docstr_tok: ["Test","``","classical_jacobian","``","with","``","argnum=","<","int",">","``","and","JAX","with","``","trainable_only=False","``","."]
code: def test_jax_not_trainable_only(self, diff_method, interface):
      r"""Test ``classical_jacobian`` with ``argnum=<int>`` and JAX
      with ``trainable_only=False`` ."""
      dev = qml.device("default.qubit", wires=2)
      qnode = qml.QNode(
          circuit_0, dev, interface=interface, diff_method=diff_method
      )
      jac = classical_jacobian(qnode, argnum=0, trainable_only=False)(a)
      assert np.allclose(jac, expected_jac_not_trainable_only)
  
imports: import pytest
  import numpy
  import pennylane
  from pennylane import numpy
  from pennylane.gradients.classical_jacobian import classical_jacobian
label_desc: Use image node_id 5 for calling the TestJax obj's underlying member method code with example usage: obj.test_jax_not_trainable_only(diff_method, interface) without return types