code: def expected_shapes(n_layers, n_wires):
      # compute the expected shapes for a given number of wires
      n_if = n_wires * (n_wires - 1) // 2
      expected = (
          [(n_layers, n_if)] * 2
          + [(n_layers, n_wires)] * 3
          + [(n_layers, n_if)] * 2
          + [(n_layers, n_wires)] * 4
      )
      return expected
  
imports: import pytest
  import numpy
  import pennylane
  from pennylane import numpy
  from pennylane.devices import DefaultGaussian
label_desc: Use image node_id 1 for calling a global function with example usage: expected_shapes(n_layers, n_wires) and returns: expected