def test_get_state_histogram_multi_1():
qubits = cirq.LineQubit.range(4)
c = cirq.Circuit(
cirq.X.on_each(*qubits[1:]),
cirq.measure(*qubits) # One multi-qubit measurement
)
r = cirq.sample(c, repetitions=5)
values_to_plot = state_histogram.get_state_histogram(r)
expected_values = [0, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 0, 0]
np.testing.assert_equal(values_to_plot, expected_values)
def test_get_state_histogram():
simulator = cirq.Simulator()
q0 = GridQubit(0, 0)
q1 = GridQubit(1, 0)
circuit = cirq.Circuit()
circuit.append([cirq.X(q0), cirq.X(q1)])
circuit.append([cirq.measure(q0, key='q0'), cirq.measure(q1, key='q1')])
result = simulator.run(program=circuit, repetitions=5)
values_to_plot = state_histogram.get_state_histogram(result)
expected_values = [0.0, 0.0, 0.0, 5.0]
np.testing.assert_equal(values_to_plot, expected_values)
