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

cirq/ops/identity.py

@@ -13,9 +13,11 @@
 # limitations under the License.
 """IdentityGate."""
 
+from __future__ import annotations
+
 import numbers
 from types import NotImplementedType
-from typing import Any, Dict, Optional, Sequence, Tuple, TYPE_CHECKING, Union
+from typing import Any, Sequence, TYPE_CHECKING
 
 import numpy as np
 import sympy
@@ -39,7 +41,7 @@ class IdentityGate(raw_types.Gate):
     """
 
     def __init__(
-        self, num_qubits: Optional[int] = None, qid_shape: Optional[Tuple[int, ...]] = None
+        self, num_qubits: int | None = None, qid_shape: tuple[int, ...] | None = None
     ) -> None:
         """Inits IdentityGate.
 
@@ -63,10 +65,10 @@ class IdentityGate(raw_types.Gate):
         if len(self._qid_shape) != num_qubits:
             raise ValueError('len(qid_shape) != num_qubits')
 
-    def _act_on_(self, sim_state: 'cirq.SimulationStateBase', qubits: Sequence['cirq.Qid']):
+    def _act_on_(self, sim_state: cirq.SimulationStateBase, qubits: Sequence[cirq.Qid]):
         return True
 
-    def _qid_shape_(self) -> Tuple[int, ...]:
+    def _qid_shape_(self) -> tuple[int, ...]:
         return self._qid_shape
 
     def num_qubits(self) -> int:
@@ -77,7 +79,7 @@ class IdentityGate(raw_types.Gate):
             return self
         return NotImplemented
 
-    def _commutes_(self, other: Any, *, atol: float = 1e-8) -> Union[bool, NotImplementedType]:
+    def _commutes_(self, other: Any, *, atol: float = 1e-8) -> bool | NotImplementedType:
         """The identity gate commutes with all other gates."""
         if not isinstance(other, raw_types.Gate):
             return NotImplemented
@@ -89,7 +91,7 @@ class IdentityGate(raw_types.Gate):
     def _unitary_(self) -> np.ndarray:
         return np.identity(np.prod(self._qid_shape, dtype=np.int64).item())
 
-    def _apply_unitary_(self, args: 'protocols.ApplyUnitaryArgs') -> Optional[np.ndarray]:
+    def _apply_unitary_(self, args: protocols.ApplyUnitaryArgs) -> np.ndarray | None:
         return args.target_tensor
 
     def _pauli_expansion_(self) -> value.LinearDict[str]:
@@ -104,7 +106,7 @@ class IdentityGate(raw_types.Gate):
             return f'cirq.IdentityGate({len(self._qid_shape)})'
         return f'cirq.IdentityGate(qid_shape={self._qid_shape!r})'
 
-    def _decompose_(self, qubits) -> 'cirq.OP_TREE':
+    def _decompose_(self, qubits) -> cirq.OP_TREE:
         return []
 
     def __str__(self) -> str:
@@ -118,13 +120,13 @@ class IdentityGate(raw_types.Gate):
     def _trace_distance_bound_(self) -> float:
         return 0.0
 
-    def _json_dict_(self) -> Dict[str, Any]:
+    def _json_dict_(self) -> dict[str, Any]:
         other = {}
         if not all(d == 2 for d in self._qid_shape):
             other['qid_shape'] = self._qid_shape
         return {'num_qubits': len(self._qid_shape), **other}
 
-    def _mul_with_qubits(self, qubits: Tuple['cirq.Qid', ...], other):
+    def _mul_with_qubits(self, qubits: tuple[cirq.Qid, ...], other):
         if isinstance(other, raw_types.Operation):
             return other
         if isinstance(other, numbers.Complex):
@@ -135,12 +137,12 @@ class IdentityGate(raw_types.Gate):
 
     _rmul_with_qubits = _mul_with_qubits
 
