class_docstr: Tests for the state function
class_docstr_tok: ["Tests","for","the","state","function"]
docstr: Test that the shape is correct for qml.state with the shot vector too.
docstr_tok: ["Test","that","the","shape","is","correct","for","qml.state","with","the","shot","vector","too","."]
code: def test_shape_shot_vector(self, s_vec):
      """Test that the shape is correct for qml.state with the shot vector too."""
      dev = qml.device("default.qubit.legacy", wires=3, shots=s_vec)
      res = qml.state()
      assert res.shape(dev, Shots(s_vec)) == (
          (2**3,),
          (2**3,),
          (2**3,),
      )
  
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 18 for calling the TestState obj's underlying member method code with example usage: obj.test_shape_shot_vector(s_vec) without return types