def test_simulation_retrieve_samples(batch_size):
simulation = StateVectorSimulation(2, 10000, batch_size)
simulation.evolve(
[gate_operations.Hadamard([0]),
gate_operations.CX([0, 1])])
counter = Counter(simulation.retrieve_samples())
assert simulation.qubit_count == 2
assert counter.keys() == {0, 3}
assert 0.4 < counter[0] / (counter[0] + counter[3]) < 0.6
assert 0.4 < counter[3] / (counter[0] + counter[3]) < 0.6
assert counter[0] + counter[3] == 10000
def _formatted_measurements(
simulation: StateVectorSimulation) -> List[List[str]]:
""" Retrieves formatted measurements obtained from the specified simulation.
Args:
simulation (StateVectorSimulation): Simulation to use for obtaining the measurements.
Returns:
List[List[str]]: List containing the measurements, where each measurement consists
of a list of measured values of qubits.
"""
return [
list("{number:0{width}b}".format(number=sample,
width=simulation.qubit_count))
for sample in simulation.retrieve_samples()
]