test: simplify array expectation

rigetti · Sep 14, 2023 · 21f7636 · 21f7636
1 parent 82963e2
commit 21f7636
Showing 1 changed file with 2 additions and 16 deletions.
diff --git a/pyquil/paulis.py b/pyquil/paulis.py
@@ -988,29 +988,15 @@ def exponentiate_pauli_sum(
     >>> coeff = phi/(-4*pi) # -0.25
     >>> hamiltonian = PauliTerm("Z", 0) * PauliTerm("Z", 1) - 1*PauliTerm("Z", 0) - 1*PauliTerm("Z", 1)
     >>> exponentiate_pauli_sum(coeff*hamiltonian)
-    array([[ 1.-1.11022302e-16j,  0.+0.00000000e+00j,  0.+0.00000000e+00j,
-             0.+0.00000000e+00j],
-           [ 0.+0.00000000e+00j,  1.-1.11022302e-16j,  0.+0.00000000e+00j,
-             0.+0.00000000e+00j],
-           [ 0.+0.00000000e+00j,  0.+0.00000000e+00j,  1.-1.11022302e-16j,
-             0.+0.00000000e+00j],
-           [ 0.+0.00000000e+00j,  0.+0.00000000e+00j,  0.+0.00000000e+00j,
-            -1.+2.22044605e-16j]])
+    array([[...]])
 
     To produce the Quil XY(theta) gate, you can use:
 
     >>> theta = pi/2
     >>> coeff = theta/(-4*pi)
     >>> hamiltonian = PauliTerm("X", 0) * PauliTerm("X", 1) + PauliTerm("Y", 0) * PauliTerm("Y", 1)
     >>> exponentiate_pauli_sum(coeff*hamiltonian)
-    array([[1.        +0.j        , 0.        +0.j        ,
-            0.        +0.j        , 0.        +0.j        ],
-           [0.        +0.j        , 0.70710678+0.j        ,
-            0.        +0.70710678j, 0.        +0.j        ],
-           [0.        +0.j        , 0.        +0.70710678j,
-            0.70710678+0.j        , 0.        +0.j        ],
-           [0.        +0.j        , 0.        +0.j        ,
-            0.        +0.j        , 1.        +0.j        ]])
+    array([[...]])
 
     A global phase is applied to the unitary such that the [0,0] element is always real.