more tests

Author: kaihsin

test/qasm2/emit/test_extended_noise.py

@@ -63,3 +63,76 @@ def main():
 """
 
     assert out == expected
+
+
+def test_loss():
+
+    @qasm2.extended
+    def main():
+        qreg = qasm2.qreg(4)
+
+        qasm2.cx(qreg[0], qreg[1])
+        qasm2.u(qreg[2], theta=0.1, phi=0.2, lam=0.3)
+
+        noise.native.atom_loss_channel(qargs=[qreg[0], qreg[1]], prob=0.2)
+
+        qasm2.u(qreg[2], theta=0.1, phi=0.2, lam=0.3)
+
+    main.print()
+
+    target = qasm2.emit.QASM2(allow_noise=True)
+    out = target.emit_str(main)
+
+    expected = """OPENQASM 2.0;
+include "qelib1.inc";
+qreg qreg[4];
+CX qreg[0], qreg[1];
+U(0.1, 0.2, 0.3) qreg[2];
+// native.Atomloss(p=0.2)
+//  -: qargs: qreg[0], qreg[1]
+U(0.1, 0.2, 0.3) qreg[2];
+"""
+
+    assert out == expected
+
+
+def test_cz_noise():
+
+    @qasm2.extended
+    def main():
+        qreg = qasm2.qreg(4)
+
+        qasm2.cx(qreg[0], qreg[1])
+        qasm2.u(qreg[2], theta=0.1, phi=0.2, lam=0.3)
+
+        noise.native.cz_pauli_channel(
+            ctrls=[qreg[0], qreg[1]],
+            qargs=[qreg[2], qreg[3]],
+            px_ctrl=0.1,
+            py_ctrl=0.2,
+            pz_ctrl=0.3,
+            px_qarg=0.4,
+            py_qarg=0.5,
+            pz_qarg=0.6,
+            paired=True,
+        )
+
+        qasm2.u(qreg[2], theta=0.1, phi=0.2, lam=0.3)
+
+    main.print()
+
+    target = qasm2.emit.QASM2(allow_noise=True)
+    out = target.emit_str(main)
+    print(out)
+    expected = """OPENQASM 2.0;
+include "qelib1.inc";
+qreg qreg[4];
+CX qreg[0], qreg[1];
+U(0.1, 0.2, 0.3) qreg[2];
+// native.CZPauliChannel(paired=True, p_ctrl=[x:0.1, y:0.2, z:0.3], p_qarg[x:0.6, y:0.5, z:0.6])
+//  -: ctrls: qreg[0], qreg[1]
+//  -: qargs: qreg[2], qreg[3]
+U(0.1, 0.2, 0.3) qreg[2];
+"""
+
+    assert out == expected