test dictionary

quantumlib · NoureldinYosri · Nov 13, 2023 · Apr 2, 2023 · Apr 3, 2023 · Apr 4, 2023
commit 399ff683a57e61ecd85642e1e93c0a89dd9b1348
diff --git a/cirq-core/cirq/sim/classical_simulator_test.py b/cirq-core/cirq/sim/classical_simulator_test.py
@@ -27,9 +27,13 @@ def test_x_gate(self):
         circuit.append(cirq.measure((q0, q1), key='key'))
         expected_results = {'key': np.array([[[1, 0]]], dtype=np.uint8)}
         sim = cirq.ClassicalStateSimulator()
-        results = sim.run(circuit, param_resolver=None, repetitions=1).measurements
-        measurements_array = np.array([results[key] for key in sorted(results.keys())])
-        np.testing.assert_equal(measurements_array, expected_results)
+        results = sim.run(circuit, param_resolver=None, repetitions=1)
+        results_dict = {}
+        for key, measurements in results.measurements.items():
+            measurements_list = [inst.astype(np.uint8).tolist() for inst in results]
+            results_dict[key] = measurements_list
+
+        np.testing.assert_equal(results_dict, expected_results)
 
     def test_CNOT(self):
         q0, q1 = cirq.LineQubit.range(2)