class_docstr: Unit tests for the counts.process_samples method
docstr: Test that the counts function outputs counts of the right size
  for multiple wires
docstr_tok: ["Test","that","the","counts","function","outputs","counts","of","the","right","size","for","multiple","wires"]
code: def test_counts_shape_multi_wires(self):
      """Test that the counts function outputs counts of the right size
      for multiple wires"""
      shots = 1000
      samples = np.random.choice([0, 1], size=(shots, 2)).astype(
          np.int64
      )
  
      result = qml.counts(wires=[0, 1]).process_samples(
          samples, wire_order=[0, 1]
      )
  
      assert len(result) == 4
      assert set(result.keys()) == {"00", "01", "10", "11"}
      assert result["00"] == np.count_nonzero(
          np.logical_and(samples[:, 0] == 0, samples[:, 1] == 0)
      )
      assert result["01"] == np.count_nonzero(
          np.logical_and(samples[:, 0] == 0, samples[:, 1] == 1)
      )
      assert result["10"] == np.count_nonzero(
          np.logical_and(samples[:, 0] == 1, samples[:, 1] == 0)
      )
      assert result["11"] == np.count_nonzero(
          np.logical_and(samples[:, 0] == 1, samples[:, 1] == 1)
      )
  
imports: import copy
  import numpy
  import pytest
  import pennylane
  from pennylane.measurements import AllCounts, Counts, CountsMP
  from pennylane.wires import Wires
label_desc: Use image node_id 2 for calling the TestProcessSamples obj's underlying member method code with example usage: obj.test_counts_shape_multi_wires() without return types