/
test_gatesetconstruction.py
70 lines (53 loc) · 2.95 KB
/
test_gatesetconstruction.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
import unittest
import numpy as np
import pygsti
from pygsti.models import modelconstruction
from pygsti.baseobjs import Basis
from ..testutils import BaseTestCase
class TestGateSetConstructionMethods(BaseTestCase):
def setUp(self):
super(TestGateSetConstructionMethods, self).setUp()
#OK for these tests, since we test user interface?
#Set Model objects to "strict" mode for testing
pygsti.models.ExplicitOpModel._strict = False
def test_build_gatesets(self):
SQ2 = 1/np.sqrt(2)
for defParamType in ("full", "full TP", "static"):
gateset_simple = pygsti.models.ExplicitOpModel(['Q0'], 'pp', defParamType)
gateset_simple['rho0'] = [SQ2, 0, 0, SQ2]
gateset_simple['Mdefault'] = pygsti.modelmembers.povms.UnconstrainedPOVM([('0', [SQ2, 0, 0, -SQ2])], evotype='default')
gateset_simple['Gi'] = [ [1, 0, 0, 0],
[0, 1, 0, 0],
[0, 0, 1, 0],
[0, 0, 0, 1] ]
with self.assertRaises(TypeError):
gateset_simple['rho0'] = 3.0
with self.assertRaises(ValueError):
gateset_simple['rho0'] = [3.0]
with self.assertRaises(ValueError):
gateset_simple['Gx'] = [1,2,3,4]
with self.assertRaises(ValueError):
gateset_simple['Gx'] = [[1,2,3,4],[5,6,7]]
gateset_badDefParam = pygsti.models.ExplicitOpModel(['Q0'], "pp", "full")
gateset_badDefParam.preps.default_param = "foobar"
gateset_badDefParam.operations.default_param = "foobar"
with self.assertRaises(ValueError):
gateset_badDefParam['rho0'] = [1, 0, 0, 0]
with self.assertRaises(ValueError):
gateset_badDefParam['Gi'] = np.identity(4,'d')
stateSpace = [(4,)] # density matrix is a 2x2 matrix
spaceLabels = [('Q0',)] # interpret the 2x2 density matrix as a single qubit named 'Q0'
with self.assertRaises(AssertionError):
modelconstruction.create_identity_vec(Basis.cast("foobar", stateSpace))
gateset_povm_first = pygsti.models.ExplicitOpModel(['Q0']) #set effect vector first
gateset_povm_first['Mdefault'] = pygsti.modelmembers.povms.TPPOVM(
[('0', modelconstruction.create_spam_vector("0", stateSpace, "gm")),
('1', modelconstruction.create_spam_vector("1", stateSpace, "gm"))], evotype='default' )
with self.assertRaises(ValueError):
gateset_povm_first['rhoBad'] = np.array([1,2,3],'d') #wrong dimension
with self.assertRaises(ValueError):
gateset_povm_first['Mdefault'] = pygsti.modelmembers.povms.UnconstrainedPOVM(
[('0', np.array([1, 2, 3], 'd'))], evotype='default',
state_space=pygsti.baseobjs.StateSpace.cast([('L0',),('L1',),('L2',)])) #wrong dimension
if __name__ == "__main__":
unittest.main(verbosity=2)