Skip to content

Commit c91f74d

Browse files
committed
Fix error in statistics and adjust tests
1 parent 1c38094 commit c91f74d

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

projectq/cengines/_gate_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -862,8 +862,8 @@ def _add_to_execute_list(node):
862862
self._stats['simul_exec'][-1] += 1
863863
key = frozenset((mapping[node.logical_id0],
864864
mapping[node.logical_id1]))
865-
self._stats['2qubit_gates_loc'][key] = self._stats.get(
866-
node.logical_ids, 0) + 1
865+
self._stats['2qubit_gates_loc'][key] \
866+
= self._stats['2qubit_gates_loc'].get(key, 0) + 1
867867
for cmd in node.compatible_successor_cmds:
868868
if len([
869869
qubit.id for qureg in cmd.all_qubits

projectq/cengines/_gate_manager_test.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -972,6 +972,9 @@ def test_qubit_manager_str():
972972
Command(engine=None, gate=X, qubits=([qb[1]], )),
973973
Command(engine=None, gate=X, qubits=([qb[4]], ), controls=[qb[5]]),
974974
Command(engine=None, gate=X, qubits=([qb[5]], ), controls=[qb[4]]),
975+
Command(engine=None, gate=X, qubits=([qb[3]], ), controls=[qb[4]]),
976+
Command(engine=None, gate=X, qubits=([qb[5]], ), controls=[qb[4]]),
977+
Command(engine=None, gate=X, qubits=([qb[3]], ), controls=[qb[4]]),
975978
Command(engine=None, gate=X, qubits=([qb[5]], ), controls=[qb[6]]),
976979
Command(engine=None, gate=X, qubits=([qb[6]], ), controls=[qb[7]]),
977980
Command(engine=None, gate=X, qubits=([qb[7]], ), controls=[qb[8]]),
@@ -1008,5 +1011,8 @@ def test_qubit_manager_str():
10081011
if m:
10091012
num_of_2qubit_gates += int(m.group(1))
10101013

1014+
edge34_count = int(re.search(r'\s+\[3, 4\]:\s+([0-9]+)',
1015+
str_repr).group(1))
10111016
assert num_of_2qubit_gates == num_of_2qubit_gates_ref
1012-
assert str_repr.count("[4, 5]: 2") == 1
1017+
assert edge34_count > 1
1018+
assert str_repr.count("[4, 5]: 3") == 1

0 commit comments

Comments
 (0)