fix: resolve diff

qiboteam · Oct 29, 2024 · eac4e14 · eac4e14
1 parent 1b3f5f8
commit eac4e14
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/test_backends.py b/tests/test_backends.py
@@ -63,10 +63,6 @@ def test_natives():
 
 def test_natives_no_cz_cnot():
     platform = create_platform("dummy")
-    for p in platform.pairs:
-        platform.pairs[p].native_gates.CZ = None
-        platform.pairs[p].native_gates.CNOT = None
-
     backend = QibolabBackend(platform)
     assert set(backend.natives) == {
         "I",
@@ -76,8 +72,15 @@ def test_natives_no_cz_cnot():
         "GPI2",
         "GPI",
         "M",
+        "CZ",
+        "CNOT",
     }
 
+    for gate in ["CZ", "CNOT"]:
+        for p in platform.pairs:
+            setattr(platform.pairs[p].native_gates, gate, None)
+        assert gate not in set(backend.natives)
+
 
 def test_execute_circuit_initial_state():
     backend = QibolabBackend("dummy")