cirq-core 1.6.0.dev20250515174529__py3-none-any.whl → 1.6.0.dev20250516154249__py3-none-any.whl

cirq/contrib/qasm_import/qasm.py

@@ -14,11 +14,15 @@
 
 from __future__ import annotations
 
-from cirq import circuits
+from typing import TYPE_CHECKING
+
 from cirq.contrib.qasm_import._parser import QasmParser
 
+if TYPE_CHECKING:
+    import cirq
+
 
-def circuit_from_qasm(qasm: str) -> circuits.Circuit:
+def circuit_from_qasm(qasm: str) -> cirq.Circuit:
     """Parses an OpenQASM string to `cirq.Circuit`.
 
     Args:

cirq/contrib/qasm_import/qasm_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 numpy as np
 
 import cirq

cirq/contrib/qcircuit/qcircuit_diagram_info.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from typing import Optional, Tuple
 
 from cirq import ops, protocols

cirq/contrib/qcircuit/qcircuit_diagram_info_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 cirq
 import cirq.contrib.qcircuit as ccq
 

cirq/contrib/qcircuit/qcircuit_pdf.py

@@ -13,17 +13,22 @@
 # limitations under the License.
 
 
+from __future__ import annotations
+
 import errno
 import os
+from typing import TYPE_CHECKING
 
 from pylatex import Document, NoEscape, Package
 
-from cirq import circuits
 from cirq.contrib.qcircuit.qcircuit_diagram import circuit_to_latex_using_qcircuit
 
+if TYPE_CHECKING:
+    import cirq
+
 
 def circuit_to_pdf_using_qcircuit_via_tex(
-    circuit: circuits.Circuit,
+    circuit: cirq.Circuit,
     filepath: str,
     pdf_kwargs=None,
     qcircuit_kwargs=None,

cirq/contrib/qcircuit/qcircuit_pdf_test.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from unittest import mock
 
 import pylatex

cirq/contrib/qcircuit/qcircuit_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 cirq
 import cirq.contrib.qcircuit as ccq
 import cirq.testing as ct

cirq/contrib/quantum_volume/quantum_volume.py

@@ -3,6 +3,8 @@
 https://arxiv.org/abs/1811.12926.
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Callable, Dict, List, Optional, Set, Tuple
 

cirq/contrib/quantum_volume/quantum_volume_test.py

@@ -1,6 +1,8 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
 """Tests for the Quantum Volume utilities."""
 
+from __future__ import annotations
+
 import io
 from unittest.mock import MagicMock, Mock
 

cirq/contrib/quimb/density_matrix_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import numpy as np
 
 import cirq

cirq/contrib/quimb/grid_circuits.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from typing import Iterator
 
 import networkx as nx

cirq/contrib/quimb/grid_circuits_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import networkx as nx
 import numpy as np
 import pytest

cirq/contrib/quimb/mps_simulator_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import itertools
 import math
 

cirq/contrib/quimb/state_vector_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import functools
 import operator
 

cirq/contrib/quirk/export_to_quirk.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import json
 import urllib.parse
 from typing import Any, cast, Iterable, List, Tuple

cirq/contrib/quirk/export_to_quirk_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
 import sympy
 

cirq/contrib/quirk/linearize_circuit.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from typing import Callable
 
 from cirq import circuits, devices, ops

cirq/contrib/routing/device.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import itertools
 from typing import Dict, Iterable, Tuple
 

cirq/contrib/routing/device_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 networkx as nx
 import pytest
 

cirq/contrib/routing/greedy_test.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from multiprocessing import Process
 
 import pytest

cirq/contrib/routing/initialization_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 random
 
 import networkx as nx

cirq/contrib/routing/router.py

@@ -14,13 +14,15 @@
 
 from __future__ import annotations
 
-from typing import Callable, Optional
+from typing import Callable, Optional, TYPE_CHECKING
 
 import networkx as nx
 
 from cirq import circuits, protocols
 from cirq.contrib.routing.greedy import route_circuit_greedily
-from cirq.contrib.routing.swap_network import SwapNetwork
+
+if TYPE_CHECKING:
+    from cirq.contrib.routing.swap_network import SwapNetwork
 
 ROUTERS = {'greedy': route_circuit_greedily}
 

cirq/contrib/routing/router_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 itertools
 import random
 

cirq/contrib/routing/swap_network_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 itertools
 
 import pytest

cirq/contrib/routing/utils_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 networkx as nx
 import pytest
 

cirq/contrib/shuffle_circuits/shuffle_circuits_with_readout_benchmarking.py

@@ -17,13 +17,15 @@
 from __future__ import annotations
 
 import time
-from typing import Dict, List, Optional, Tuple, Union
+from typing import Dict, List, Optional, Tuple, TYPE_CHECKING, Union
 
 import numpy as np
 
 from cirq import circuits, ops, protocols, work
 from cirq.experiments import SingleQubitReadoutCalibrationResult
-from cirq.study import ResultDict
+
+if TYPE_CHECKING:
+    from cirq.study import ResultDict
 
 
 def _validate_input(

cirq/contrib/shuffle_circuits/shuffle_circuits_with_readout_benchmarking_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
 
 import numpy as np

cirq/contrib/svg/svg_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import IPython.display
 import numpy as np
 import pytest

cirq/devices/device_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import networkx as nx
 
 import cirq

cirq/devices/grid_device_metadata_test.py

@@ -11,8 +11,11 @@
 # 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.
+
 """Tests for GridDevicemetadata."""
 
+from __future__ import annotations
+
 import networkx as nx
 import pytest
 

cirq/interop/quirk/cells/parse_test.py

@@ -19,7 +19,7 @@ import sympy
 from cirq.interop.quirk.cells.parse import parse_complex, parse_formula, parse_matrix
 
 
-def test_parse_matrix():
+def test_parse_matrix() -> None:
     s = np.sqrt(0.5)
     np.testing.assert_allclose(
         parse_matrix('{{√½,√½},{-√½,√½}}'), np.array([[s, s], [-s, s]]), atol=1e-8
@@ -32,7 +32,7 @@ def test_parse_matrix():
     )
 
 
-def test_parse_matrix_failures():
+def test_parse_matrix_failures() -> None:
     with pytest.raises(ValueError, match='Not surrounded by {{}}'):
         _ = parse_matrix('1')
     with pytest.raises(ValueError, match='Not surrounded by {{}}'):
@@ -45,7 +45,7 @@ def test_parse_matrix_failures():
         _ = parse_matrix('{{x}}')
 
 
-def test_parse_real_formula():
+def test_parse_real_formula() -> None:
     t = sympy.Symbol('t')
     assert parse_formula('1/2') == 0.5
     assert parse_formula('t*t + ln(t)') == t * t + sympy.ln(t)
@@ -58,12 +58,12 @@ def test_parse_real_formula():
         _ = parse_formula('i')
 
 
-def test_parse_formula_failures():
+def test_parse_formula_failures() -> None:
     with pytest.raises(TypeError, match='formula must be a string'):
-        _ = parse_formula(2)
+        _ = parse_formula(2)  # type: ignore[arg-type]
 
     with pytest.raises(TypeError, match='formula must be a string'):
-        _ = parse_formula([])
+        _ = parse_formula([])  # type: ignore[arg-type]
 
     with pytest.raises(ValueError, match='Unrecognized token'):
         _ = parse_formula('5*__**DSA **)SADD')
@@ -72,7 +72,7 @@ def test_parse_formula_failures():
         _ = parse_formula('5*x')
 
 
-def test_parse_complex():
+def test_parse_complex() -> None:
     assert parse_complex('0') == 0
     assert parse_complex('1') == 1
     assert parse_complex('i') == 1j
@@ -84,7 +84,7 @@ def test_parse_complex():
     np.testing.assert_allclose(parse_complex('exp 2'), np.e**2, atol=1e-8)
 
 
-def test_parse_complex_raw_cases_from_quirk():
+def test_parse_complex_raw_cases_from_quirk() -> None:
     assert parse_complex("0") == 0
     assert parse_complex("1") == 1
     assert parse_complex("-1") == -1
@@ -116,7 +116,7 @@ def test_parse_complex_raw_cases_from_quirk():
     np.testing.assert_allclose(parse_complex("2       pi"), 2 * np.pi)
 
 
-def test_parse_complex_expression_cases_from_quirk():
+def test_parse_complex_expression_cases_from_quirk() -> None:
     np.testing.assert_allclose(parse_complex("1/3"), 1 / 3)
     np.testing.assert_allclose(parse_complex("2/3/5"), (2 / 3) / 5)
     np.testing.assert_allclose(parse_complex("2/3/5*7/13"), ((((2 / 3) / 5)) * 7) / 13)
@@ -152,7 +152,7 @@ def test_parse_complex_expression_cases_from_quirk():
     np.testing.assert_allclose(parse_complex("cos(acos(0.5))"), 0.5, atol=1e-8)
 
 
-def test_parse_complex_expression_failures():
+def test_parse_complex_expression_failures() -> None:
     with pytest.raises(ValueError, match='Incomplete expression'):
         _ = parse_formula('(')
     with pytest.raises(ValueError, match=r"unmatched '\)'"):

cirq/ops/pauli_string_phasor.py

@@ -199,7 +199,7 @@ class PauliStringPhasor(gate_operation.GateOperation):
         syms = tuple(sym(qubit) for qubit in qubits)
         return protocols.CircuitDiagramInfo(wire_symbols=syms, exponent=self.exponent_relative)
 
-    def conjugated_by(self, clifford: 'cirq.OP_TREE') -> 'PauliStringPhasor':
+    def conjugated_by(self, clifford: cirq.OP_TREE) -> PauliStringPhasor:
         r"""Returns the Pauli string conjugated by a clifford operation.
 
         The PauliStringPhasor $P$ conjugated by the Clifford operation $C$ is

