class_docstr: Tests for the state function
class_docstr_tok: ["Tests","for","the","state","function"]
docstr: Test that the state is of correct size and dtype for a trivial circuit
docstr_tok: ["Test","that","the","state","is","of","correct","size","and","dtype","for","a","trivial","circuit"]
code: def test_state_shape_and_dtype(self, wires):
      """Test that the state is of correct size and dtype for a trivial circuit"""
  
      dev = qml.device("default.qubit.legacy", wires=wires)
  
      @qml.qnode(dev)
      def func():
          return state()
  
      state_val = func()
      assert state_val.shape == (2**wires,)
      assert state_val.dtype == np.complex128
  
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 1 for calling the TestState obj's underlying member method code with example usage: obj.test_state_shape_and_dtype(wires) and returns: state