class_docstr: Unit tests for the pwc function
class_docstr_tok: ["Unit","tests","for","the","pwc","function"]
docstr: Test that jax.jit can be used on the callable produced by pwc_from_function
docstr_tok: ["Test","that","jax.jit","can","be","used","on","the","callable","produced","by","pwc_from_function"]
code: def test_function_call_is_jittable(self):
      """Test that jax.jit can be used on the callable produced by pwc_from_function"""
      import jax
  
      f = qml.pulse.pwc(10)
      assert jax.jit(f)([1.2, 2.3], 2) != 0
      assert jax.jit(f)([1.2, 2.3], 13) == 0
  
imports: import inspect
  from functools import partial
  import numpy
  import pytest
  import pennylane
  from pennylane.pulse.parametrized_hamiltonian import ParametrizedHamiltonian
label_desc: Use image node_id 5 for calling the TestPWC obj's underlying member method code with example usage: obj.test_function_call_is_jittable() without return types