cirq/testing/equals_tester_test.py

@@ -19,7 +19,7 @@ import pytest
 from cirq.testing.equals_tester import EqualsTester
 
 
-def test_add_equality_group_correct():
+def test_add_equality_group_correct() -> None:
     eq = EqualsTester()
 
     eq.add_equality_group(fractions.Fraction(1, 1))
@@ -39,7 +39,7 @@ def test_add_equality_group_correct():
     eq.add_equality_group('unrelated')
 
 
-def test_assert_make_equality_group():
+def test_assert_make_equality_group() -> None:
     eq = EqualsTester()
 
     with pytest.raises(AssertionError, match="can't be in the same"):
@@ -57,20 +57,20 @@ def test_assert_make_equality_group():
         eq.make_equality_group(lambda: 3)
 
 
-def test_add_equality_group_not_equivalent():
+def test_add_equality_group_not_equivalent() -> None:
     eq = EqualsTester()
     with pytest.raises(AssertionError, match="can't be in the same"):
         eq.add_equality_group(1, 2)
 
 
-def test_add_equality_group_not_disjoint():
+def test_add_equality_group_not_disjoint() -> None:
     eq = EqualsTester()
     eq.add_equality_group(1)
     with pytest.raises(AssertionError, match="can't be in different"):
         eq.add_equality_group(1)
 
 
