Update tests/test_pauli_string.py

Co-authored-by: James E T Smith <[email protected]>
qognitive · Aug 7, 2024 · 8064110 · 8064110
1 parent bbfbd4c
commit 8064110
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_pauli_string.py b/tests/test_pauli_string.py
@@ -248,9 +248,7 @@ def test_expected_value(
 
         coeff = generate_random_complex(1)[0]
         psis = generate_random_complex(n_dim, n_states)
-        expected = [
-            coeff * naive_pauli_converter(s).dot(psi).dot(psi.conj()) for psi in psis.T
-        ]
+        expected = np.einsum("ti,ij,tj->t", psi.conj(), naive_pauil_converter(s), psi)
         np.testing.assert_allclose(
             PauliString(s).expected_value(psis, coeff),
             expected,