fix tests by using complex dtype for orbital rotations

qiskit-community · Oct 13, 2024 · 3b850d3 · 3b850d3
1 parent 681012c
commit 3b850d3
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/python/variational/ucj_test.py b/tests/python/variational/ucj_test.py
@@ -35,9 +35,11 @@ def test_n_params():
     ):
         diag_coulomb_mats_alpha_alpha = np.zeros((n_reps, norb, norb))
         diag_coulomb_mats_alpha_beta = np.zeros((n_reps, norb, norb))
-        orbital_rotations = np.stack([np.eye(norb) for _ in range(n_reps)])
+        orbital_rotations = np.stack(
+            [np.eye(norb, dtype=complex) for _ in range(n_reps)]
+        )
 
-        final_orbital_rotation = np.eye(norb)
+        final_orbital_rotation = np.eye(norb, dtype=complex)
         operator = ffsim.UCJOperator(
             diag_coulomb_mats_alpha_alpha=diag_coulomb_mats_alpha_alpha,
             diag_coulomb_mats_alpha_beta=diag_coulomb_mats_alpha_beta,
@@ -254,9 +256,11 @@ def test_real_ucj_n_params():
     ):
         diag_coulomb_mats_alpha_alpha = np.zeros((n_reps, norb, norb))
         diag_coulomb_mats_alpha_beta = np.zeros((n_reps, norb, norb))
-        orbital_rotations = np.stack([np.eye(norb) for _ in range(n_reps)])
+        orbital_rotations = np.stack(
+            [np.eye(norb, dtype=complex) for _ in range(n_reps)]
+        )
 
-        final_orbital_rotation = np.eye(norb)
+        final_orbital_rotation = np.eye(norb, dtype=complex)
         operator = ffsim.RealUCJOperator(
             diag_coulomb_mats_alpha_alpha=diag_coulomb_mats_alpha_alpha,
             diag_coulomb_mats_alpha_beta=diag_coulomb_mats_alpha_beta,