code: def test_draw_batch_transform():
      """Test that drawing a batch transform works correctly."""
  
      @partial(qml.gradients.param_shift, shifts=[(0.2,)])
      @qml.qnode(qml.device("default.qubit", wires=1))
      def circ(x):
          qml.Hadamard(wires=0)
          qml.RX(x, wires=0)
          return qml.expval(qml.PauliZ(0))
  
      expected = "0: ──H──RX(0.8)─┤  <Z>\n\n0: ──H──RX(0.4)─┤  <Z>"
      assert (
          draw(circ, decimals=1)(np.array(0.6, requires_grad=True))
          == expected
      )
  
imports: from functools import partial
  import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.drawer import draw
label_desc: Use image node_id 2 for calling a global function with example usage: test_draw_batch_transform() and returns: qml