cirq-core 1.5.0.dev20250409222543__py3-none-any.whl → 1.6.0__py3-none-any.whl

cirq/interop/quirk/cells/control_cells.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Iterable, Iterator, List, Optional, TYPE_CHECKING, Union
+from __future__ import annotations
+
+from typing import Any, Iterable, Iterator, TYPE_CHECKING
 
 from cirq import ops, value
 from cirq.interop.quirk.cells.cell import Cell, CellMaker
@@ -25,7 +27,7 @@ if TYPE_CHECKING:
 class ControlCell(Cell):
     """A modifier that adds controls to other cells in the column."""
 
-    def __init__(self, qubit: 'cirq.Qid', basis_change: Iterable['cirq.Operation']):
+    def __init__(self, qubit: cirq.Qid, basis_change: Iterable[cirq.Operation]):
         self.qubit = qubit
         self._basis_change = tuple(basis_change)
 
@@ -42,7 +44,7 @@ class ControlCell(Cell):
     def gate_count(self) -> int:
         return 0
 
-    def with_line_qubits_mapped_to(self, qubits: List['cirq.Qid']) -> 'Cell':
+    def with_line_qubits_mapped_to(self, qubits: list[cirq.Qid]) -> Cell:
         return ControlCell(
             qubit=Cell._replace_qubit(self.qubit, qubits),
             basis_change=tuple(
@@ -51,13 +53,13 @@ class ControlCell(Cell):
             ),
         )
 
-    def modify_column(self, column: List[Optional['Cell']]):
+    def modify_column(self, column: list[Cell | None]):
         for i in range(len(column)):
             gate = column[i]
             if gate is not None:
                 column[i] = gate.controlled_by(self.qubit)
 
-    def basis_change(self) -> 'cirq.OP_TREE':
+    def basis_change(self) -> cirq.OP_TREE:
         return self._basis_change
 
 
