class_docstr: Test that the state reconstruction is correct for a variety of states
class_docstr_tok: ["Test","that","the","state","reconstruction","is","correct","for","a","variety","of","states"]
docstr: Test that the state reconstruction is correct for a subset
  of the qubits
docstr_tok: ["Test","that","the","state","reconstruction","is","correct","for","a","subset","of","the","qubits"]
code: def test_partial_state(self, diffable):
      """Test that the state reconstruction is correct for a subset
      of the qubits"""
      wires_list = [[0], [0, 1]]
  
      circuit = max_entangled_circuit(3)
      circuit = qml.shadows.shadow_state(
          circuit, wires=wires_list, diffable=diffable
      )
  
      actual = circuit()
  
      expected = [
          np.array([[0.5, 0], [0, 0.5]]),
          np.array(
              [
                  [0.5, 0, 0, 0],
                  [0, 0, 0, 0],
                  [0, 0, 0, 0],
                  [0, 0, 0, 0.5],
              ]
          ),
      ]
  
      assert qml.math.allclose(actual[0], expected[0], atol=1e-1)
      assert qml.math.allclose(actual[1], expected[1], atol=1e-1)
  
imports: from functools import partial
  import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.shadows.transforms import _replace_obs
label_desc: Use image node_id 3 for calling the TestStateForward obj's underlying member method code with example usage: obj.test_partial_state(diffable) without return types