-    def _circuit_diagram_info_(self, args) -> Tuple[str, ...]:
+    def _circuit_diagram_info_(self, args) -> tuple[str, ...]:
         if self.num_qubits() <= 0:
             return NotImplemented
         return ('I',) * self.num_qubits()
 
-    def _qasm_(self, args: 'cirq.QasmArgs', qubits: Tuple['cirq.Qid', ...]) -> Optional[str]:
+    def _qasm_(self, args: cirq.QasmArgs, qubits: tuple[cirq.Qid, ...]) -> str | None:
         args.validate_version('2.0', '3.0')
         return ''.join([args.format('id {0};\n', qubit) for qubit in qubits])
 
@@ -163,7 +165,7 @@ document(
 )
 
 
-def identity_each(*qubits: 'cirq.Qid') -> 'cirq.Operation':
+def identity_each(*qubits: cirq.Qid) -> cirq.Operation:
     """Returns a single IdentityGate applied to all the given qubits.
 
     Args:

cirq/ops/identity_test.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
+
 import itertools
 from typing import Any
 from unittest import mock
@@ -23,7 +26,7 @@ import cirq
 
 
 @pytest.mark.parametrize('num_qubits', [1, 2, 4])
-def test_identity_init(num_qubits):
+def test_identity_init(num_qubits) -> None:
     assert cirq.IdentityGate(num_qubits).num_qubits() == num_qubits
     assert cirq.qid_shape(cirq.IdentityGate(num_qubits)) == (2,) * num_qubits
     assert cirq.qid_shape(cirq.IdentityGate(3, (1, 2, 3))) == (1, 2, 3)
@@ -34,7 +37,7 @@ def test_identity_init(num_qubits):
         cirq.IdentityGate()
 
 
-def test_identity_on_each():
+def test_identity_on_each() -> None:
     q0, q1, q2 = cirq.LineQubit.range(3)
     assert cirq.I.on_each(q0, q1, q2) == [cirq.I(q0), cirq.I(q1), cirq.I(q2)]
     assert cirq.I.on_each([q0, [q1], q2]) == [cirq.I(q0), cirq.I(q1), cirq.I(q2)]
@@ -43,7 +46,7 @@ def test_identity_on_each():
         cirq.I.on_each('abc')
 
 
-def test_identity_on_each_iter_second():
+def test_identity_on_each_iter_second() -> None:
     class Q(cirq.Qid):
         @property
         def dimension(self) -> int:
@@ -60,7 +63,7 @@ def test_identity_on_each_iter_second():
     assert cirq.I.on_each(q) == [cirq.I(q)]
 
 
-def test_identity_on_each_only_single_qubit():
+def test_identity_on_each_only_single_qubit() -> None:
     q0, q1 = cirq.LineQubit.range(2)
     q0_3, q1_3 = q0.with_dimension(3), q1.with_dimension(3)
     assert cirq.I.on_each(q0, q1) == [cirq.I.on(q0), cirq.I.on(q1)]
@@ -70,7 +73,7 @@ def test_identity_on_each_only_single_qubit():
     ]
 
 
-def test_identity_on_each_two_qubits():
+def test_identity_on_each_two_qubits() -> None:
     q0, q1, q2, q3 = cirq.LineQubit.range(4)
     q0_3, q1_3 = q0.with_dimension(3), q1.with_dimension(3)
     assert cirq.IdentityGate(2).on_each([(q0, q1)]) == [cirq.IdentityGate(2)(q0, q1)]
@@ -93,14 +96,14 @@ def test_identity_on_each_two_qubits():
 
 
 @pytest.mark.parametrize('num_qubits', [1, 2, 4])
-def test_identity_unitary(num_qubits):
+def test_identity_unitary(num_qubits) -> None:
     i = cirq.IdentityGate(num_qubits)
     assert np.allclose(cirq.unitary(i), np.identity(2**num_qubits))
     i3 = cirq.IdentityGate(num_qubits, (3,) * num_qubits)
     assert np.allclose(cirq.unitary(i3), np.identity(3**num_qubits))
 
 
-def test_identity_str():
+def test_identity_str() -> None:
     assert str(cirq.IdentityGate(1)) == 'I'
     assert str(cirq.IdentityGate(2)) == 'I(2)'
     # Qid shape is not included in str
@@ -108,13 +111,13 @@ def test_identity_str():
     assert str(cirq.IdentityGate(2, (1, 2))) == 'I(2)'
 
 
-def test_identity_repr():
+def test_identity_repr() -> None:
     assert repr(cirq.I) == 'cirq.I'
     assert repr(cirq.IdentityGate(5)) == 'cirq.IdentityGate(5)'
     assert repr(cirq.IdentityGate(qid_shape=(2, 3))) == 'cirq.IdentityGate(qid_shape=(2, 3))'
 
 
-def test_identity_apply_unitary():
+def test_identity_apply_unitary() -> None:
     v = np.array([1, 0])
     result = cirq.apply_unitary(cirq.I, cirq.ApplyUnitaryArgs(v, np.array([0, 1]), (0,)))
     assert result is v
@@ -126,7 +129,7 @@ def test_identity_apply_unitary():
     assert result is v
 
 
-def test_identity_eq():
+def test_identity_eq() -> None:
     equals_tester = cirq.testing.EqualsTester()
     equals_tester.make_equality_group(
         lambda: cirq.I, lambda: cirq.IdentityGate(1), lambda: cirq.IdentityGate(1, (2,))
@@ -137,12 +140,12 @@ def test_identity_eq():
     equals_tester.add_equality_group(cirq.IdentityGate(4, (1, 2, 3, 4)))
 
 
-def test_identity_trace_distance_bound():
+def test_identity_trace_distance_bound() -> None:
     assert cirq.I._trace_distance_bound_() == 0
     assert cirq.IdentityGate(num_qubits=2)._trace_distance_bound_() == 0
 
 
-def test_identity_pow():
+def test_identity_pow() -> None:
     I = cirq.I
     q = cirq.NamedQubit('q')
 
@@ -156,28 +159,29 @@ def test_identity_pow():
         _ = I(q) ** q
 
 
-def test_pauli_expansion_notimplemented():
+def test_pauli_expansion_notimplemented() -> None:
     assert cirq.IdentityGate(1, (3,))._pauli_expansion_() == NotImplemented
 
 
 @pytest.mark.parametrize(
     'gate_type, num_qubits', itertools.product((cirq.IdentityGate,), range(1, 5))
 )
-def test_consistent_protocols(gate_type, num_qubits):
+def test_consistent_protocols(gate_type, num_qubits) -> None:
     gate = gate_type(num_qubits=num_qubits)
     cirq.testing.assert_implements_consistent_protocols(gate, qubit_count=num_qubits)
 
 
-def test_identity_global():
+def test_identity_global() -> None:
     qubits = cirq.LineQubit.range(3)
     assert cirq.identity_each(*qubits) == cirq.IdentityGate(3).on(*qubits)
     qids = cirq.LineQid.for_qid_shape((1, 2, 3))
     assert cirq.identity_each(*qids) == cirq.IdentityGate(3, (1, 2, 3)).on(*qids)
     with pytest.raises(ValueError, match='Not a cirq.Qid'):
-        cirq.identity_each(qubits)  # The user forgot to expand the list for example.
+        # The user forgot to expand the list for example.
+        cirq.identity_each(qubits)  # type: ignore[arg-type]
 
 
-def test_identity_mul():
+def test_identity_mul() -> None:
     class UnknownGate(cirq.testing.SingleQubitGate):
         pass
 
@@ -204,19 +208,19 @@ def test_identity_mul():
     assert 1j * i == cirq.PauliString(coefficient=1j)
 
 
-def test_identity_short_circuits_act_on():
+def test_identity_short_circuits_act_on() -> None:
     args = mock.Mock(cirq.SimulationState)
     args._act_on_fallback_.side_effect = mock.Mock(side_effect=Exception('No!'))
     cirq.act_on(cirq.IdentityGate(1)(cirq.LineQubit(0)), args)
 
 
-def test_identity_commutes():
+def test_identity_commutes() -> None:
     assert cirq.commutes(cirq.I, cirq.X)
     with pytest.raises(TypeError):
         cirq.commutes(cirq.I, "Gate")
 
 
-def test_identity_diagram():
+def test_identity_diagram() -> None:
     cirq.testing.assert_has_diagram(
         cirq.Circuit(cirq.IdentityGate(3).on_each(cirq.LineQubit.range(3))),
         """

