class_docstr: Test that all operations in the ``supports_broadcasting`` attribute
  actually support broadcasting.
docstr: Test that AmplitudeEmbedding, which is marked as supporting parameter broadcasting,
  actually does support broadcasting.
docstr_tok: ["Test","that","AmplitudeEmbedding",",","which","is","marked","as","supporting","parameter","broadcasting",",","actually","does","support","broadcasting","."]
code: def test_amplitude_embedding(self, state, num_wires):
      """Test that AmplitudeEmbedding, which is marked as supporting parameter broadcasting,
      actually does support broadcasting."""
  
      features = np.array([state])
      op = qml.AmplitudeEmbedding(
          features, wires=list(range(num_wires))
      )
      assert op.batch_size == 1
      qml.AmplitudeEmbedding.compute_decomposition(
          features, list(range(num_wires))
      )
      op.decomposition()
  
      features = np.array([state] * 3)
      op = qml.AmplitudeEmbedding(
          features, wires=list(range(num_wires))
      )
      assert op.batch_size == 3
      qml.AmplitudeEmbedding.compute_decomposition(
          features, list(range(num_wires))
      )
      op.decomposition()
  
imports: import itertools
  import pytest
  import numpy
  from scipy.stats import unitary_group
  import pennylane
  from pennylane.ops.qubit.attributes import Attribute, has_unitary_generator
  from pennylane.operation import AnyWires
label_desc: Use image node_id 14 for calling the TestSupportsBroadcasting obj's underlying member method code with example usage: obj.test_amplitude_embedding(state, num_wires) without return types