simplify givens rotation test

qiskit-community · May 8, 2024 · e139f89 · e139f89
1 parent b80e6a0
commit e139f89
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/python/linalg/givens_test.py b/tests/python/linalg/givens_test.py
@@ -53,11 +53,9 @@ def test_givens_decomposition_reconstruct(dim: int):
         for i, phase_shift in enumerate(phase_shifts):
             reconstructed[i] *= phase_shift
         for c, s, i, j in givens_rotations[::-1]:
-            reconstructed = reconstructed.T.copy()
-            reconstructed[j], reconstructed[i] = zrot(
-                reconstructed[j], reconstructed[i], c, s.conjugate()
+            reconstructed[:, j], reconstructed[:, i] = zrot(
+                reconstructed[:, j], reconstructed[:, i], c, s.conjugate()
             )
-            reconstructed = reconstructed.T
         np.testing.assert_allclose(reconstructed, mat)