class_docstr: Tests for the qml.simplify method used with callables.
class_docstr_tok: ["Tests","for","the","qml.simplify","method","used","with","callables","."]
docstr: Test that we can jit qnodes that have a simplified quantum function.
docstr_tok: ["Test","that","we","can","jit","qnodes","that","have","a","simplified","quantum","function","."]
code: def test_jitting_simplified_qfunc(self):
      """Test that we can jit qnodes that have a simplified quantum function."""
  
      import jax
  
      @jax.jit
      @qml.qnode(qml.device("default.qubit", wires=1))
      @qml.simplify
      def circuit(x):
          qml.adjoint(qml.RX(x, wires=0))
          _ = qml.PauliX(0) ** 2
          return qml.expval(qml.PauliY(0))
  
      x = jax.numpy.array(4 * jax.numpy.pi + 0.1)
      res = circuit(x)
      assert qml.math.allclose(res, jax.numpy.sin(x))
  
      grad = jax.grad(circuit)(x)
      assert qml.math.allclose(grad, jax.numpy.cos(x))
  
file_path: pennylane/tests/ops/functions/test_simplify.py
imports: import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.tape import QuantumScript
label_desc: Use image node_id 2 for calling the TestSimplifyCallables obj's underlying member method code with example usage: obj.test_jitting_simplified_qfunc() and returns: qml