code: def test_angle_embedding(mocker):
      """Test that batching works for AngleEmbedding"""
      dev = qml.device("default.qubit", wires=3)
  
      @qml.batch_params
      @qml.qnode(dev, interface="autograd")
      def circuit(data):
          qml.templates.AngleEmbedding(data, wires=[0, 1, 2])
          qml.RY(0.2, wires=1)
          return qml.probs(wires=[0, 2])
  
      batch_size = 5
      data = np.random.random((batch_size, 3))
  
      spy = mocker.spy(circuit.device, "execute")
      res = circuit(data)
      assert res.shape == (batch_size, 4)
      assert len(spy.call_args[0][0]) == batch_size
  
imports: import functools
  import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 5 for calling a global function with example usage: test_angle_embedding(mocker) and returns: qml