From ba005ad7fae85480dcac9e684563ec45f408a38a Mon Sep 17 00:00:00 2001
From: a_corni <antoine.cornillot@pasqal.com>
Date: Thu, 15 Feb 2024 16:51:54 +0100
Subject: [PATCH] Fix tests

---
 tests/test_simulation.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/test_simulation.py b/tests/test_simulation.py
index 209b8515f..133372769 100644
--- a/tests/test_simulation.py
+++ b/tests/test_simulation.py
@@ -972,7 +972,10 @@ def test_noisy_xy(matrices, masked_qubit, noise, result, n_collapse_ops):
         "atom2": True,
         "atom3": False,
     }
-    assert len(sim._hamiltonian._collapse_ops) // 4 == n_collapse_ops
+    assert (
+        len(sim._hamiltonian._collapse_ops) // len(simple_reg.qubits)
+        == n_collapse_ops
+    )
     assert sim.run().sample_final_state() == Counter(result)