add more gates for the tests

ShellyGarion · Nov 12, 2024 · ee3e88e · ee3e88e
1 parent 06c3da4
commit ee3e88e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/python/synthesis/test_synthesis.py b/test/python/synthesis/test_synthesis.py
@@ -46,6 +46,8 @@
     RZXGate,
     CPhaseGate,
     CRZGate,
+    CRXGate,
+    CRYGate,
     RXGate,
     RYGate,
     RZGate,
@@ -1426,7 +1428,7 @@ class TestTwoQubitControlledUDecompose(CheckDecompositions):
     def test_correct_unitary(self, seed):
         """Verify unitary for different gates in the decomposition"""
         unitary = random_unitary(4, seed=seed)
-        for gate in [RXXGate, RYYGate, RZZGate, RZXGate, CPhaseGate, CRZGate]:
+        for gate in [RXXGate, RYYGate, RZZGate, RZXGate, CPhaseGate, CRZGate, CRXGate, CRYGate]:
             decomposer = TwoQubitControlledUDecomposer(gate)
             circ = decomposer(unitary)
             self.assertEqual(Operator(unitary), Operator(circ))