def METHOD_NAME(self):
    """Test QNode in autograd interface."""
    def qfunc(amplitude):
        qml.AmplitudeEmbedding(amplitude, wires=0)
        qml.AmplitudeEmbedding(amplitude, wires=1)
        return qml.state()
    dev = qml.device("default.qubit", wires=2)
    optimized_qfunc = qml.transforms.merge_amplitude_embedding(qfunc)
    optimized_qnode = qml.QNode(optimized_qfunc, dev)
    amplitude = np.array([0.0, 1.0], requires_grad=True)
    # Check the state |11> is being generated.
    assert optimized_qnode(amplitude)[-1] == 1