Update quread/gates.py

quread/gates.py

@@ -1,7 +1,9 @@
 # quread/gates.py
 import numpy as np
 
-I2 = np.eye(2, dtype=complex)
+# Identity
+I = np.array([[1, 0],
+              [0, 1]], dtype=complex)
 
 H = (1 / np.sqrt(2)) * np.array([[1, 1], [1, -1]], dtype=complex)
 X = np.array([[0, 1], [1, 0]], dtype=complex)
@@ -18,7 +20,7 @@ Tdg = np.array([[1, 0], [0, np.exp(-1j * np.pi / 4)]], dtype=complex)
 SQRTX = 0.5 * np.array([[1 + 1j, 1 - 1j],
                         [1 - 1j, 1 + 1j]], dtype=complex)
 
-# √Z (principal square root): diag(1, e^{iπ/2}) = diag(1, i) == S
+# √Z (principal square root): diag(1, i) == S
 SQRTZ = S
 
 
@@ -41,50 +43,30 @@ def RZ(theta: float) -> np.ndarray:
                      [0, np.exp(1j * theta / 2)]], dtype=complex)
 
 
-def single_qubit_gate_matrix(gate: str) -> np.ndarray:
-    """
-    Supported gate strings:
-    H, X, Y, Z, S, Sdg, T, Tdg, SQRTX, SQRTZ,
-    RX_PI, RX_PI_2, RY_PI, RY_PI_2, RZ_PI, RZ_PI_2,
-    I
-    """
-    g = gate.strip()
-
-    if g == "I":
-        return I2
-    if g == "H":
-        return H
-    if g == "X":
-        return X
-    if g == "Y":
-        return Y
-    if g == "Z":
-        return Z
-    if g == "S":
-        return S
-    if g in ("S†", "Sdg"):
-        return Sdg
-    if g == "T":
-        return T
-    if g in ("T†", "Tdg"):
-        return Tdg
-    if g in ("√X", "SQRTX"):
-        return SQRTX
-    if g in ("√Z", "SQRTZ"):
-        return SQRTZ
-
-    # rotations (fixed angles from your palette)
-    if g == "RX(π)":
-        return RX(np.pi)
-    if g == "RX(π/2)":
-        return RX(np.pi / 2)
-    if g == "RY(π)":
-        return RY(np.pi)
-    if g == "RY(π/2)":
-        return RY(np.pi / 2)
-    if g == "RZ(π)":
-        return RZ(np.pi)
-    if g == "RZ(π/2)":
-        return RZ(np.pi / 2)
-
-    raise ValueError(f"Unsupported gate: {gate}")
+# --- aliases expected by engine.py ---
+def rx(theta: float) -> np.ndarray:
+    return RX(theta)
+
+
+def ry(theta: float) -> np.ndarray:
+    return RY(theta)
+
+
+def rz(theta: float) -> np.ndarray:
+    return RZ(theta)
+
+
+# --- map expected by engine.py ---
+SINGLE_QUBIT_GATES = {
+    "I": I,
+    "H": H,
+    "X": X,
+    "Y": Y,
+    "Z": Z,
+    "S": S,
+    "Sdg": Sdg,
+    "T": T,
+    "Tdg": Tdg,
+    "SQRTX": SQRTX,
+    "SQRTZ": SQRTZ,
+}