-def test_add_equality_group_bad_hash():
+def test_add_equality_group_bad_hash() -> None:
     class KeyHash:
         def __init__(self, k, h):
             self._k = k
@@ -94,7 +94,7 @@ def test_add_equality_group_bad_hash():
         eq.add_equality_group(KeyHash('c', 2), KeyHash('c', 3))
 
 
-def test_add_equality_group_exception_hash():
+def test_add_equality_group_exception_hash() -> None:
     class FailHash:
         def __hash__(self):
             raise ValueError('injected failure')
@@ -104,7 +104,7 @@ def test_add_equality_group_exception_hash():
         eq.add_equality_group(FailHash())
 
 
-def test_fails_when_forgot_type_check():
+def test_fails_when_forgot_type_check() -> None:
     eq = EqualsTester()
 
     class NoTypeCheckEqualImplementation:
@@ -124,11 +124,12 @@ def test_fails_when_forgot_type_check():
         eq.add_equality_group(NoTypeCheckEqualImplementation())
 
 
-def test_fails_when_equal_to_everything():
+def test_fails_when_equal_to_everything() -> None:
     eq = EqualsTester()
 
     class AllEqual:
-        __hash__ = None
+        def __hash__(self) -> int:
+            return 0
 
         def __eq__(self, other):
             return True
