Fix tests

pasqal-io · Feb 15, 2024 · ba005ad · ba005ad
1 parent c60eb9a
commit ba005ad
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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)