cirq/ops/kraus_channel.py

@@ -1,5 +1,8 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
-from typing import Any, Dict, FrozenSet, Iterable, Mapping, Tuple, TYPE_CHECKING, Union
+
+from __future__ import annotations
+
+from typing import Any, Iterable, Mapping, TYPE_CHECKING
 
 import numpy as np
 
@@ -29,7 +32,7 @@ class KrausChannel(raw_types.Gate):
     def __init__(
         self,
         kraus_ops: Iterable[np.ndarray],
-        key: Union[str, 'cirq.MeasurementKey', None] = None,
+        key: str | cirq.MeasurementKey | None = None,
         validate: bool = False,
     ):
         kraus_ops = list(kraus_ops)
@@ -56,7 +59,7 @@ class KrausChannel(raw_types.Gate):
         self._key = key
 
     @staticmethod
-    def from_channel(channel: 'cirq.Gate', key: Union[str, 'cirq.MeasurementKey', None] = None):
+    def from_channel(channel: cirq.Gate, key: str | cirq.MeasurementKey | None = None):
         """Creates a copy of a channel with the given measurement key."""
         return KrausChannel(kraus_ops=list(protocols.kraus(channel)), key=key)
 
@@ -80,7 +83,7 @@ class KrausChannel(raw_types.Gate):
             return NotImplemented
         return str(self._key)
 
