class_docstr: Tests for the state function
class_docstr_tok: ["Tests","for","the","state","function"]
docstr: Test the state when custom wire labels are used
docstr_tok: ["Test","the","state","when","custom","wire","labels","are","used"]
code: def test_custom_wire_labels(self, wires):
      """Test the state when custom wire labels are used"""
      dev = qml.device("default.qubit.legacy", wires=wires)
  
      @qml.qnode(dev, diff_method="parameter-shift")
      def func():
          for i in range(4):
              qml.Hadamard(wires[i])
          return state()
  
      state_expected = 0.25 * np.ones(16)
      state_val = func()
  
      assert np.allclose(state_expected, state_val)
  
imports: import numpy
  import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.devices import DefaultQubitLegacy
  from pennylane.measurements import State, Shots, density_matrix, expval, state
label_desc: Use image node_id 16 for calling the TestState obj's underlying member method code with example usage: obj.test_custom_wire_labels(wires) and returns: state