cirq/sim/act_on_args_container_test.py

43     def _on_copy(self, args):  argument
74         args = EmptyActOnArgs(qubits, log)
76             args_map[q] = args
77         args_map[None] = args if not split_untangled_states else EmptyActOnArgs((), log)
82     args = create_container(qs2)
83     assert len(set(args.values())) == 3
84     args.apply_operation(cirq.CNOT(q0, q1))
85     assert len(set(args.values())) == 2
86     assert args[q0] is args[q1]
87     assert args[None] is not args[q0]
91     args = create_container(qs2)
92     assert len(set(args.values())) == 3
93     args.apply_operation(cirq.CircuitOperation(cirq.FrozenCircuit(cirq.CNOT(q0, q1))))
94     assert len(set(args.values())) == 2
95     assert args[q0] is args[q1]
99     args = create_container(qs3)
100     assert len(set(args.values())) == 4
101     args.apply_operation(cirq.CircuitOperation(cirq.FrozenCircuit(cirq.CNOT(q0, q1))))
102     assert len(set(args.values())) == 3
103     assert args[q0] is args[q1]
104     args.apply_operation(cirq.CircuitOperation(cirq.FrozenCircuit(cirq.CNOT(q0, q2))))
105     assert len(set(args.values())) == 2
106     assert args[q0] is args[q1] is args[q2]
110     args = create_container(qs2)
111     assert len(set(args.values())) == 3
112     args.apply_operation(cirq.IdentityGate(2)(q0, q1))
113     assert len(set(args.values())) == 3
114     assert args[q0] is not args[q1]
115     assert args[q0] is not args[None]
119     args = create_container(qs2)
120     assert len(set(args.values())) == 3
121     args._act_on_fallback_(cirq.I, (q0, q1))
122     assert len(set(args.values())) == 3
123     assert args[q0] is not args[q1]
124     assert args[q0] is not args[None]
128     args = create_container(qs2)
129     assert len(set(args.values())) == 3
130     args.apply_operation(cirq.CircuitOperation(cirq.FrozenCircuit(cirq.IdentityGate(2)(q0, q1))))
131     assert len(set(args.values())) == 3
132     assert args[q0] is not args[q1]
136     args = create_container(qs2)
137     args.apply_operation(cirq.CNOT(q0, q1))
138     assert len(set(args.values())) == 2
139     args.apply_operation(cirq.measure(q0))
140     assert len(set(args.values())) == 3
141     assert args[q0] is not args[q1]
142     assert args[q0] is not args[None]
146     args = create_container(qs2)
147     args.apply_operation(cirq.CNOT(q0, q1))
148     assert len(set(args.values())) == 2
149     args.apply_operation(cirq.CircuitOperation(cirq.FrozenCircuit(cirq.measure(q0))))
150     assert len(set(args.values())) == 3
151     assert args[q0] is not args[q1]
155     args = create_container(qs3)
156     args.apply_operation(cirq.CNOT(q0, q1))
157     args.apply_operation(cirq.CNOT(q0, q2))
158     assert len(set(args.values())) == 2
159     args.apply_operation(cirq.CircuitOperation(cirq.FrozenCircuit(cirq.measure(q0))))
160     assert len(set(args.values())) == 3
161     assert args[q0] is not args[q1]
162     args.apply_operation(cirq.CircuitOperation(cirq.FrozenCircuit(cirq.measure(q1))))
163     assert len(set(args.values())) == 4
164     assert args[q0] is not args[q1]
165     assert args[q0] is not args[q2]
166     assert args[q1] is not args[q2]
170     args = create_container(qs2)
171     args.apply_operation(cirq.CNOT(q0, q1))
172     assert len(set(args.values())) == 2
173     args.apply_operation(cirq.reset(q0))
174     assert len(set(args.values())) == 3
175     assert args[q0] is not args[q1]
176     assert args[q0] is not args[None]
180     args = create_container(qs2, False)
181     args.apply_operation(cirq.CNOT(q0, q1))
182     assert len(set(args.values())) == 1
183     args.apply_operation(cirq.measure(q0))
184     assert len(set(args.values())) == 1
185     assert args[q1] is args[q0]
186     assert args[None] is args[q0]
190     args = create_container(qs2, False)
191     args.apply_operation(cirq.CNOT(q0, q1))
192     assert len(set(args.values())) == 1
193     args.apply_operation(cirq.reset(q0))
194     assert len(set(args.values())) == 1
195     assert args[q1] is args[q0]
196     assert args[None] is args[q0]
200     args = create_container(qs2)
201     args.apply_operation(cirq.CNOT(q0, q1))
202     assert len(set(args.values())) == 2
203     args.apply_operation(cirq.measure(q0, q1))
204     assert len(set(args.values())) == 3
205     assert args[q0] is not args[q1]
206     assert args[q0] is not args[None]
210     args = create_container([q0])
211     assert len(set(args.values())) == 2
212     args.apply_operation(cirq.measure(q0))
213     assert len(set(args.values())) == 2
214     assert args[q0] is not args[None]
218     args = create_container(qs2, False)
219     reordered = args[q0].transpose_to_qubit_order([q1, q0])
224     args = create_container(qs2, False)
225     copied = args[q0].copy()
230     args = create_container(qs2, False)
231     merged = args.create_merged_state()
236     args = create_container(qs2)
237     old_q0 = args[q0]
238     old_q1 = args[q1]
239     args.apply_operation(cirq.SWAP(q0, q1))
240     assert len(set(args.values())) == 3
241     assert args[q0] is not old_q0
242     assert args[q1] is old_q0
243     assert args[q1] is not old_q1
244     assert args[q0] is old_q1
245     assert args[q0].qubits == (q0,)
246     assert args[q1].qubits == (q1,)
250     args = create_container(qs2)
251     args.apply_operation(cirq.SWAP(q0, q1) ** 0.5)
252     assert len(set(args.values())) == 2
253     assert args[q0] is args[q1]
257     args = create_container(qs2)
258     args.apply_operation(cirq.CX(q0, q1))
259     assert len(set(args.values())) == 2
260     assert args[q0].qubits == (q0, q1)
261     args.apply_operation(cirq.SWAP(q0, q1))
262     assert len(set(args.values())) == 2
263     assert args[q0] is args[q1]
264     assert args[q0].qubits == (q1, q0)
268     args = create_container(qs2)
269     assert len(set(args.values())) == 3
270     cirq.act_on(cirq.X, args, [q0])
271     assert len(set(args.values())) == 3
272     assert args[q0] is not args[q1]
273     assert args[q0] is not args[None]