fix pre-commit test

BQSKit · May 21, 2024 · 44d49ea · 44d49ea
1 parent 7ce1a1f
commit 44d49ea
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 8 deletions.
diff --git a/bqskit/ir/gates/constant/ecr.py b/bqskit/ir/gates/constant/ecr.py
@@ -7,6 +7,7 @@
 from bqskit.ir.gates.qubitgate import QubitGate
 from bqskit.qis.unitary.unitarymatrix import UnitaryMatrix
 
+
 class ECRGate(ConstantGate, QubitGate):
     """
     The echoed cross-resonance gate (ECR).
@@ -27,16 +28,23 @@ class ECRGate(ConstantGate, QubitGate):
     _num_qudits = 2
     _qasm_name = 'ecr'
     _utry = UnitaryMatrix([
-            [0, 0, 1*1/math.sqrt(2), 1j*1/math.sqrt(2)],
-            [0, 0, 1j*1/math.sqrt(2), 1*1/math.sqrt(2)],
-            [1*1/math.sqrt(2), -1j*1/math.sqrt(2), 0, 0],
-            [-1j*1/math.sqrt(2), 1*1/math.sqrt(2), 0, 0],
+        [0, 0, 1 * 1 / math.sqrt(2), 1j * 1 / math.sqrt(2)],
+        [0, 0, 1j * 1 / math.sqrt(2), 1 * 1 / math.sqrt(2)],
+        [1 * 1 / math.sqrt(2), -1j * 1 / math.sqrt(2), 0, 0],
+        [-1j * 1 / math.sqrt(2), 1 * 1 / math.sqrt(2), 0, 0],
     ])
 
     def __init__(self) -> None:
         pass
 
     def get_qasm_gate_def(self) -> str:
-        qasm_rzx = 'gate rzx(param0) q0,q1 { h q1; cx q0,q1; rz(param0) q1; cx q0,q1; h q1; }'
+        qasm_rzx = (
+            'gate rzx(param0) q0,q1 {'
+            'h q1;'
+            'cx q0,q1;'
+            'rz(param0) q1;'
+            'cx q0,q1;'
+            'h q1; }'
+        )
         qasm_ecr = 'gate ecr q0,q1 { rzx(pi/4) q0,q1; x q0; rzx(-pi/4) q0,q1; }'
-        return qasm_rzx +'\n' + qasm_ecr
+        return qasm_rzx + '\n' + qasm_ecr
diff --git a/tests/ir/gates/constant/test_ecr.py b/tests/ir/gates/constant/test_ecr.py
@@ -10,24 +10,26 @@
 from bqskit.ir.gates import XGate
 from bqskit.ir.lang.qasm2 import OPENQASM2Language
 
+
 def test_ecr() -> None:
     c = Circuit(2)
     c.append_gate(HGate(), 1)
     c.append_gate(CXGate(), (0, 1))
-    c.append_gate(RZGate(), 1, [np.pi/4.0])
+    c.append_gate(RZGate(), 1, [np.pi / 4.0])
     c.append_gate(CXGate(), (0, 1))
     c.append_gate(HGate(), 1)
 
     c.append_gate(XGate(), 0)
 
     c.append_gate(HGate(), 1)
     c.append_gate(CXGate(), (0, 1))
-    c.append_gate(RZGate(), 1, [-np.pi/4.0])
+    c.append_gate(RZGate(), 1, [-np.pi / 4.0])
     c.append_gate(CXGate(), (0, 1))
     c.append_gate(HGate(), 1)
 
     assert c.get_unitary().get_distance_from(ECRGate().get_unitary()) < 3e-8
 
+
 def test_ecr_encode_decode() -> None:
     c = Circuit(2)
     c.append_gate(ECRGate(), (0, 1))