class_docstr: Tests for the probs function
class_docstr_tok: ["Tests","for","the","probs","function"]
docstr: Test that the shape is correct.
docstr_tok: ["Test","that","the","shape","is","correct","."]
code: def test_shape(self, wires, shots):
      """Test that the shape is correct."""
      dev = qml.device("default.qubit.legacy", wires=3, shots=shots)
      res = qml.probs(wires=wires)
      assert res.shape(dev, Shots(shots)) == (2 ** len(wires),)
  
imports: import numpy
  import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.measurements import ProbabilityMP, Shots
  from pennylane.devices.qubit.measure import flatten_state
label_desc: Use image node_id 2 for calling the TestProbs obj's underlying member method code with example usage: obj.test_shape(wires, shots) without return types