-    def _measurement_key_obj_(self) -> 'cirq.MeasurementKey':
+    def _measurement_key_obj_(self) -> cirq.MeasurementKey:
         if self._key is None:
             return NotImplemented
         return self._key
@@ -92,16 +95,16 @@ class KrausChannel(raw_types.Gate):
             return self
         return KrausChannel(kraus_ops=self._kraus_ops, key=key_map[str(self._key)])
 
-    def _with_key_path_(self, path: Tuple[str, ...]):
+    def _with_key_path_(self, path: tuple[str, ...]):
         return KrausChannel(kraus_ops=self._kraus_ops, key=protocols.with_key_path(self._key, path))
 
-    def _with_key_path_prefix_(self, prefix: Tuple[str, ...]):
+    def _with_key_path_prefix_(self, prefix: tuple[str, ...]):
         return KrausChannel(
             kraus_ops=self._kraus_ops, key=protocols.with_key_path_prefix(self._key, prefix)
         )
 
     def _with_rescoped_keys_(
-        self, path: Tuple[str, ...], bindable_keys: FrozenSet['cirq.MeasurementKey']
+        self, path: tuple[str, ...], bindable_keys: frozenset[cirq.MeasurementKey]
     ):
         return KrausChannel(
             kraus_ops=self._kraus_ops,
@@ -119,7 +122,7 @@ class KrausChannel(raw_types.Gate):
             args.append(f'key=\'{self._key}\'')
         return f'cirq.KrausChannel({", ".join(args)})'
 
-    def _json_dict_(self) -> Dict[str, Any]:
+    def _json_dict_(self) -> dict[str, Any]:
         return protocols.obj_to_dict_helper(self, ['_kraus_ops', '_key'])
 
     @classmethod

cirq/ops/kraus_channel_test.py

@@ -1,11 +1,14 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import numpy as np
 import pytest
 
 import cirq
 
 
-def test_kraus_channel_from_channel():
+def test_kraus_channel_from_channel() -> None:
     q0 = cirq.LineQubit(0)
     dp = cirq.depolarize(0.1)
     kc = cirq.KrausChannel.from_channel(dp, key='dp')
@@ -21,7 +24,7 @@ def test_kraus_channel_from_channel():
     assert results.measurements['dp'] in range(4)
 
 
-def test_kraus_channel_equality():
+def test_kraus_channel_equality() -> None:
     dp_pt1 = cirq.depolarize(0.1)
     dp_pt2 = cirq.depolarize(0.2)
     kc_a1 = cirq.KrausChannel.from_channel(dp_pt1, key='a')
@@ -44,7 +47,7 @@ def test_kraus_channel_equality():
     assert x_meas != x_meas_inv
 
 
-def test_kraus_channel_remap_keys():
+def test_kraus_channel_remap_keys() -> None:
     dp = cirq.depolarize(0.1)
     kc = cirq.KrausChannel.from_channel(dp)
     with pytest.raises(TypeError):
@@ -62,7 +65,7 @@ def test_kraus_channel_remap_keys():
     assert cirq.with_measurement_key_mapping(kc_a, {'a': 'b'}) == kc_b
 
 
-def test_kraus_channel_from_kraus():
+def test_kraus_channel_from_kraus() -> None:
     q0 = cirq.LineQubit(0)
     # This is equivalent to an X-basis measurement.
     ops = [np.array([[1, 1], [1, 1]]) * 0.5, np.array([[1, -1], [-1, 1]]) * 0.5]
@@ -77,7 +80,7 @@ def test_kraus_channel_from_kraus():
     assert results.measurements['x_meas'] == 0
 
 
-def test_kraus_channel_str():
+def test_kraus_channel_str() -> None:
     # This is equivalent to an X-basis measurement.
     ops = [np.array([[1, 1], [1, 1]]) * 0.5, np.array([[1, -1], [-1, 1]]) * 0.5]
     x_meas = cirq.KrausChannel(ops)
@@ -96,7 +99,7 @@ def test_kraus_channel_str():
     )
 
 
