-
Notifications
You must be signed in to change notification settings - Fork 55
/
test_explicitmodel.py
155 lines (134 loc) · 7.7 KB
/
test_explicitmodel.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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
import numpy as np
import pygsti.models.explicitmodel as mdl
from pygsti.baseobjs import ExplicitStateSpace
from pygsti.models.modelconstruction import create_explicit_model_from_expressions, create_operation
from pygsti.modelpacks.legacy import std1Q_XYI as std
from ..util import BaseCase
class ExplicitOpModelStrictAccessTester(BaseCase):
def setUp(self):
mdl.ExplicitOpModel._strict = True
self.model = std.target_model().randomize_with_unitary(0.001, seed=1234)
def test_strict_access(self):
#test strict mode, which forbids all these accesses
with self.assertRaises(KeyError):
self.model['identity'] = [1, 0, 0, 0]
with self.assertRaises(KeyError):
self.model['Gx'] = np.identity(4, 'd')
with self.assertRaises(KeyError):
self.model['E0'] = [1, 0, 0, 0]
with self.assertRaises(KeyError):
self.model['rho0'] = [1, 0, 0, 0]
with self.assertRaises(KeyError):
self.model['identity']
with self.assertRaises(KeyError):
self.model['Gx']
with self.assertRaises(KeyError):
self.model['E0']
with self.assertRaises(KeyError):
self.model['rho0']
class ExplicitOpModelToolTester(BaseCase):
def setUp(self):
mdl.ExplicitOpModel._strict = False
# XXX can these be constructed directly? EGN: yes, some model-construction tests should do it.
self.model = create_explicit_model_from_expressions([('Q0',)], ['Gi', 'Gx', 'Gy'],
["I(Q0)", "X(pi/2,Q0)", "Y(pi/2,Q0)"])
self.gateset_2q = create_explicit_model_from_expressions(
[('Q0', 'Q1')], ['GIX', 'GIY', 'GXI', 'GYI', 'GCNOT'],
["I(Q0):X(pi/2,Q1)", "I(Q0):Y(pi/2,Q1)", "X(pi/2,Q0):I(Q1)", "Y(pi/2,Q0):I(Q1)", "CX(pi,Q0,Q1)"])
def test_randomize_with_unitary(self):
gateset_randu = self.model.randomize_with_unitary(0.01)
gateset_randu = self.model.randomize_with_unitary(0.01, seed=1234)
# TODO assert correctness
def test_rotate_1q(self):
sslbls = ExplicitStateSpace("Q0")
rotXPi = create_operation("X(pi,Q0)", sslbls, "pp")
rotXPiOv2 = create_operation("X(pi/2,Q0)", sslbls, "pp")
rotYPiOv2 = create_operation("Y(pi/2,Q0)", sslbls, "pp")
gateset_rot = self.model.rotate((np.pi / 2, 0, 0)) # rotate all gates by pi/2 about X axis
self.assertArraysAlmostEqual(gateset_rot['Gi'], rotXPiOv2)
self.assertArraysAlmostEqual(gateset_rot['Gx'], rotXPi)
self.assertArraysAlmostEqual(gateset_rot['Gx'], np.dot(rotXPiOv2, rotXPiOv2))
self.assertArraysAlmostEqual(gateset_rot['Gy'], np.dot(rotXPiOv2, rotYPiOv2))
def test_rotate_2q(self):
gateset_2q_rot = self.gateset_2q.rotate(rotate=list(np.zeros(15, 'd')))
gateset_2q_rot_same = self.gateset_2q.rotate(rotate=(0.01,) * 15)
gateset_2q_randu = self.gateset_2q.randomize_with_unitary(0.01)
gateset_2q_randu = self.gateset_2q.randomize_with_unitary(0.01, seed=1234)
# TODO assert correctness
def test_depolarize(self):
Gi_dep = np.array([[1, 0, 0, 0],
[0, 0.9, 0, 0],
[0, 0, 0.9, 0],
[0, 0, 0, 0.9]], 'd')
Gx_dep = np.array([[1, 0, 0, 0],
[0, 0.9, 0, 0],
[0, 0, 0, -0.9],
[0, 0, 0.9, 0]], 'd')
Gy_dep = np.array([[1, 0, 0, 0],
[0, 0, 0, 0.9],
[0, 0, 0.9, 0],
[0, -0.9, 0, 0]], 'd')
gateset_dep = self.model.depolarize(op_noise=0.1)
self.assertArraysAlmostEqual(gateset_dep['Gi'], Gi_dep)
self.assertArraysAlmostEqual(gateset_dep['Gx'], Gx_dep)
self.assertArraysAlmostEqual(gateset_dep['Gy'], Gy_dep)
def test_depolarize_with_spam_noise(self):
gateset_spam = self.model.depolarize(spam_noise=0.1)
self.assertAlmostEqual(float(np.dot(self.model['Mdefault']['0'].to_dense().T, self.model['rho0'].to_dense())), 1.0)
# Since np.ndarray doesn't implement __round__... (assertAlmostEqual() doesn't work)
# Compare the single element dot product result to 0.095 instead (coverting the array's contents ([[ 0.095 ]]) to a **python** float (0.095))
# print("DEBUG gateset_spam = ")
# print(gateset_spam['Mdefault']['0'].T)
# print(gateset_spam['rho0'].T)
# print(gateset_spam)
# print(gateset_spam['Mdefault']['0'].T)
# print(gateset_spam['rho0'].T)
# not 0.905 b/c effecs aren't depolarized now
self.assertAlmostEqual(np.dot(gateset_spam['Mdefault']['0'].to_dense().T, gateset_spam['rho0'].to_dense()).reshape(-1,)[0], 0.95)
self.assertArraysAlmostEqual(gateset_spam['rho0'].to_dense(), 1 / np.sqrt(2) * np.array([1, 0, 0, 0.9]))
#self.assertArraysAlmostEqual(gateset_spam['Mdefault']['0'], 1/np.sqrt(2)*np.array([1,0,0,0.9]).reshape(-1,1) ) #not depolarized now
print(gateset_spam['Mdefault']['0'].to_dense())
self.assertArraysAlmostEqual(gateset_spam['Mdefault']['0'].to_dense(), 1 / np.sqrt(2) * np.array([1, 0, 0, 1])) # not depolarized now
def test_random_rotate_1q(self):
gateset_rand_rot = self.model.rotate(max_rotate=0.2)
gateset_rand_rot = self.model.rotate(max_rotate=0.2, seed=1234)
def test_random_rotate_2q(self):
gateset_2q_rand_rot = self.gateset_2q.rotate(max_rotate=0.2)
gateset_2q_rand_rot = self.gateset_2q.rotate(max_rotate=0.2, seed=1234)
# TODO assert correctness
def test_random_depolarize(self):
gateset_rand_dep = self.model.depolarize(max_op_noise=0.1)
gateset_rand_dep = self.model.depolarize(max_op_noise=0.1, seed=1234)
# TODO assert correctness
def test_random_depolarize_with_spam_noise(self):
gateset_rand_spam = self.model.depolarize(max_spam_noise=0.1)
gateset_rand_spam = self.model.depolarize(max_spam_noise=0.1, seed=1234)
# TODO assert correctness
def test_rotate_raises_on_bad_arg_spec(self):
with self.assertRaises(ValueError):
self.model.rotate(rotate=(0.2,) * 3, max_rotate=0.2) # can't specify both
with self.assertRaises(ValueError):
self.model.rotate() # must specify rotate or max_rotate
with self.assertRaises(ValueError):
self.gateset_2q.rotate(rotate=(0.2,) * 15, max_rotate=0.2) # can't specify both
with self.assertRaises(ValueError):
self.gateset_2q.rotate() # must specify rotate or max_rotate
def test_rotate_raises_on_bad_dim(self):
with self.assertRaises(AssertionError):
self.model.rotate((1, 2, 3, 4)) # tuple must be length 3
with self.assertRaises(AssertionError):
self.gateset_2q.rotate(rotate=(0, 0, 0)) # wrong dimension model
with self.assertRaises(AssertionError):
self.gateset_2q.rotate((1, 2, 3, 4)) # tuple must be length 15
with self.assertRaises(AssertionError):
self.model.rotate(rotate=np.zeros(15, 'd')) # wrong dimension model
def test_rotate_raises_on_bad_type(self):
with self.assertRaises(AssertionError):
self.model.rotate("a string!") # must be a 3-tuple
with self.assertRaises(AssertionError):
self.gateset_2q.rotate("a string!") # must be a 15-tuple
def test_depolarize_raises_on_bad_arg_spec(self):
with self.assertRaises(ValueError):
self.model.depolarize(op_noise=0.1, max_op_noise=0.1, spam_noise=0) # can't specify both
with self.assertRaises(ValueError):
self.model.depolarize(spam_noise=0.1, max_spam_noise=0.1) # can't specify both