Update tests/test_backends.py

Co-authored-by: Alessandro Candido <candido.ale@gmail.com>
qiboteam · Oct 25, 2024 · 93ec160 · 93ec160
1 parent 6471dfb
commit 93ec160
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/test_backends.py b/tests/test_backends.py
@@ -62,10 +62,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",
@@ -75,7 +71,14 @@ 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():