-
Notifications
You must be signed in to change notification settings - Fork 2
/
test_units.py
136 lines (128 loc) · 5.47 KB
/
test_units.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
import circtester
import circgenerator
from qiskit import QuantumCircuit
import utilities
import checksfinder
from qiskit.quantum_info import state_fidelity
from qiskit.quantum_info import Statevector, partial_trace
def test_checksfinder_get_error_mitig_circ():
qasmstr='''OPENQASM 2.0;
include "qelib1.inc";
qreg q0[2];
y q0[0];
x q0[1];
h q0[1];
cx q0[0],q0[1];
h q0[0];
s q0[1];
cx q0[1], q0[0];
sdg q0[0];
h q0[0];
h q0[1];
s q0[1];
z q0[1];'''
num_layers=2
circ1=QuantumCircuit.from_qasm_str(qasmstr)
qasmstr2=checksfinder.get_error_mitig_circ(qasmstr, num_layers)
circ2=QuantumCircuit.from_qasm_str(qasmstr2)
circ2.remove_final_measurements()
zero1=Statevector(circ1)
zero2=Statevector(circ2)
zero1.evolve(circ1)
zero2.evolve(circ2)
zero2= partial_trace(zero2, list(range(circ2.num_qubits))[2:])
assert state_fidelity(zero1, zero2) >= 0.999
def test_circrunner_getsso():
dist1={"1111": .9, "0000": 0.1}
dist2={"1111": .9, "0000": 0.1}
assert circtester.get_sso(dist1, dist2) == 1.0
def test_count_gate():
gate_count=5
circ = QuantumCircuit(gate_count)
for idx in range(gate_count):
circ.x(idx)
circ.y(idx)
circ.z(idx)
circ.h(idx)
circ.cx(0, 1)
circ.swap(0, 1)
assert circgenerator.count_gate(circ, "x") == gate_count
assert circgenerator.count_gate(circ, "y") == gate_count
assert circgenerator.count_gate(circ, "z") == gate_count
assert circgenerator.count_gate(circ, "h") == gate_count
assert circgenerator.count_gate(circ, "cx") == gate_count
assert circgenerator.count_gate(circ, "swap") == gate_count
assert circgenerator.count_gate(circ, "not_a_gate") == 0
def test_get_single_weight_pauli_table_at_qubit_idx():
qubit_idx=0
number_of_qubits=5
single_weights=checksfinder.get_single_weight_pauli_table_at_qubit_idx(qubit_idx, number_of_qubits)
assert single_weights[0]=="XIIII"
assert single_weights[1]=="ZIIII"
qubit_idx=4
number_of_qubits=5
single_weights=checksfinder.get_single_weight_pauli_table_at_qubit_idx(qubit_idx, number_of_qubits)
assert single_weights[0]=="IIIIX"
assert single_weights[1]=="IIIIZ"
def test_get_min_checks_linear():
circ=QuantumCircuit(2)
# Test pushing through X.
circ.x(0)
p2s=["IX", "IY", "IZ"]
layers=len(p2s)
finder= checksfinder.ChecksFinder(2, circ)
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["+1IX", "-1IY", "-1IZ"] and pauli_str_p2s==["+1IX", "+1IY", "+1IZ"]
# Test pushing through Y.
circ=QuantumCircuit(2)
circ.y(0)
finder= checksfinder.ChecksFinder(2, circ)
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["-1IX", "+1IY", "-1IZ"] and pauli_str_p2s==["+1IX", "+1IY", "+1IZ"]
# Test pushing through Z.
circ=QuantumCircuit(2)
circ.z(0)
finder= checksfinder.ChecksFinder(2, circ)
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["-1IX", "-1IY", "+1IZ"] and pauli_str_p2s==["+1IX", "+1IY", "+1IZ"]
# Test pushing through H.
circ=QuantumCircuit(2)
circ.h(0)
finder= checksfinder.ChecksFinder(2, circ)
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["+1IZ", "-1IY", "+1IX"] and pauli_str_p2s==["+1IX", "+1IY", "+1IZ"]
# Test pushing through S.
circ=QuantumCircuit(2)
circ.s(0)
finder= checksfinder.ChecksFinder(2, circ)
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["-1IY", "+1IX", "+1IZ"] and pauli_str_p2s==["+1IX", "+1IY", "+1IZ"]
# Test pushing through Sdg.
circ=QuantumCircuit(2)
circ.sdg(0)
finder= checksfinder.ChecksFinder(2, circ)
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["+1IY", "-1IX", "+1IZ"] and pauli_str_p2s==["+1IX", "+1IY", "+1IZ"]
# Test pushing through CX. #Note that we are using little endian because Qiskit uses it.
circ=QuantumCircuit(2)
circ.cx(0,1)
finder= checksfinder.ChecksFinder(2, circ)
p2s=["IX", "IY", "IZ", "XI"]
layers=len(p2s)
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["+1XX", "+1XY", "+1IZ", "+1XI"] and pauli_str_p2s==["+1IX", "+1IY", "+1IZ", "+1XI"]
p2s=["XX", "XY", "XZ"]
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["+1IX", "+1IY", "+1XZ"] and pauli_str_p2s==["+1XX", "+1XY", "+1XZ"]
p2s=["YI", "YX", "YY", "YZ"]
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["+1YZ", "+1ZY", "-1ZX", "+1YI"] and pauli_str_p2s==["+1YI", "+1YX", "+1YY", "+1YZ"]
p2s=["ZI", "ZX", "ZY", "ZZ"]
_, _, pauli_str_p1s, pauli_str_p2s=finder.get_min_checks_linear(p2s, layers)
assert pauli_str_p1s==["+1ZZ", "-1YY", "+1YX", "+1ZI"] and pauli_str_p2s==["+1ZI", "+1ZX", "+1ZY", "+1ZZ"]
# def test_generate_checks_file_name_from_raw_qasm():
# file_name="some_name_raw_.qasm"
# assert checksfinder_new.generate_checks_file_name_from_raw_qasm(file_name) == "some_name_.qasm"
# def test_generate_obj_file_name_from_checks_file_name():
# file_name="some_name_.qasm"
# assert checksfinder_new.generate_obj_file_name_from_checks_file_name(file_name) == "some_name_.obj"