From 124a2da17c8b62551dc1d039c1c8add03c4b7732 Mon Sep 17 00:00:00 2001 From: anishbhujbal7 Date: Wed, 16 Sep 2026 06:47:34 +0000 Subject: [PATCH] fix: pass explicit qubit_order to final_state_vector in state_preparation_test --- .../circuits/primitives/state_preparation_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/openfermion/circuits/primitives/state_preparation_test.py b/src/openfermion/circuits/primitives/state_preparation_test.py index e9bc83a42..9c1375281 100644 --- a/src/openfermion/circuits/primitives/state_preparation_test.py +++ b/src/openfermion/circuits/primitives/state_preparation_test.py @@ -57,7 +57,7 @@ def test_prepare_gaussian_state( ) if isinstance(initial_state, list): initial_state = sum(1 << (n_qubits - 1 - i) for i in initial_state) - state = circuit.final_state_vector(initial_state=initial_state) + state = circuit.final_state_vector(initial_state=initial_state, qubit_order=qubits) # Check that the result is an eigenstate with the correct eigenvalue numpy.testing.assert_allclose(quad_ham_sparse.dot(state), energy * state, atol=atol) @@ -104,7 +104,7 @@ def test_prepare_gaussian_state_with_spin_symmetry( if isinstance(initial_state, list): initial_state = sum(1 << (n_qubits - 1 - i) for i in initial_state) - state = circuit.final_state_vector(initial_state=initial_state) + state = circuit.final_state_vector(initial_state=initial_state, qubit_order=qubits) # Check that the result is an eigenstate with the correct eigenvalue numpy.testing.assert_allclose(quad_ham_sparse.dot(state), energy * state, atol=atol) @@ -150,6 +150,6 @@ def test_prepare_slater_determinant( ) if isinstance(initial_state, list): initial_state = sum(1 << (n_qubits - 1 - i) for i in initial_state) - state = circuit.final_state_vector(initial_state=initial_state) + state = circuit.final_state_vector(initial_state=initial_state, qubit_order=qubits) assert cirq.allclose_up_to_global_phase(state, correct_state, atol=atol)