class_docstr: Integration tests for the ParametrizedEvolution class.
class_docstr_tok: ["Integration","tests","for","the","ParametrizedEvolution","class","."]
code: def test_time_input_formats(
      self, device_class, time, time_interface, use_jit
  ):
      import jax
      import jax.numpy as jnp
  
      if time_interface == "jax":
          time = jnp.array(time)
      elif time_interface == "numpy":
          time = np.array(time)
      H = qml.pulse.ParametrizedHamiltonian([2], [qml.PauliX(0)])
  
      dev = device_class(wires=1)
  
      @qml.qnode(dev, interface="jax")
      def circuit(t):
          qml.evolve(H)([], t)
          return qml.expval(qml.PauliZ(0))
  
      if use_jit:
          circuit = jax.jit(circuit)
  
      res = circuit(time)
      duration = time if qml.math.ndim(time) == 0 else time[1] - time[0]
      assert qml.math.isclose(res, qml.math.cos(4 * duration))
  
imports: from functools import reduce
  import numpy
  import pytest
  import pennylane
  from pennylane.operation import AnyWires
  from pennylane.ops import QubitUnitary
  from pennylane.pulse import ParametrizedEvolution, ParametrizedHamiltonian
  from pennylane.tape import QuantumTape
  from pennylane.devices import DefaultQubit, DefaultQubitLegacy
label_desc: Use image node_id 1 for calling the TestIntegration obj's underlying member method code with example usage: obj.test_time_input_formats(device_class, time, time_interface, use_jit) and returns: qml