inline some signatures

CQCL · Aug 12, 2024 · 502f3de · 502f3de
1 parent cfc0fdb
commit 502f3de
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/hugr-py/tests/conftest.py b/hugr-py/tests/conftest.py
@@ -20,40 +20,35 @@
     from hugr.ops import ComWire
 
 EXTENSION = ext.Extension("pytest.quantum,", ext.Version(0, 1, 0))
-_SINGLE_QUBIT = ext.OpDefSig(tys.FunctionType.endo([tys.Qubit]))
-_TWO_QUBIT = ext.OpDefSig(tys.FunctionType.endo([tys.Qubit] * 2))
-_MEAS_SIG = ext.OpDefSig(tys.FunctionType([tys.Qubit], [tys.Qubit, tys.Bool]))
-_RZ_SIG = ext.OpDefSig(tys.FunctionType([tys.Qubit, FLOAT_T], [tys.Qubit]))
-
 EXTENSION.add_op_def(
     ext.OpDef(
         name="H",
         description="Hadamard gate",
-        signature=_SINGLE_QUBIT,
+        signature=ext.OpDefSig(tys.FunctionType.endo([tys.Qubit])),
     )
 )
 
 EXTENSION.add_op_def(
     ext.OpDef(
         name="CX",
         description="CNOT gate",
-        signature=_TWO_QUBIT,
+        signature=ext.OpDefSig(tys.FunctionType.endo([tys.Qubit] * 2)),
     )
 )
 
 EXTENSION.add_op_def(
     ext.OpDef(
         name="Measure",
         description="Measurement operation",
-        signature=_MEAS_SIG,
+        signature=ext.OpDefSig(tys.FunctionType([tys.Qubit], [tys.Qubit, tys.Bool])),
     )
 )
 
 EXTENSION.add_op_def(
     ext.OpDef(
         name="Rz",
         description="Rotation around the z-axis",
-        signature=_RZ_SIG,
+        signature=ext.OpDefSig(tys.FunctionType([tys.Qubit, FLOAT_T], [tys.Qubit])),
     )
 )