cirq-core 1.7.0.dev20250924231107__py3-none-any.whl → 1.7.0.dev20251203004401__py3-none-any.whl

cirq/transformers/optimize_for_target_gateset_test.py

@@ -25,7 +25,7 @@ if TYPE_CHECKING:
     from cirq.protocols.decompose_protocol import DecomposeResult
 
 
-def test_decompose_operations_raises_on_stuck():
+def test_decompose_operations_raises_on_stuck() -> None:
     c_orig = cirq.Circuit(cirq.X(cirq.NamedQubit("q")).with_tags("ignore"))
     gateset = cirq.Gateset(cirq.Y)
     with pytest.raises(ValueError, match="Unable to convert"):
@@ -41,7 +41,7 @@ def test_decompose_operations_raises_on_stuck():
     cirq.testing.assert_same_circuits(c_orig, c_new)
 
 
-def test_decompose_operations_to_target_gateset_default():
+def test_decompose_operations_to_target_gateset_default() -> None:
     q = cirq.LineQubit.range(2)
     c_orig = cirq.Circuit(
         cirq.T(q[0]),
@@ -77,7 +77,7 @@ m: ═════════════════════════
     )
 
 
-def test_decompose_operations_to_target_gateset():
+def test_decompose_operations_to_target_gateset() -> None:
     q = cirq.LineQubit.range(2)
     c_orig = cirq.Circuit(
         cirq.T(q[0]),
@@ -130,7 +130,7 @@ class MatrixGateTargetGateset(cirq.CompilationTargetGateset):
         return cirq.MatrixGate(cirq.unitary(op), name="M").on(*op.qubits)
 
 
-def test_optimize_for_target_gateset_default():
+def test_optimize_for_target_gateset_default() -> None:
     q = cirq.LineQubit.range(2)
     c_orig = cirq.Circuit(
         cirq.T(q[0]), cirq.SWAP(*q), cirq.T(q[0]), cirq.SWAP(*q).with_tags("ignore")
@@ -148,7 +148,7 @@ def test_optimize_for_target_gateset_default():
     cirq.testing.assert_circuits_with_terminal_measurements_are_equivalent(c_orig, c_new, atol=1e-6)
 
 
-def test_optimize_for_target_gateset():
+def test_optimize_for_target_gateset() -> None:
     q = cirq.LineQubit.range(4)
     c_orig = cirq.Circuit(
         cirq.QuantumFourierTransformGate(4).on(*q),
@@ -201,7 +201,7 @@ m: ═════════════════════════
         )
 
 
-def test_optimize_for_target_gateset_deep():
+def test_optimize_for_target_gateset_deep() -> None:
     q0, q1 = cirq.LineQubit.range(2)
     c_nested = cirq.FrozenCircuit(cirq.CX(q0, q1))
     c_orig = cirq.Circuit(
@@ -209,10 +209,12 @@ def test_optimize_for_target_gateset_deep():
             cirq.FrozenCircuit(cirq.H(q0), cirq.CircuitOperation(c_nested).repeat(3))
         ).repeat(5)
     )
+    phxz_from_h_unitary = cirq.single_qubit_matrix_to_phxz(cirq.unitary(cirq.H(q0)))
+    assert phxz_from_h_unitary is not None
     c_expected = cirq.Circuit(
         cirq.CircuitOperation(
             cirq.FrozenCircuit(
-                cirq.single_qubit_matrix_to_phxz(cirq.unitary(cirq.H(q0))).on(q0),
+                phxz_from_h_unitary.on(q0),
                 cirq.CircuitOperation(
                     cirq.FrozenCircuit(
                         cirq.MatrixGate(c_nested.unitary(qubit_order=[q0, q1]), name="M").on(q0, q1)
@@ -252,7 +254,7 @@ def test_optimize_for_target_gateset_deep():
 
 
 @pytest.mark.parametrize('max_num_passes', [2, None])
-def test_optimize_for_target_gateset_multiple_passes(max_num_passes: int | None):
+def test_optimize_for_target_gateset_multiple_passes(max_num_passes: int | None) -> None:
     gateset = cirq.CZTargetGateset()
 
     input_circuit = cirq.Circuit(
@@ -332,7 +334,7 @@ def test_optimize_for_target_gateset_multiple_passes(max_num_passes: int | None)
 @pytest.mark.parametrize('max_num_passes', [2, None])
 def test_optimize_for_target_gateset_multiple_passes_dont_preserve_moment_structure(
     max_num_passes: int | None,
-):
+) -> None:
     gateset = cirq.CZTargetGateset(preserve_moment_structure=False)
 
     input_circuit = cirq.Circuit(

cirq/transformers/qubit_management_transformers_test.py

@@ -14,6 +14,8 @@
 
 from __future__ import annotations
 
+from collections.abc import Callable
+
 import cirq
 
 
@@ -57,8 +59,10 @@ class GateAllocAndBorrowInDecompose(cirq.Gate):
         qm.qfree(qa + qb)
 
 
-def get_decompose_func(gate_type, qm):
-    def decompose_func(op: cirq.Operation, _):
+def get_decompose_func(
+    gate_type: type[cirq.Gate], qm: cirq.QubitManager
+) -> Callable[[cirq.Operation, int], cirq.OP_TREE]:
+    def decompose_func(op: cirq.Operation, _: int) -> cirq.OP_TREE:
         return (
             cirq.decompose_once(op, context=cirq.DecompositionContext(qm))
             if isinstance(op.gate, gate_type)

cirq/transformers/routing/mapping_manager.py

@@ -16,7 +16,8 @@
 
 from __future__ import annotations
 
-from typing import Sequence, TYPE_CHECKING
+from collections.abc import Sequence
+from typing import TYPE_CHECKING
 
 import networkx as nx
 import numpy as np

cirq/transformers/routing/route_circuit_cqc.py

@@ -17,7 +17,8 @@
 from __future__ import annotations
 
 import itertools
-from typing import Any, Sequence, TYPE_CHECKING
+from collections.abc import Sequence
+from typing import Any, TYPE_CHECKING
 
 import networkx as nx
 

cirq/transformers/stratify.py

@@ -17,7 +17,8 @@
 from __future__ import annotations
 
 import itertools
-from typing import Callable, Iterable, Sequence, TYPE_CHECKING, Union
+from collections.abc import Callable, Iterable, Sequence
+from typing import TYPE_CHECKING, Union
 
 from cirq import _import, circuits, ops, protocols
 from cirq.transformers import transformer_api

cirq/transformers/symbolize.py

@@ -17,7 +17,8 @@
 from __future__ import annotations
 
 import re
-from typing import Hashable, TYPE_CHECKING
+from collections.abc import Hashable
+from typing import TYPE_CHECKING
 
 import attrs
 import sympy

cirq/transformers/tag_transformers.py

@@ -15,7 +15,8 @@
 from __future__ import annotations
 
 import itertools
-from typing import Callable, Hashable, TYPE_CHECKING
+from collections.abc import Callable, Hashable
+from typing import TYPE_CHECKING
 
 from cirq.transformers import transformer_api, transformer_primitives
 

cirq/transformers/target_gatesets/compilation_target_gateset.py

@@ -17,7 +17,8 @@
 from __future__ import annotations
 
 import abc
-from typing import Hashable, TYPE_CHECKING
+from collections.abc import Hashable
+from typing import TYPE_CHECKING
 
 from cirq import circuits, ops, protocols, transformers
 from cirq.transformers import merge_k_qubit_gates, merge_single_qubit_gates

cirq/transformers/target_gatesets/cz_gateset.py

@@ -16,7 +16,8 @@
 
 from __future__ import annotations
 
-from typing import Any, Sequence, TYPE_CHECKING
+from collections.abc import Sequence
+from typing import Any, TYPE_CHECKING
 
 from cirq import ops, protocols
 from cirq.transformers.analytical_decompositions import two_qubit_to_cz

cirq/transformers/target_gatesets/cz_gateset_test.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Sequence
+from collections.abc import Sequence
 
 import numpy as np
 import pytest

cirq/transformers/target_gatesets/sqrt_iswap_gateset.py

@@ -16,7 +16,8 @@
 
 from __future__ import annotations
 
-from typing import Any, Sequence, TYPE_CHECKING
+from collections.abc import Sequence
+from typing import Any, TYPE_CHECKING
 
 from cirq import ops, protocols
 from cirq.transformers.analytical_decompositions import two_qubit_to_sqrt_iswap

cirq/transformers/transformer_api.py

@@ -21,7 +21,8 @@ import enum
 import functools
 import inspect
 import textwrap
-from typing import Any, Callable, cast, Hashable, overload, Protocol, TYPE_CHECKING, TypeVar
+from collections.abc import Callable, Hashable
+from typing import Any, cast, overload, Protocol, TYPE_CHECKING, TypeVar
 
 from cirq import circuits