cirq-core 1.6.0.dev20250509215532__py3-none-any.whl → 1.6.0.dev20250514001158__py3-none-any.whl

cirq/transformers/analytical_decompositions/two_qubit_to_ms_test.py

@@ -130,7 +130,7 @@ def assert_ms_depth_below(operations, threshold):
     (2, _random_double_MS_effect()) for _ in range(10)
 ])
 # yapf: enable
-def test_two_to_ops(max_ms_depth: int, effect: np.ndarray):
+def test_two_to_ops(max_ms_depth: int, effect: np.ndarray) -> None:
     q0 = cirq.NamedQubit('q0')
     q1 = cirq.NamedQubit('q1')
 

cirq/transformers/drop_empty_moments_test.py

@@ -15,7 +15,7 @@
 import cirq
 
 
-def test_drop():
+def test_drop() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     cirq.testing.assert_same_circuits(
@@ -28,7 +28,7 @@ def test_drop():
     )
 
 
-def test_drop_empty_moments():
+def test_drop_empty_moments() -> None:
     q1, q2 = cirq.LineQubit.range(2)
     c_nested = cirq.FrozenCircuit(
         cirq.Moment(), cirq.Moment(), cirq.Moment([cirq.CNOT(q1, q2)]), cirq.Moment()

cirq/transformers/eject_phased_paulis_test.py

@@ -98,7 +98,7 @@ def quick_circuit(*moments: Iterable[cirq.OP_TREE]) -> cirq.Circuit:
     )
 
 
-def test_absorbs_z():
+def test_absorbs_z() -> None:
     q = cirq.NamedQubit('q')
     x = sympy.Symbol('x')
 
@@ -165,7 +165,7 @@ def test_absorbs_z():
     )
 
 
-def test_crosses_czs():
+def test_crosses_czs() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     x = sympy.Symbol('x')
@@ -232,7 +232,7 @@ def test_crosses_czs():
     )
 
 
-def test_toggles_measurements():
+def test_toggles_measurements() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     x = sympy.Symbol('x')
@@ -289,7 +289,7 @@ def test_toggles_measurements():
     )
 
 
-def test_eject_phased_xz():
+def test_eject_phased_xz() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     c = cirq.Circuit(
@@ -305,7 +305,7 @@ def test_eject_phased_xz():
     cirq.testing.assert_circuits_with_terminal_measurements_are_equivalent(c, c_expected, 1e-8)
 
 
-def test_cancels_other_full_w():
+def test_cancels_other_full_w() -> None:
     q = cirq.NamedQubit('q')
     x = sympy.Symbol('x')
     y = sympy.Symbol('y')
@@ -365,7 +365,7 @@ def test_cancels_other_full_w():
     )
 
 
-def test_phases_partial_ws():
+def test_phases_partial_ws() -> None:
     q = cirq.NamedQubit('q')
     x = sympy.Symbol('x')
     y = sympy.Symbol('y')
@@ -420,7 +420,7 @@ def test_phases_partial_ws():
 
 
 @pytest.mark.parametrize('sym', [sympy.Symbol('x'), sympy.Symbol('x') + 1])
-def test_blocked_by_unknown_and_symbols(sym):
+def test_blocked_by_unknown_and_symbols(sym) -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
 
@@ -442,7 +442,7 @@ def test_blocked_by_unknown_and_symbols(sym):
     )
 
 
-def test_blocked_by_nocompile_tag():
+def test_blocked_by_nocompile_tag() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
 
@@ -453,7 +453,7 @@ def test_blocked_by_nocompile_tag():
     )
 
 
-def test_zero_x_rotation():
+def test_zero_x_rotation() -> None:
     a = cirq.NamedQubit('a')
 
     assert_optimizes(before=quick_circuit([cirq.rx(0)(a)]), expected=quick_circuit([cirq.rx(0)(a)]))

cirq/transformers/heuristic_decompositions/two_qubit_gate_tabulation_test.py

