class_docstr: Tests class attributes and methods.
class_docstr_tok: ["Tests","class","attributes","and","methods","."]
docstr: Test that the shape method returns the correct shape of the weights tensor
docstr_tok: ["Test","that","the","shape","method","returns","the","correct","shape","of","the","weights","tensor"]
code: def test_shape(self, n_wires, expected_shape):
      """Test that the shape method returns the correct shape of the weights tensor"""
  
      shape = qml.ArbitraryUnitary.shape(n_wires)
      assert shape == expected_shape
  
imports: import pytest
  import numpy
  from pennylane import numpy
  import pennylane
  from pennylane.templates.subroutines.arbitrary_unitary import _all_pauli_words_but_identity, _tuple_to_word, _n_k_gray_code
label_desc: Use image node_id 1 for calling the TestAttributes obj's underlying member method code with example usage: obj.test_shape(n_wires, expected_shape) without return types