-def test_kraus_channel_repr():
+def test_kraus_channel_repr() -> None:
     # This is equivalent to an X-basis measurement.
     ops = [
         np.array([[1, 1], [1, 1]], dtype=np.complex64) * 0.5,
@@ -113,14 +116,14 @@ key='x_meas')"""
     )
 
 
-def test_empty_ops_fails():
-    ops = []
+def test_empty_ops_fails() -> None:
+    ops: list[np.ndarray] = []
 
     with pytest.raises(ValueError, match='must have at least one operation'):
         _ = cirq.KrausChannel(kraus_ops=ops, key='m')
 
 
-def test_ops_mismatch_fails():
+def test_ops_mismatch_fails() -> None:
     op2 = np.zeros((4, 4))
     op2[1][1] = 1
     ops = [np.array([[1, 0], [0, 0]]), op2]
@@ -129,14 +132,14 @@ def test_ops_mismatch_fails():
         _ = cirq.KrausChannel(kraus_ops=ops, key='m')
 
 
-def test_nonqubit_kraus_ops_fails():
+def test_nonqubit_kraus_ops_fails() -> None:
     ops = [np.array([[1, 0, 0], [0, 0, 0]]), np.array([[0, 0, 0], [0, 1, 0]])]
 
     with pytest.raises(ValueError, match='Input Kraus ops'):
         _ = cirq.KrausChannel(kraus_ops=ops, key='m')
 
 
-def test_validate():
+def test_validate() -> None:
     # Not quite CPTP.
     ops = [np.array([[1, 0], [0, 0]]), np.array([[0, 0], [0, 0.9]])]
     with pytest.raises(ValueError, match='CPTP map'):