@@ -41,7 +41,7 @@ _random_2Q_unitaries = np.array([random_special_unitary(4, random_state=_rng) fo
 
 @pytest.mark.parametrize('tabulation', [sycamore_tabulation, sqrt_iswap_tabulation])
 @pytest.mark.parametrize('target', _random_2Q_unitaries)
-def test_gate_compilation_matches_expected_max_infidelity(tabulation, target):
+def test_gate_compilation_matches_expected_max_infidelity(tabulation, target) -> None:
     result = tabulation.compile_two_qubit_gate(target)
 
     assert result.success
@@ -50,7 +50,7 @@ def test_gate_compilation_matches_expected_max_infidelity(tabulation, target):
 
 
 @pytest.mark.parametrize('tabulation', [sycamore_tabulation, sqrt_iswap_tabulation])
-def test_gate_compilation_on_base_gate_standard(tabulation):
+def test_gate_compilation_on_base_gate_standard(tabulation) -> None:
     base_gate = tabulation.base_gate
 
     result = tabulation.compile_two_qubit_gate(base_gate)
@@ -61,7 +61,7 @@ def test_gate_compilation_on_base_gate_standard(tabulation):
     assert fidelity > 0.99999
 
 
-def test_gate_compilation_on_base_gate_identity():
+def test_gate_compilation_on_base_gate_identity() -> None:
     tabulation = two_qubit_gate_product_tabulation(np.eye(4), 0.25)
     base_gate = tabulation.base_gate
 
@@ -73,7 +73,7 @@ def test_gate_compilation_on_base_gate_identity():
     assert fidelity > 0.99999
 
 
-def test_gate_compilation_missing_points_raises_error():
+def test_gate_compilation_missing_points_raises_error() -> None:
     with pytest.raises(ValueError, match='Failed to tabulate a'):
         two_qubit_gate_product_tabulation(
             np.eye(4), 0.4, allow_missed_points=False, random_state=_rng
@@ -81,7 +81,7 @@ def test_gate_compilation_missing_points_raises_error():
 
 
 @pytest.mark.parametrize('seed', [0, 1])
-def test_sycamore_gate_tabulation(seed):
+def test_sycamore_gate_tabulation(seed) -> None:
     base_gate = cirq.unitary(cirq.FSimGate(np.pi / 2, np.pi / 6))
     tab = two_qubit_gate_product_tabulation(
         base_gate, 0.1, sample_scaling=2, random_state=np.random.RandomState(seed)
@@ -90,7 +90,7 @@ def test_sycamore_gate_tabulation(seed):
     assert result.success
 
 
-def test_sycamore_gate_tabulation_repr():
+def test_sycamore_gate_tabulation_repr() -> None:
     simple_tabulation = TwoQubitGateTabulation(
         np.array([[(1 + 0j), 0j, 0j, 0j]], dtype=np.complex128),
         np.array([[(1 + 0j), 0j, 0j, 0j]], dtype=np.complex128),
@@ -102,7 +102,7 @@ def test_sycamore_gate_tabulation_repr():
     assert_equivalent_repr(simple_tabulation)
 
 
-def test_sycamore_gate_tabulation_eq():
+def test_sycamore_gate_tabulation_eq() -> None:
     assert sycamore_tabulation == sycamore_tabulation
     assert sycamore_tabulation != sqrt_iswap_tabulation
     assert sycamore_tabulation != 1

cirq/transformers/insertion_sort_test.py

@@ -16,7 +16,7 @@ import cirq
 import cirq.transformers
 
 
-def test_insertion_sort():
+def test_insertion_sort() -> None:
     c = cirq.Circuit(
         cirq.CZ(cirq.q(2), cirq.q(1)),
         cirq.CZ(cirq.q(2), cirq.q(4)),

cirq/transformers/merge_k_qubit_gates_test.py

@@ -37,7 +37,7 @@ def assert_optimizes(optimized: cirq.AbstractCircuit, expected: cirq.AbstractCir
     cirq.testing.assert_same_circuits(optimized, expected)
 
 
-def test_merge_1q_unitaries():
+def test_merge_1q_unitaries() -> None:
     q, q2 = cirq.LineQubit.range(2)
     # 1. Combines trivial 1q sequence.
     c = cirq.Circuit(cirq.X(q) ** 0.5, cirq.Z(q) ** 0.5, cirq.X(q) ** -0.5)
@@ -57,7 +57,7 @@ def test_merge_1q_unitaries():
     assert isinstance(c[-1][q].gate, cirq.MatrixGate)
 
 
-def test_respects_nocompile_tags():
+def test_respects_nocompile_tags() -> None:
     q = cirq.NamedQubit("q")
     c = cirq.Circuit(
         [cirq.Z(q), cirq.H(q), cirq.X(q), cirq.H(q), cirq.X(q).with_tags("nocompile"), cirq.H(q)]
@@ -70,12 +70,12 @@ def test_respects_nocompile_tags():
     assert isinstance(c[-1][q].gate, cirq.MatrixGate)
 
 
-def test_ignores_2qubit_target():
+def test_ignores_2qubit_target() -> None:
     c = cirq.Circuit(cirq.CZ(*cirq.LineQubit.range(2)))
     assert_optimizes(optimized=cirq.merge_k_qubit_unitaries(c, k=1), expected=c)
 
 
-def test_ignore_unsupported_gate():
+def test_ignore_unsupported_gate() -> None:
     class UnsupportedExample(cirq.testing.SingleQubitGate):
         pass
 
@@ -83,7 +83,7 @@ def test_ignore_unsupported_gate():
     assert_optimizes(optimized=cirq.merge_k_qubit_unitaries(c, k=1), expected=c)
 
 
-def test_1q_rewrite():
+def test_1q_rewrite() -> None:
     q0, q1 = cirq.LineQubit.range(2)
     circuit = cirq.Circuit(
         cirq.X(q0), cirq.Y(q0), cirq.X(q1), cirq.CZ(q0, q1), cirq.Y(q1), cirq.measure(q0, q1)
@@ -98,12 +98,12 @@ def test_1q_rewrite():
     )
 
 
-def test_merge_k_qubit_unitaries_raises():
+def test_merge_k_qubit_unitaries_raises() -> None:
     with pytest.raises(ValueError, match="k should be greater than or equal to 1"):
         _ = cirq.merge_k_qubit_unitaries(cirq.Circuit())
 
 
-def test_merge_complex_circuit_preserving_moment_structure():
+def test_merge_complex_circuit_preserving_moment_structure() -> None:
     q = cirq.LineQubit.range(3)
     c_orig = cirq.Circuit(
         cirq.Moment(cirq.H.on_each(*q)),
@@ -191,7 +191,7 @@ a: ═════════════════════════
     )
 
 
-def test_merge_k_qubit_unitaries_deep():
+def test_merge_k_qubit_unitaries_deep() -> None:
     q = cirq.LineQubit.range(2)
     h_cz_y = [cirq.H(q[0]), cirq.CZ(*q), cirq.Y(q[1])]
     c_orig = cirq.Circuit(
@@ -253,7 +253,7 @@ def test_merge_k_qubit_unitaries_deep():
     cirq.testing.assert_same_circuits(c_new_matrix, c_expected_matrix)
 
 
-def test_merge_k_qubit_unitaries_deep_recurses_on_large_circuit_op():
+def test_merge_k_qubit_unitaries_deep_recurses_on_large_circuit_op() -> None:
     q = cirq.LineQubit.range(2)
     c_orig = cirq.Circuit(
         cirq.CircuitOperation(cirq.FrozenCircuit(cirq.X(q[0]), cirq.H(q[0]), cirq.CNOT(*q)))

cirq/transformers/qubit_management_transformers_test.py

@@ -66,7 +66,7 @@ def get_decompose_func(gate_type, qm):
     return decompose_func
 
 
-def test_map_clean_and_borrowable_qubits_greedy_types():
+def test_map_clean_and_borrowable_qubits_greedy_types() -> None:
     qm = cirq.ops.SimpleQubitManager()
     q = cirq.LineQubit.range(2)
     g = GateAllocInDecompose(1)
@@ -130,7 +130,7 @@ ancilla_1: ───X───X───
     )
 
 
-def test_map_clean_and_borrowable_qubits_borrows():
+def test_map_clean_and_borrowable_qubits_borrows() -> None:
     qm = cirq.ops.SimpleQubitManager()
     op = GateAllocAndBorrowInDecompose(3).on(cirq.NamedQubit("original"))
     extra = cirq.LineQubit.range(3)
@@ -233,7 +233,7 @@ original: ────@───@───@───@─────@───@
     )
 
 
-def test_map_clean_and_borrowable_qubits_deallocates_only_once():
+def test_map_clean_and_borrowable_qubits_deallocates_only_once() -> None:
     q = [cirq.ops.BorrowableQubit(i) for i in range(2)] + [cirq.q('q')]
     circuit = cirq.Circuit(cirq.X.on_each(*q), cirq.Y(q[1]), cirq.Z(q[1]))
     greedy_mm = cirq.GreedyQubitManager(prefix="a", size=2)

cirq/transformers/randomized_measurements_test.py

@@ -18,7 +18,7 @@ import cirq
 import cirq.transformers.randomized_measurements as rand_meas
 
 
-def test_randomized_measurements_appends_two_moments_on_returned_circuit():
+def test_randomized_measurements_appends_two_moments_on_returned_circuit() -> None:
     # Create a 4-qubit circuit
     q0, q1, q2, q3 = cirq.LineQubit.range(4)
     circuit_pre = cirq.Circuit(
@@ -34,7 +34,9 @@ def test_randomized_measurements_appends_two_moments_on_returned_circuit():
         assert num_moments_post == num_moments_pre + 2
 
 
-def test_append_randomized_measurements_leaves_qubits_not_in_specified_subsystem_unchanged():
+def test_append_randomized_measurements_leaves_qubits_not_in_specified_subsystem_unchanged() -> (
+    None
+):
     # Create a 4-qubit circuit
     q0, q1, q2, q3 = cirq.LineQubit.range(4)
     circuit = cirq.Circuit([cirq.H(q0), cirq.CNOT(q0, q1), cirq.CNOT(q1, q2), cirq.CNOT(q2, q3)])
@@ -46,7 +48,9 @@ def test_append_randomized_measurements_leaves_qubits_not_in_specified_subsystem
     assert circuit.operation_at(q3, 4) is None
 
 
-def test_append_randomized_measurements_leaves_qubits_not_in_noncontinuous_subsystem_unchanged():
+def test_append_random_measurements_leaves_qubits_not_in_noncontinuous_subsystem_unchanged() -> (
+    None
+):
     # Create a 4-qubit circuit
     q0, q1, q2, q3 = cirq.LineQubit.range(4)
     circuit = cirq.Circuit([cirq.H(q0), cirq.CNOT(q0, q1), cirq.CNOT(q1, q2), cirq.CNOT(q2, q3)])
@@ -59,7 +63,7 @@ def test_append_randomized_measurements_leaves_qubits_not_in_noncontinuous_subsy
     assert circuit.operation_at(q3, 4) is None
 
 
-def test_exception():
+def test_exception() -> None:
     q0, q1, q2, q3 = cirq.LineQubit.range(4)
     circuit = cirq.Circuit([cirq.H(q0), cirq.CNOT(q0, q1), cirq.CNOT(q1, q2), cirq.CNOT(q2, q3)])
 

cirq/transformers/routing/line_initial_mapper_test.py

@@ -66,7 +66,7 @@ def construct_valid_circuit():
     )
 
 
-def test_valid_circuit():
+def test_valid_circuit() -> None:
     # Any circuit with a (full connectivity) graph of disjoint lines should be directly
     # executable after mapping a a supporting device topology without the need for inserting
     # any swaps.
@@ -79,7 +79,7 @@ def test_valid_circuit():
     device.validate_circuit(mapped_circuit)
 
 
-def test_long_line_on_grid_device():
+def test_long_line_on_grid_device() -> None:
     # tests
     #   -if strategy is able to map a single long line onto the device whenever the device topology
     #   supports it (i.e. is Hamiltonian)
@@ -106,7 +106,7 @@ def test_long_line_on_grid_device():
         mapper.initial_mapping(step_circuit)
 
 
-def test_small_circuit_on_grid_device():
+def test_small_circuit_on_grid_device() -> None:
     circuit = construct_small_circuit()
     device_graph = cirq.testing.construct_grid_device(7, 7).metadata.nx_graph
     mapper = cirq.LineInitialMapper(device_graph)
@@ -126,7 +126,7 @@ def test_small_circuit_on_grid_device():
     cirq.testing.assert_same_circuits(circuit.transform_qubits(mapping), expected_circuit)
 
 
-def test_small_circuit_on_ring_device():
+def test_small_circuit_on_ring_device() -> None:
     circuit = construct_small_circuit()
     device_graph = cirq.testing.construct_ring_device(10, directed=True).metadata.nx_graph
 
@@ -187,7 +187,7 @@ def test_large_random_circuits_grid_device(
     assert nx.is_connected(nx.induced_subgraph(glob_device_graph, mapping.values()))
 
 
-def test_repr():
+def test_repr() -> None:
     device_graph = cirq.testing.construct_grid_device(7, 7).metadata.nx_graph
     mapper = cirq.LineInitialMapper(device_graph)
     cirq.testing.assert_equivalent_repr(mapper, setup_code='import cirq\nimport networkx as nx')

cirq/transformers/routing/route_circuit_cqc_test.py

@@ -17,7 +17,7 @@ import pytest
 import cirq
 
 
-def test_directed_device():
+def test_directed_device() -> None:
     device = cirq.testing.construct_ring_device(10, directed=True)
     device_graph = device.metadata.nx_graph
     with pytest.raises(ValueError, match="Device graph must be undirected."):
@@ -33,7 +33,7 @@ def test_directed_device():
         for op_density in [0.3, 0.5, 0.7]
     ],
 )
-def test_route_small_circuit_random(n_qubits, n_moments, op_density, seed):
+def test_route_small_circuit_random(n_qubits, n_moments, op_density, seed) -> None:
     c_orig = cirq.testing.random_circuit(
         qubits=n_qubits, n_moments=n_moments, op_density=op_density, random_state=seed
     )
@@ -47,7 +47,7 @@ def test_route_small_circuit_random(n_qubits, n_moments, op_density, seed):
     )
 
 
-def test_high_qubit_count():
+def test_high_qubit_count() -> None:
     c_orig = cirq.testing.random_circuit(qubits=40, n_moments=350, op_density=0.4, random_state=0)
     device = cirq.testing.construct_grid_device(7, 7)
     device_graph = device.metadata.nx_graph
@@ -56,7 +56,7 @@ def test_high_qubit_count():
     device.validate_circuit(c_routed)
 
 
-def test_multi_qubit_gate_inputs():
+def test_multi_qubit_gate_inputs() -> None:
     device = cirq.testing.construct_grid_device(4, 4)
     device_graph = device.metadata.nx_graph
     router = cirq.RouteCQC(device_graph)
@@ -97,7 +97,7 @@ def test_multi_qubit_gate_inputs():
     device.validate_circuit(c_routed)
 
 
-def test_circuit_with_measurement_gates():
+def test_circuit_with_measurement_gates() -> None:
     device = cirq.testing.construct_ring_device(3)
     device_graph = device.metadata.nx_graph
     q = cirq.LineQubit.range(3)
@@ -108,7 +108,7 @@ def test_circuit_with_measurement_gates():
     cirq.testing.assert_same_circuits(routed_circuit, circuit)
 
 
-def test_circuit_with_two_qubit_intermediate_measurement_gate():
+def test_circuit_with_two_qubit_intermediate_measurement_gate() -> None:
     device = cirq.testing.construct_ring_device(2)
     device_graph = device.metadata.nx_graph
     router = cirq.RouteCQC(device_graph)
@@ -121,7 +121,7 @@ def test_circuit_with_two_qubit_intermediate_measurement_gate():
     device.validate_circuit(routed_circuit)
 
 
-def test_circuit_with_multi_qubit_intermediate_measurement_gate_and_with_default_key():
+def test_circuit_with_multi_qubit_intermediate_measurement_gate_and_with_default_key() -> None:
     device = cirq.testing.construct_ring_device(3)
     device_graph = device.metadata.nx_graph
     router = cirq.RouteCQC(device_graph)
@@ -135,7 +135,7 @@ def test_circuit_with_multi_qubit_intermediate_measurement_gate_and_with_default
     cirq.testing.assert_same_circuits(routed_circuit, expected)
 
 
-def test_circuit_with_multi_qubit_intermediate_measurement_gate_with_custom_key():
+def test_circuit_with_multi_qubit_intermediate_measurement_gate_with_custom_key() -> None:
     device = cirq.testing.construct_ring_device(3)
     device_graph = device.metadata.nx_graph
     router = cirq.RouteCQC(device_graph)
@@ -150,7 +150,7 @@ def test_circuit_with_multi_qubit_intermediate_measurement_gate_with_custom_key(
         )
 
 
-def test_circuit_with_non_unitary_and_global_phase():
+def test_circuit_with_non_unitary_and_global_phase() -> None:
     device = cirq.testing.construct_ring_device(4)
     device_graph = device.metadata.nx_graph
     q = cirq.LineQubit.range(3)
@@ -176,7 +176,7 @@ def test_circuit_with_non_unitary_and_global_phase():
     cirq.testing.assert_same_circuits(routed_circuit, expected)
 
 
-def test_circuit_with_tagged_ops():
+def test_circuit_with_tagged_ops() -> None:
     device = cirq.testing.construct_ring_device(4)
     device_graph = device.metadata.nx_graph
     q = cirq.LineQubit.range(3)
@@ -205,7 +205,7 @@ def test_circuit_with_tagged_ops():
     cirq.testing.assert_same_circuits(routed_circuit, expected)
 
 
-def test_already_valid_circuit():
+def test_already_valid_circuit() -> None:
     device = cirq.testing.construct_ring_device(10)
     device_graph = device.metadata.nx_graph
     circuit = cirq.Circuit(
@@ -222,7 +222,7 @@ def test_already_valid_circuit():
     cirq.testing.assert_same_circuits(routed_circuit, circuit)
 
 
-def test_empty_circuit():
+def test_empty_circuit() -> None:
     device = cirq.testing.construct_grid_device(5, 5)
     device_graph = device.metadata.nx_graph
     empty_circuit = cirq.Circuit()
@@ -235,7 +235,7 @@ def test_empty_circuit():
     )
 
 
-def test_repr():
+def test_repr() -> None:
     device = cirq.testing.construct_ring_device(10)
     device_graph = device.metadata.nx_graph
     router = cirq.RouteCQC(device_graph)

cirq/transformers/routing/visualize_routed_circuit_test.py

@@ -17,7 +17,7 @@ import pytest
 import cirq
 
 
-def test_routed_circuit_with_mapping_simple():
+def test_routed_circuit_with_mapping_simple() -> None:
     q = cirq.LineQubit.range(2)
     circuit = cirq.Circuit([cirq.Moment(cirq.SWAP(q[0], q[1]).with_tags(cirq.RoutingSwapTag()))])
     expected_diagram = """
@@ -57,7 +57,7 @@ def test_routed_circuit_with_mapping_simple():
         cirq.routed_circuit_with_mapping(circuit)
 
 
-def test_routed_circuit_with_mapping_multi_swaps():
+def test_routed_circuit_with_mapping_multi_swaps() -> None:
     q = cirq.LineQubit.range(6)
     circuit = cirq.Circuit(
         [

cirq/transformers/synchronize_terminal_measurements_test.py

@@ -48,14 +48,14 @@ def assert_optimizes(before, after, measure_only_moment=True, with_context=False
     cirq.testing.assert_same_circuits(transformed_circuit, c_expected)
 
 
-def test_no_move():
+def test_no_move() -> None:
     q1 = cirq.NamedQubit('q1')
     before = cirq.Circuit([cirq.Moment([cirq.H(q1)])])
     after = before
     assert_optimizes(before=before, after=after)
 
 
-def test_simple_align():
+def test_simple_align() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     before = cirq.Circuit(
@@ -76,7 +76,7 @@ def test_simple_align():
     assert_optimizes(before=before, after=before, with_context=True)
 
 
-def test_simple_partial_align():
+def test_simple_partial_align() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     before = cirq.Circuit(
@@ -96,7 +96,7 @@ def test_simple_partial_align():
     assert_optimizes(before=before, after=before, with_context=True)
 
 
-def test_slide_forward_one():
+def test_slide_forward_one() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     q3 = cirq.NamedQubit('q3')
@@ -119,7 +119,7 @@ def test_slide_forward_one():
     assert_optimizes(before=before, after=after_no_compile, with_context=True)
 
 
-def test_no_slide_forward_one():
+def test_no_slide_forward_one() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     q3 = cirq.NamedQubit('q3')
@@ -128,7 +128,7 @@ def test_no_slide_forward_one():
     assert_optimizes(before=before, after=after, measure_only_moment=False)
 
 
-def test_blocked_shift_one():
+def test_blocked_shift_one() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     before = cirq.Circuit(
@@ -150,7 +150,7 @@ def test_blocked_shift_one():
     assert_optimizes(before=before, after=before, with_context=True)
 
 
-def test_complex_move():
+def test_complex_move() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     q3 = cirq.NamedQubit('q3')
@@ -182,7 +182,7 @@ def test_complex_move():
     assert_optimizes(before=before, after=before, with_context=True)
 
 
-def test_complex_move_no_slide():
+def test_complex_move_no_slide() -> None:
     q1 = cirq.NamedQubit('q1')
     q2 = cirq.NamedQubit('q2')
     q3 = cirq.NamedQubit('q3')
@@ -208,13 +208,13 @@ def test_complex_move_no_slide():
     assert_optimizes(before=before, after=before, measure_only_moment=False, with_context=True)
 
 
-def test_multi_qubit():
+def test_multi_qubit() -> None:
     q0, q1 = cirq.LineQubit.range(2)
     circuit = cirq.Circuit(cirq.H(q1), cirq.measure(q0, q1, key='m'))
     assert_optimizes(before=circuit, after=circuit)
 
 
-def test_classically_controlled_op():
+def test_classically_controlled_op() -> None:
     q0, q1 = cirq.LineQubit.range(2)
     circuit = cirq.Circuit(
         cirq.H(q0), cirq.measure(q0, key='m'), cirq.X(q1).with_classical_controls('m')

cirq/transformers/target_gatesets/compilation_target_gateset_test.py

@@ -24,7 +24,7 @@ if TYPE_CHECKING:
     from cirq.protocols.decompose_protocol import DecomposeResult
 
 
-def test_compilation_target_gateset():
+def test_compilation_target_gateset() -> None:
     class ExampleTargetGateset(cirq.CompilationTargetGateset):
         def __init__(self):
             super().__init__(cirq.AnyUnitaryGateFamily(2))
@@ -94,7 +94,7 @@ class ExampleCXTargetGateset(cirq.TwoQubitCompilationTargetGateset):
         )
 
 
-def test_two_qubit_compilation_leaves_single_gates_in_gateset():
+def test_two_qubit_compilation_leaves_single_gates_in_gateset() -> None:
     q = cirq.LineQubit.range(2)
     gateset = ExampleCXTargetGateset()
 
@@ -105,7 +105,7 @@ def test_two_qubit_compilation_leaves_single_gates_in_gateset():
     cirq.testing.assert_same_circuits(cirq.optimize_for_target_gateset(c, gateset=gateset), c)
 
 
-def test_two_qubit_compilation_merges_runs_of_single_qubit_gates():
+def test_two_qubit_compilation_merges_runs_of_single_qubit_gates() -> None:
     q = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.CNOT(*q), cirq.X(q[0]), cirq.Y(q[0]), cirq.CNOT(*q))
     cirq.testing.assert_same_circuits(
@@ -118,7 +118,7 @@ def test_two_qubit_compilation_merges_runs_of_single_qubit_gates():
     )
 
 
-def test_two_qubit_compilation_decompose_operation_not_implemented():
+def test_two_qubit_compilation_decompose_operation_not_implemented() -> None:
     gateset = ExampleCXTargetGateset()
     q = cirq.LineQubit.range(3)
     assert gateset.decompose_to_target_gateset(cirq.measure(q[0]), 1) is NotImplemented
@@ -130,7 +130,7 @@ def test_two_qubit_compilation_decompose_operation_not_implemented():
     assert gateset.decompose_to_target_gateset(cirq.CCZ(*q), 1) is NotImplemented
 
 
-def test_two_qubit_compilation_merge_and_replace_to_target_gateset():
+def test_two_qubit_compilation_merge_and_replace_to_target_gateset() -> None:
     q = cirq.LineQubit.range(2)
     c_orig = cirq.Circuit(
         cirq.Moment(cirq.Z(q[1]), cirq.X(q[0])),
@@ -164,7 +164,7 @@ def test_two_qubit_compilation_merge_and_replace_to_target_gateset():
     )
 
 
-def test_two_qubit_compilation_merge_and_replace_inefficient_component():
+def test_two_qubit_compilation_merge_and_replace_inefficient_component() -> None:
     q = cirq.LineQubit.range(2)
     c_orig = cirq.Circuit(
         cirq.Moment(cirq.X(q[0])),
@@ -208,7 +208,7 @@ m: ═════════════════════════
     )
 
 
-def test_two_qubit_compilation_replaces_only_if_2q_gate_count_is_less():
+def test_two_qubit_compilation_replaces_only_if_2q_gate_count_is_less() -> None:
     class ExampleTargetGateset(cirq.TwoQubitCompilationTargetGateset):
         def __init__(self):
             super().__init__(cirq.X, cirq.CNOT)
@@ -228,7 +228,7 @@ def test_two_qubit_compilation_replaces_only_if_2q_gate_count_is_less():
     cirq.testing.assert_same_circuits(c_new, c_expected)
 
 
-def test_create_transformer_with_kwargs_raises():
+def test_create_transformer_with_kwargs_raises() -> None:
     with pytest.raises(SyntaxError, match="must not contain `context`"):
         cirq.create_transformer_with_kwargs(
             cirq.merge_k_qubit_unitaries, k=2, context=cirq.TransformerContext()