class_docstr: Tests for the probs function
class_docstr_tok: ["Tests","for","the","probs","function"]
docstr: Tests that process_state functions as expected with all interfaces with batching.
code: def test_process_state_batched(
      self, interface, subset_wires, expected
  ):
      """Tests that process_state functions as expected with all interfaces with batching."""
      states = qml.math.array(
          [[1 / 2, 0] * 4, [0, 1 / 2] * 4], like=interface
      )
      wires = qml.wires.Wires(range(3))
      subset_probs = qml.probs(wires=subset_wires).process_state(
          states, wires
      )
      assert subset_probs.shape == qml.math.shape(expected)
      assert qml.math.allclose(subset_probs, expected)
  
imports: import numpy
  import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.measurements import MeasurementProcess, Probability, ProbabilityMP, Shots
  from pennylane.queuing import AnnotatedQueue
label_desc: Use image node_id 13 for calling the TestProbs obj's underlying member method code with example usage: obj.test_process_state_batched(interface, subset_wires, expected) without return types