code: def test_shape(shots, shape):
      """Test that the shape is correct."""
      dev = qml.device("default.qubit.legacy", wires=3, shots=shots)
      res = qml.mutual_info(wires0=[0], wires1=[1])
      assert res.shape(dev, Shots(shots)) == shape
  
imports: import numpy
  import pytest
  import pennylane
  from pennylane.workflow import INTERFACE_MAP
  from pennylane.measurements import Shots
label_desc: Use image node_id 1 for calling a global function with example usage: test_shape(shots, shape) without return types