added more test cases

tests/utils_test.py

@@ -1,7 +1,8 @@
 import numpy as np
 import unittest
 import cirq
-from qca.utils.utils import count_gates
+from qca.utils.utils import count_gates, count_T_gates, get_T_depth, get_T_depth_wire
+from pyLIQTR.gate_decomp.cirq_transforms import clifford_plus_t_direct_transform
 
 class UtilsTest(unittest.TestCase):
     def test_count_gates(self):
@@ -14,8 +15,44 @@ def test_count_gates(self):
         circuit.append(layer_rz)
         circuit.append(layer_rx)
         circuit.append(layer_measurement)
-
         self.assertEqual(count_gates(circuit), 12)
+
+    def test_T_gate_info(self):
+        qubits = [cirq.LineQubit(i) for i in range(4)]
+
+        layer_rz = [cirq.Rz(rads = np.pi/4).on(qubit) for qubit in qubits]
+        layer_rx = [cirq.Rx(rads = np.pi/4).on(qubit) for qubit in qubits]
+        layer_measurement = [cirq.measure(qubit) for qubit in qubits]
+        circuit = cirq.Circuit()
+        circuit.append(layer_rz)
+        circuit.append(layer_rx)
+        circuit.append(layer_measurement)
+        circ_cpt =  clifford_plus_t_direct_transform(circuit)
+
+        self.assertEqual(count_T_gates(circ_cpt), 8)
+        self.assertEqual(get_T_depth(circ_cpt), 2)
+        self.assertEqual(get_T_depth_wire(circ_cpt), 2)
+
+
+    def test_clifford_info(self):
+        qubits = [cirq.LineQubit(i) for i in range(4)]
+
+        layer_rz = [cirq.Rz(rads = np.pi/4).on(qubit) for qubit in qubits]
+        layer_rx = [cirq.Rx(rads = np.pi/4).on(qubit) for qubit in qubits]
+        layer_measurement = [cirq.measure(qubit) for qubit in qubits]
+        circuit = cirq.Circuit()
+        circuit.append(layer_rz)
+        circuit.append(layer_rx)
+        circuit.append(layer_measurement)
+        circ_cpt =  clifford_plus_t_direct_transform(circuit)
+
+        gate_count = count_gates(circ_cpt)
+        t_count = count_T_gates(circ_cpt)
+        clifford_count = gate_count - t_count
+        circ_depth = len(circ_cpt)
+        self.assertEqual(circ_depth, 5)
+        self.assertEqual(clifford_count, 12)
+
 
 if __name__ == '__main__':
     unittest.main()