cirq/ops/pauli_string_test.py

891     q_map = {q0: 0, q1: 1}
895         im_ps.expectation_from_state_vector(wf, q_map)
898         ps.expectation_from_state_vector(np.array([1, 0], dtype=int), q_map)
924         ps.expectation_from_state_vector(np.arange(7, dtype=np.complex64), q_map)
931     _ = ps.expectation_from_state_vector(rho_or_wf, q_map)
943     q_map = {q0: 0, q1: 1, q2: 2, q3: 3}
946         ps.expectation_from_state_vector(np.arange(16, dtype=np.complex64), q_map)
949         np.arange(16, dtype=np.complex64), q_map, check_preconditions=False
956     q_map = {q0: 0}
959         x0.expectation_from_state_vector(np.array([1, 0], dtype=complex), q_map), 0, atol=1e-7
962         x0.expectation_from_state_vector(np.array([0, 1], dtype=complex), q_map), 0, atol=1e-7
965         x0.expectation_from_state_vector(np.array([1, 1], dtype=complex) / np.sqrt(2), q_map),
970         x0.expectation_from_state_vector(np.array([1, -1], dtype=complex) / np.sqrt(2), q_map),
977         y0.expectation_from_state_vector(np.array([1, 1j], dtype=complex) / np.sqrt(2), q_map),
982         y0.expectation_from_state_vector(np.array([1, -1j], dtype=complex) / np.sqrt(2), q_map),
987         y0.expectation_from_state_vector(np.array([1, 1], dtype=complex) / np.sqrt(2), q_map),
992         y0.expectation_from_state_vector(np.array([1, -1], dtype=complex) / np.sqrt(2), q_map),
1004     q_map = {q0: 0, q1: 1}
1007         np.testing.assert_allclose(z0z1.expectation_from_state_vector(state, q_map), -1)
1008         np.testing.assert_allclose(x0x1.expectation_from_state_vector(state, q_map), 1)
1012         np.testing.assert_allclose(z0z1.expectation_from_state_vector(state, q_map), 1)
1013         np.testing.assert_allclose(x0x1.expectation_from_state_vector(state, q_map), 1)
1017         np.testing.assert_allclose(z0z1.expectation_from_state_vector(state, q_map), 0)
1018         np.testing.assert_allclose(x0x1.expectation_from_state_vector(state, q_map), 1)
1040     q_map = {q: i for i, q in enumerate(qubits)}
1059         np.testing.assert_allclose(z0z1.expectation_from_state_vector(state, q_map), -1)
1060         np.testing.assert_allclose(z0z2.expectation_from_state_vector(state, q_map), 0)
1061         np.testing.assert_allclose(z0z3.expectation_from_state_vector(state, q_map), 0)
1062         np.testing.assert_allclose(z0x1.expectation_from_state_vector(state, q_map), 0)
1063         np.testing.assert_allclose(z1x2.expectation_from_state_vector(state, q_map), -1)
1064         np.testing.assert_allclose(x0z1.expectation_from_state_vector(state, q_map), 0)
1065         np.testing.assert_allclose(x3.expectation_from_state_vector(state, q_map), -1)
1070     q_map = {q: i for i, q in enumerate(qs)}
1085         np.testing.assert_allclose(z0z1.expectation_from_state_vector(state, q_map), -0.123)
1086         np.testing.assert_allclose(z0z2.expectation_from_state_vector(state, q_map), 0)
1087         np.testing.assert_allclose(z1x2.expectation_from_state_vector(state, q_map), 1)
1095     q_map = {q0: 0, q1: 1}
1099         im_ps.expectation_from_density_matrix(rho, q_map)
1102         ps.expectation_from_density_matrix(0.5 * np.eye(2, dtype=int), q_map)
1127         ps.expectation_from_density_matrix(1j * np.eye(4), q_map)
1129         ps.expectation_from_density_matrix(np.eye(4, dtype=np.complex64), q_map)
1135             q_map,
1140         ps.expectation_from_density_matrix(np.ones((4, 5), dtype=np.complex64), q_map)
1149         ps.expectation_from_density_matrix(np.array([1, 0], dtype=np.complex64), q_map)
1151         ps.expectation_from_density_matrix(wf, q_map)
1156     _ = ps.expectation_from_density_matrix(rho_or_wf, q_map)
1162     q_map = {q0: 0, q1: 1}
1169             q_map,
1174         q_map,
1183     q_map = {q0: 0}
1185         x0.expectation_from_density_matrix(np.array([[1, 0], [0, 0]], dtype=complex), q_map), 0
1188         x0.expectation_from_density_matrix(np.array([[0, 0], [0, 1]], dtype=complex), q_map), 0
1191         x0.expectation_from_density_matrix(np.array([[1, 1], [1, 1]], dtype=complex) / 2, q_map),
1195         x0.expectation_from_density_matrix(np.array([[1, -1], [-1, 1]], dtype=complex) / 2, q_map),
1206     q_map = {q0: 0, q1: 1}
1211         np.testing.assert_allclose(z0z1.expectation_from_density_matrix(state, q_map), -1)
1212         np.testing.assert_allclose(x0x1.expectation_from_density_matrix(state, q_map), 1)
1217         np.testing.assert_allclose(z0z1.expectation_from_density_matrix(state, q_map), 1)
1218         np.testing.assert_allclose(x0x1.expectation_from_density_matrix(state, q_map), 1)
1223         np.testing.assert_allclose(z0z1.expectation_from_density_matrix(state, q_map), 0)
1224         np.testing.assert_allclose(x0x1.expectation_from_density_matrix(state, q_map), 1)
1256     q_map = {q: i for i, q in enumerate(qubits)}
1276         np.testing.assert_allclose(z0z1.expectation_from_density_matrix(state, q_map), -1)
1277         np.testing.assert_allclose(z0z2.expectation_from_density_matrix(state, q_map), 0)
1278         np.testing.assert_allclose(z0z3.expectation_from_density_matrix(state, q_map), 0)
1279         np.testing.assert_allclose(z0x1.expectation_from_density_matrix(state, q_map), 0)
1280         np.testing.assert_allclose(z1x2.expectation_from_density_matrix(state, q_map), -1)
1281         np.testing.assert_allclose(x0z1.expectation_from_density_matrix(state, q_map), 0)
1282         np.testing.assert_allclose(x3.expectation_from_density_matrix(state, q_map), -1)
1287     q_map = {q: i for i, q in enumerate(qs)}
1303         np.testing.assert_allclose(z0z1.expectation_from_density_matrix(state, q_map), -0.123)
1304         np.testing.assert_allclose(z0z2.expectation_from_density_matrix(state, q_map), 0)
1305         np.testing.assert_allclose(z1x2.expectation_from_density_matrix(state, q_map), 1)
1318     q_map = {q: i for i, q in enumerate(qubits)}
1322     a = pauli_string.expectation_from_state_vector(state_vector1, q_map)
1323     b = pauli_string.expectation_from_state_vector(state_vector2, q_map)
1324     c = pauli_string.expectation_from_density_matrix(density_matrix, q_map)