magic-state-cultivation/src/latte/lattice_script_test.py at main · Strilanc/magic-state-cultivation · GitHub

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
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
from latte.lattice_script import LatticeScript


def test_lattice_script_t_comparison():
    script = LatticeScript.from_str("""
          /   /
         T   T
    =====================================
        measures m0
              /
         Z-*-Z
        /   /
    =====================================
        replace A with Z else Y if m0
        measures m1
        discard_shot_if m1

        .    A
            /
    """)
    hits = 0
    for _ in range(10):
        result, state = script.simulate()
        hits += result == 'correct'
        assert state.keys() == {
            'm0',
            'm1',
        }
    assert hits == 10


def test_lattice_script_t_comparison_bad():
    script = LatticeScript.from_str("""
          /   /
         T   T
    =====================================
        measures cmp
              /
         Z-*-Z
        /   /
    =====================================
        replace A with Y else Y if cmp
        measures parity
        discard_shot_if parity

        .    A
            /
    """)

    for _ in range(100):
        result, state = script.simulate()
        if result != 'correct':
            break
    else:
        assert False, "All correct"


def test_lattice_script_actual_factory():
    script = LatticeScript.from_str("""
          /   /   /
         T   T   T   .

         .   .   .   .
          /   /   /
         T   T   T   .
    =====================================
        measures mz2456

          /   /   /   /
         Z   Z   Z   T
        /   /   /|
         X-*-X---X   .
         |/  |/  |/   /
         Z   Z   Z   T
        /   /   /
    =====================================
        measures mz1257

          /   /   /   /
         Z   Z   Z   Z
        /   /|  /|  /
         .   X---X-*-X
          /  |/   /  |/
         Z   Z   Z   Z
        /   /   /   /
    =====================================
        measures mz0145
        measures mz2367

          /   /   /   /
         Z   Z   Z   Z
        /|  /|  /|  /|
         X-*-X   X-*-X
         |/  |/  |/  |/
         Z   Z   Z   Z
        /   /   /   /
    =====================================
        let s0 = mz2456 ^ mz2367 ^ mz0145
        let s2 = mz1257
        let s4 = mz2456 ^ mz2367
        let s6 = mz1257 ^ mz2367
        replace A with Y else Z if s0
        replace B with Y else Z if s2
        replace C with Y else Z if s4
        replace D with Y else Z if s6
        measures mxy0
        measures mxy2
        measures mxy4
        measures mxy6

              /       /
         A   Z   B   Z
        /   /   /   /
         .   .   .   .
              /       /
         C   Z   D   Z
        /   /   /   /
    =====================================
        measures mx1357
        let pass1 = mx1357 ^ mxy0 ^ mxy2 ^ mxy4 ^ mxy6
        discard_shot_if pass1

                      /
         X---X-@-Z---X
         |/ /    |  /
         X   .   Z   .
                 |    /
         .   X---Z---X
            /       /
    =====================================
        let f0 = mxy0 ^ mxy2 ^ mxy6
        let f1 = mxy0 ^ mxy4 ^ mxy6
        let f2 = mxy0 ^ mxy2 ^ mxy4
        feedback Z at 3 if f0
        feedback Z at 3+1j if f1
        feedback Z at 3+2j if f2
        feedback X at 3
        feedback X at 3+1j
        feedback X at 3+2j
        output_should_be CCZ

                     /
        .   .   .   X
                   / /
        X---X---X---X
       /             /
        .   .   .   X
                   /
    """)
    result, state = script.simulate()
    assert state.keys() == {
        'mz2456',
        'mz1257',
        'mz0145',
        'mz2367',
        'mxy0',
        'mxy4',
        'mxy2',
        'mxy6',
        'mx1357',
        'pass1',
        's0',
        's2',
        's4',
        's6',
        'f0',
        'f1',
        'f2',
    }
    assert result == 'correct'