@@ -69,7 +71,7 @@ class ParityControlCell(Cell):
     of them are individually satisfied.
     """
 
-    def __init__(self, qubits: Iterable['cirq.Qid'], basis_change: Iterable['cirq.Operation']):
+    def __init__(self, qubits: Iterable[cirq.Qid], basis_change: Iterable[cirq.Operation]):
         self.qubits = list(qubits)
         self._basis_change = list(basis_change)
 
@@ -86,7 +88,7 @@ class ParityControlCell(Cell):
     def gate_count(self) -> int:
         return 0
 
-    def with_line_qubits_mapped_to(self, qubits: List['cirq.Qid']) -> 'Cell':
+    def with_line_qubits_mapped_to(self, qubits: list[cirq.Qid]) -> Cell:
         return ParityControlCell(
             qubits=Cell._replace_qubits(self.qubits, qubits),
             basis_change=tuple(
@@ -95,7 +97,7 @@ class ParityControlCell(Cell):
             ),
         )
 
-    def modify_column(self, column: List[Optional['Cell']]):
+    def modify_column(self, column: list[Cell | None]):
         for i in range(len(column)):
             gate = column[i]
             if gate is self:
@@ -109,7 +111,7 @@ class ParityControlCell(Cell):
             elif gate is not None:
                 column[i] = gate.controlled_by(self.qubits[0])
 
-    def basis_change(self) -> Iterator['cirq.OP_TREE']:
+    def basis_change(self) -> Iterator[cirq.OP_TREE]:
         yield from self._basis_change
 
         # Temporarily move the ZZZ..Z parity observable onto a single qubit.
@@ -132,7 +134,7 @@ def generate_all_control_cell_makers() -> Iterator[CellMaker]:
     yield _reg_parity_control("zpar", basis_change=None)
 
 
-def _reg_control(identifier: str, *, basis_change: Optional['cirq.Gate']) -> CellMaker:
+def _reg_control(identifier: str, *, basis_change: cirq.Gate | None) -> CellMaker:
     return CellMaker(
         identifier=identifier,
         size=1,
@@ -142,9 +144,7 @@ def _reg_control(identifier: str, *, basis_change: Optional['cirq.Gate']) -> Cel
     )
 
 
-def _reg_parity_control(
-    identifier: str, *, basis_change: Optional['cirq.Gate'] = None
-) -> CellMaker:
+def _reg_parity_control(identifier: str, *, basis_change: cirq.Gate | None = None) -> CellMaker:
     return CellMaker(
         identifier=identifier,
         size=1,
@@ -155,8 +155,8 @@ def _reg_parity_control(
 
 
 def _basis_else_empty(
-    basis_change: Optional['cirq.Gate'], qureg: Union['cirq.Qid', Iterable['cirq.Qid']]
-) -> Iterable['cirq.Operation']:
+    basis_change: cirq.Gate | None, qureg: cirq.Qid | Iterable[cirq.Qid]
+) -> Iterable[cirq.Operation]:
     if basis_change is None:
         return ()
     return basis_change.on_each(qureg)

cirq/interop/quirk/cells/control_cells_test.py

@@ -12,11 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import cirq
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_controls():
+def test_controls() -> None:
     a, b = cirq.LineQubit.range(2)
 
     assert_url_to_circuit_returns('{"cols":[["•","X"]]}', cirq.Circuit(cirq.X(b).controlled_by(a)))
@@ -86,7 +88,7 @@ def test_controls():
     )
 
 
-def test_parity_controls():
+def test_parity_controls() -> None:
     a, b, c, d, e = cirq.LineQubit.range(5)
 
     assert_url_to_circuit_returns(
@@ -106,7 +108,7 @@ def test_parity_controls():
     )
 
 
-def test_control_with_line_qubits_mapped_to():
+def test_control_with_line_qubits_mapped_to() -> None:
     a, b = cirq.LineQubit.range(2)
     a2, b2 = cirq.NamedQubit.range(2, prefix='q')
     cell = cirq.interop.quirk.cells.control_cells.ControlCell(a, [cirq.Y(b) ** 0.5])
@@ -115,7 +117,7 @@ def test_control_with_line_qubits_mapped_to():
     assert cell.with_line_qubits_mapped_to([a2, b2]) == mapped_cell
 
 
-def test_parity_control_with_line_qubits_mapped_to():
+def test_parity_control_with_line_qubits_mapped_to() -> None:
     a, b, c = cirq.LineQubit.range(3)
     a2, b2, c2 = cirq.NamedQubit.range(3, prefix='q')
     cell = cirq.interop.quirk.cells.control_cells.ParityControlCell([a, b], [cirq.Y(c) ** 0.5])
@@ -126,7 +128,7 @@ def test_parity_control_with_line_qubits_mapped_to():
     assert cell.with_line_qubits_mapped_to([a2, b2, c2]) == mapped_cell
 
 
-def test_repr():
+def test_repr() -> None:
     a, b, c = cirq.LineQubit.range(3)
     cirq.testing.assert_equivalent_repr(
         cirq.interop.quirk.cells.control_cells.ControlCell(a, [cirq.Y(b) ** 0.5])

cirq/interop/quirk/cells/frequency_space_cells.py

@@ -11,6 +11,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from __future__ import annotations
+
 from typing import Callable, Iterator
 
 import sympy
@@ -41,9 +44,7 @@ def generate_all_frequency_space_cell_makers() -> Iterator[CellMaker]:
     )
 
 
-def _family(
-    identifier_prefix: str, gate_maker: Callable[[int], 'cirq.Gate']
-) -> Iterator[CellMaker]:
+def _family(identifier_prefix: str, gate_maker: Callable[[int], cirq.Gate]) -> Iterator[CellMaker]:
     f = lambda args: ExplicitOperationsCell([gate_maker(len(args.qubits)).on(*args.qubits)])
     yield CellMaker(identifier_prefix, 1, f)
     for i in CELL_SIZES:

cirq/interop/quirk/cells/frequency_space_cells_test.py

@@ -12,13 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import sympy
 
 import cirq
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_frequency_space_gates():
+def test_frequency_space_gates() -> None:
     a, b, c = cirq.LineQubit.range(3)
 
     assert_url_to_circuit_returns('{"cols":[["QFT3"]]}', cirq.Circuit(cirq.qft(a, b, c)))

cirq/interop/quirk/cells/ignored_cells.py

@@ -11,6 +11,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from __future__ import annotations
+
 from typing import Iterator
 
 from cirq.interop.quirk.cells.cell import CELL_SIZES, CellMaker

cirq/interop/quirk/cells/ignored_cells_test.py

@@ -12,11 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import cirq
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_displays():
+def test_displays() -> None:
     assert_url_to_circuit_returns(
         '{"cols":[["Amps2"],[1,"Amps3"],["Chance"],'
         '["Chance2"],["Density"],["Density3"],'

cirq/interop/quirk/cells/input_cells.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Iterable, Iterator, List, Optional, TYPE_CHECKING
+from __future__ import annotations
+
+from typing import Iterable, Iterator, TYPE_CHECKING
 
 from cirq.interop.quirk.cells.cell import Cell, CELL_SIZES, CellMaker
 
@@ -23,17 +25,17 @@ if TYPE_CHECKING:
 class InputCell(Cell):
     """A modifier that provides a quantum input to gates in the same column."""
 
-    def __init__(self, qubits: Iterable['cirq.Qid'], letter: str):
+    def __init__(self, qubits: Iterable[cirq.Qid], letter: str):
         self.qubits = tuple(qubits)
         self.letter = letter
 
     def gate_count(self) -> int:
         return 0
 
-    def with_line_qubits_mapped_to(self, qubits: List['cirq.Qid']) -> 'Cell':
+    def with_line_qubits_mapped_to(self, qubits: list[cirq.Qid]) -> Cell:
         return InputCell(qubits=Cell._replace_qubits(self.qubits, qubits), letter=self.letter)
 
-    def modify_column(self, column: List[Optional['Cell']]):
+    def modify_column(self, column: list[Cell | None]):
         for i in range(len(column)):
             cell = column[i]
             if cell is not None:
@@ -50,7 +52,7 @@ class SetDefaultInputCell(Cell):
     def gate_count(self) -> int:
         return 0
 
-    def with_line_qubits_mapped_to(self, qubits: List['cirq.Qid']) -> 'Cell':
+    def with_line_qubits_mapped_to(self, qubits: list[cirq.Qid]) -> Cell:
         return self
 
     def persistent_modifiers(self):

cirq/interop/quirk/cells/input_cells_test.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import pytest
 
 from cirq import quirk_url_to_circuit
@@ -19,12 +21,12 @@ from cirq.interop.quirk.cells.input_cells import SetDefaultInputCell
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_missing_input_cell():
+def test_missing_input_cell() -> None:
     with pytest.raises(ValueError, match='Missing input'):
         _ = quirk_url_to_circuit('https://algassert.com/quirk#circuit={"cols":[["+=A2"]]}')
 
 
-def test_input_cell():
+def test_input_cell() -> None:
     assert_url_to_circuit_returns(
         '{"cols":[["inputA4",1,1,1,"+=A4"]]}', maps={0x_0_0: 0x_0_0, 0x_2_3: 0x_2_5}
     )
@@ -41,7 +43,7 @@ def test_input_cell():
         )
 
 
-def test_reversed_input_cell():
+def test_reversed_input_cell() -> None:
     assert_url_to_circuit_returns(
         '{"cols":[["revinputA4",1,1,1,"+=A4"]]}',
         maps={0x_0_0: 0x_0_0, 0x_2_3: 0x_2_7, 0x_1_3: 0x_1_B},
@@ -59,7 +61,7 @@ def test_reversed_input_cell():
         )
 
 
-def test_set_default_input_cell():
+def test_set_default_input_cell() -> None:
     # Later column.
     assert_url_to_circuit_returns(
         '{"cols":[[{"id":"setA","arg":11}],["+=A4"]]}', maps={0: 11, 4: 15, 5: 0}
@@ -98,6 +100,6 @@ def test_set_default_input_cell():
         )
 
 
-def test_with_line_qubits_mapped_to():
+def test_with_line_qubits_mapped_to() -> None:
     cell = SetDefaultInputCell('a', 5)
     assert cell.with_line_qubits_mapped_to([]) is cell

cirq/interop/quirk/cells/input_rotation_cells.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Iterable, Iterator, List, Optional, Sequence, Tuple, Union
+from __future__ import annotations
+
+from typing import Any, Iterable, Iterator, Sequence
 
 import numpy as np
 
@@ -28,8 +30,8 @@ class InputRotationCell(Cell):
     def __init__(
         self,
         identifier: str,
-        register: Optional[Sequence['cirq.Qid']],
-        base_operation: 'cirq.Operation',
+        register: Sequence[cirq.Qid] | None,
+        base_operation: cirq.Operation,
         exponent_sign: int,
     ):
         self.identifier = identifier
@@ -52,7 +54,7 @@ class InputRotationCell(Cell):
     def gate_count(self) -> int:
         return 1
 
-    def with_line_qubits_mapped_to(self, qubits: List['cirq.Qid']) -> 'Cell':
+    def with_line_qubits_mapped_to(self, qubits: list[cirq.Qid]) -> Cell:
         return InputRotationCell(
             self.identifier,
             None if self.register is None else Cell._replace_qubits(self.register, qubits),
@@ -62,7 +64,7 @@ class InputRotationCell(Cell):
             exponent_sign=self.exponent_sign,
         )
 
-    def with_input(self, letter: str, register: Union[Sequence['cirq.Qid'], int]) -> 'Cell':
+    def with_input(self, letter: str, register: Sequence[cirq.Qid] | int) -> Cell:
         # Parameterized rotations use input A as their parameter.
         if self.register is None and letter == 'a':
             if isinstance(register, int):
@@ -75,7 +77,7 @@ class InputRotationCell(Cell):
             )
         return self
 
-    def controlled_by(self, qubit: 'cirq.Qid'):
+    def controlled_by(self, qubit: cirq.Qid):
         return InputRotationCell(
             self.identifier,
             self.register,
@@ -83,7 +85,7 @@ class InputRotationCell(Cell):
             self.exponent_sign,
         )
 
-    def operations(self) -> 'cirq.OP_TREE':
+    def operations(self) -> cirq.OP_TREE:
         if self.register is None:
             raise ValueError("Missing input 'a'")
         return QuirkInputRotationOperation(
@@ -98,8 +100,8 @@ class QuirkInputRotationOperation(ops.Operation):
     def __init__(
         self,
         identifier: str,
-        register: Iterable['cirq.Qid'],
-        base_operation: 'cirq.Operation',
+        register: Iterable[cirq.Qid],
+        base_operation: cirq.Operation,
         exponent_sign: int,
     ):
         if exponent_sign not in [-1, +1]:
@@ -113,7 +115,7 @@ class QuirkInputRotationOperation(ops.Operation):
         return (self.identifier, self.register, self.base_operation, self.exponent_sign)
 
     @property
-    def qubits(self) -> Tuple['cirq.Qid', ...]:
+    def qubits(self) -> tuple[cirq.Qid, ...]:
         return tuple(self.base_operation.qubits) + self.register
 
     def with_qubits(self, *new_qubits):
@@ -127,7 +129,7 @@ class QuirkInputRotationOperation(ops.Operation):
             self.exponent_sign,
         )
 
-    def _circuit_diagram_info_(self, args: 'cirq.CircuitDiagramInfoArgs'):
+    def _circuit_diagram_info_(self, args: cirq.CircuitDiagramInfoArgs):
         sub_result = cirq.circuit_diagram_info(self.base_operation)
         sign_char = '-' if self.exponent_sign == -1 else ''
         symbols = list(sub_result.wire_symbols)
@@ -142,7 +144,7 @@ class QuirkInputRotationOperation(ops.Operation):
     def _has_unitary_(self) -> bool:
         return True
 
-    def _apply_unitary_(self, args: 'cirq.ApplyUnitaryArgs'):
+    def _apply_unitary_(self, args: cirq.ApplyUnitaryArgs):
         transposed_args = args.with_axes_transposed_to_start()
 
         target_axes = transposed_args.axes[: len(self.base_operation.qubits)]
@@ -184,7 +186,7 @@ def generate_all_input_rotation_cell_makers() -> Iterator[CellMaker]:
     yield _input_rotation_gate("Z^(-A/2^n)", ops.Z, -1)
 
 
-def _input_rotation_gate(identifier: str, gate: 'cirq.Gate', exponent_sign: int) -> CellMaker:
+def _input_rotation_gate(identifier: str, gate: cirq.Gate, exponent_sign: int) -> CellMaker:
     return CellMaker(
         identifier,
         gate.num_qubits(),

cirq/interop/quirk/cells/input_rotation_cells_test.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import numpy as np
 import pytest
 
@@ -20,7 +22,7 @@ from cirq import quirk_url_to_circuit
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_input_rotation_cells():
+def test_input_rotation_cells() -> None:
     with pytest.raises(ValueError, match='classical constant'):
         _ = quirk_url_to_circuit(
             'https://algassert.com/quirk#circuit={"cols":[["Z^(A/2^n)",{"id":"setA","arg":3}]]}'
@@ -105,7 +107,7 @@ def test_input_rotation_cells():
     )
 
 
-def test_input_rotation_cells_repr():
+def test_input_rotation_cells_repr() -> None:
     circuit = quirk_url_to_circuit(
         'http://algassert.com/quirk#circuit={"cols":[["•","X^(-A/2^n)","inputA2"]]}'
     )
@@ -113,7 +115,7 @@ def test_input_rotation_cells_repr():
     cirq.testing.assert_equivalent_repr(op)
 
 
-def test_validation():
+def test_validation() -> None:
     with pytest.raises(ValueError, match='sign'):
         _ = cirq.interop.quirk.QuirkInputRotationOperation(
             identifier='xxx',
@@ -123,7 +125,7 @@ def test_validation():
         )
 
 
-def test_input_rotation_with_qubits():
+def test_input_rotation_with_qubits() -> None:
     a, b, c, d, e = cirq.LineQubit.range(5)
     x, y, z, t, w = cirq.LineQubit.range(10, 15)
     op = cirq.interop.quirk.QuirkInputRotationOperation(
@@ -143,7 +145,7 @@ def test_input_rotation_with_qubits():
     )
 
 
-def test_input_rotation_cell_with_qubits():
+def test_input_rotation_cell_with_qubits() -> None:
     a, b, c, d, e = cirq.LineQubit.range(5)
     x, y, z, t, w = cirq.LineQubit.range(10, 15)
     cell = cirq.interop.quirk.cells.input_rotation_cells.InputRotationCell(
@@ -162,7 +164,7 @@ def test_input_rotation_cell_with_qubits():
     )
 
 
-def test_input_rotation_cell_with_qubits_before_register_specified():
+def test_input_rotation_cell_with_qubits_before_register_specified() -> None:
     d, e = cirq.LineQubit.range(3, 5)
     x, y, z, t, w = cirq.LineQubit.range(10, 15)
     cell = cirq.interop.quirk.cells.input_rotation_cells.InputRotationCell(
@@ -181,7 +183,7 @@ def test_input_rotation_cell_with_qubits_before_register_specified():
     )
 
 
-def test_repr():
+def test_repr() -> None:
     a, b, c, d, e = cirq.LineQubit.range(5)
     cirq.testing.assert_equivalent_repr(
         cirq.interop.quirk.cells.input_rotation_cells.InputRotationCell(

cirq/interop/quirk/cells/measurement_cells.py

@@ -11,7 +11,10 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import cast, Iterable, Iterator, Optional, TYPE_CHECKING
+
+from __future__ import annotations
+
+from typing import cast, Iterable, Iterator, TYPE_CHECKING
 
 from cirq import ops
 from cirq.interop.quirk.cells.cell import CellMaker, ExplicitOperationsCell
@@ -27,7 +30,7 @@ def generate_all_measurement_cell_makers() -> Iterator[CellMaker]:
     yield _measurement("XDetector", basis_change=ops.Y**0.5)
 
 
-def _measurement(identifier: str, basis_change: Optional['cirq.Gate'] = None) -> CellMaker:
+def _measurement(identifier: str, basis_change: cirq.Gate | None = None) -> CellMaker:
     return CellMaker(
         identifier=identifier,
         size=1,

cirq/interop/quirk/cells/measurement_cells_test.py

@@ -12,11 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import cirq
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_measurement_gates():
+def test_measurement_gates() -> None:
     a, b, c = cirq.LineQubit.range(3)
     assert_url_to_circuit_returns(
         '{"cols":[["Measure","Measure"],["Measure","Measure"]]}',