@@ -136,11 +137,12 @@ def test_fails_when_equal_to_everything():
         def __ne__(self, other):
             return False
 
+    assert hash(AllEqual()) == 0
     with pytest.raises(AssertionError, match="can't be in different"):
         eq.add_equality_group(AllEqual())
 
 
-def test_fails_hash_is_default_and_inconsistent():
+def test_fails_hash_is_default_and_inconsistent() -> None:
     eq = EqualsTester()
 
     class DefaultHashImplementation:
@@ -161,7 +163,7 @@ def test_fails_hash_is_default_and_inconsistent():
         eq.make_equality_group(DefaultHashImplementation)
 
 
-def test_fails_when_ne_is_inconsistent():
+def test_fails_when_ne_is_inconsistent() -> None:
     eq = EqualsTester()
 
     class InconsistentNeImplementation:
@@ -185,7 +187,7 @@ def test_fails_when_ne_is_inconsistent():
         eq.make_equality_group(InconsistentNeImplementation)
 
 
-def test_fails_when_ne_is_inconsistent_due_to_not_implemented():
+def test_fails_when_ne_is_inconsistent_due_to_not_implemented() -> None:
     eq = EqualsTester()
 
     class InconsistentNeImplementation:
@@ -207,7 +209,7 @@ def test_fails_when_ne_is_inconsistent_due_to_not_implemented():
         eq.make_equality_group(InconsistentNeImplementation)
 
 
-def test_fails_when_not_reflexive():
+def test_fails_when_not_reflexive() -> None:
     eq = EqualsTester()
 
     class NotReflexiveImplementation:
@@ -226,7 +228,7 @@ def test_fails_when_not_reflexive():
         eq.add_equality_group(NotReflexiveImplementation())
 
 
-def test_fails_when_not_commutative():
+def test_fails_when_not_commutative() -> None:
     eq = EqualsTester()
 
     class NotCommutativeImplementation:
@@ -248,14 +250,14 @@ def test_fails_when_not_commutative():
         eq.add_equality_group(NotCommutativeImplementation(1), NotCommutativeImplementation(0))
 
 
-def test_works_on_types():
+def test_works_on_types() -> None:
     eq = EqualsTester()
     eq.add_equality_group(object)
     eq.add_equality_group(int)
     eq.add_equality_group(object())
 
 
-def test_returns_not_implemented_for_other_types():
+def test_returns_not_implemented_for_other_types() -> None:
     # First we demonstrate an example of the problem.
 
     # FirstClass is the class that is broken.
@@ -323,7 +325,7 @@ def test_returns_not_implemented_for_other_types():
     eq.add_equality_group(ThirdClass("a"), ThirdClass("a"))
 
 
-def test_not_implemented_error():
+def test_not_implemented_error() -> None:
     # Common bug is to return NotImplementedError instead of NotImplemented.
     class NotImplementedErrorCase:
         def __init__(self, val):

cirq/value/random_state_test.py

@@ -17,7 +17,7 @@ import numpy as np
 import cirq
 
 
-def test_parse_random_state():
+def test_parse_random_state() -> None:
     global_state = np.random.get_state()
 
     def rand(prng):
@@ -34,11 +34,11 @@ def test_parse_random_state():
     eq.add_equality_group(*vals)
 
     seed = np.random.randint(2**31)
-    prngs = [
+    prngs1 = [
         np.random.RandomState(seed),
         cirq.value.parse_random_state(np.random.RandomState(seed)),
         cirq.value.parse_random_state(seed),
     ]
-    vals = [prng.rand() for prng in prngs]
+    vals = [prng.rand() for prng in prngs1]
     eq = cirq.testing.EqualsTester()
     eq.add_equality_group(*vals)

cirq/value/timestamp.py

@@ -64,6 +64,10 @@ class Timestamp:
     def __sub__(self, other: Duration) -> 'Timestamp':
         pass
 
+    @overload
+    def __sub__(self, other: timedelta) -> 'Timestamp':
+        pass
+
     def __sub__(self, other):
         if isinstance(other, Duration):
             return Timestamp(picos=self._picos - other.total_picos())