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

cirq/transformers/target_gatesets/cz_gateset_test.py

@@ -56,7 +56,7 @@ def assert_optimization_not_broken(circuit: cirq.Circuit):
     )
 
 
-def test_convert_to_cz_preserving_moment_structure():
+def test_convert_to_cz_preserving_moment_structure() -> None:
     q = cirq.LineQubit.range(5)
     op = lambda q0, q1: cirq.H(q1).controlled_by(q0)
     c_orig = cirq.Circuit(
@@ -100,7 +100,7 @@ def test_convert_to_cz_preserving_moment_structure():
     )
 
 
-def test_clears_paired_cnot():
+def test_clears_paired_cnot() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimizes(
         before=cirq.Circuit(cirq.Moment(cirq.CNOT(a, b)), cirq.Moment(cirq.CNOT(a, b))),
@@ -108,7 +108,7 @@ def test_clears_paired_cnot():
     )
 
 
-def test_ignores_czs_separated_by_parameterized():
+def test_ignores_czs_separated_by_parameterized() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimizes(
         before=cirq.Circuit(
@@ -129,19 +129,19 @@ def test_ignores_czs_separated_by_parameterized():
     )
 
 
-def test_cnots_separated_by_single_gates_correct():
+def test_cnots_separated_by_single_gates_correct() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimization_not_broken(cirq.Circuit(cirq.CNOT(a, b), cirq.H(b), cirq.CNOT(a, b)))
 
 
-def test_czs_separated_by_single_gates_correct():
+def test_czs_separated_by_single_gates_correct() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimization_not_broken(
         cirq.Circuit(cirq.CZ(a, b), cirq.X(b), cirq.X(b), cirq.X(b), cirq.CZ(a, b))
     )
 
 
-def test_inefficient_circuit_correct():
+def test_inefficient_circuit_correct() -> None:
     t = 0.1
     v = 0.11
     a, b = cirq.LineQubit.range(2)
@@ -166,7 +166,7 @@ def test_inefficient_circuit_correct():
     )
 
 
-def test_optimizes_single_iswap():
+def test_optimizes_single_iswap() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.ISWAP(a, b))
     assert_optimization_not_broken(c)
@@ -174,7 +174,7 @@ def test_optimizes_single_iswap():
     assert len([1 for op in c.all_operations() if len(op.qubits) == 2]) == 2
 
 
-def test_optimizes_tagged_partial_cz():
+def test_optimizes_tagged_partial_cz() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit((cirq.CZ**0.5)(a, b).with_tags('mytag'))
     assert_optimization_not_broken(c)
@@ -184,7 +184,7 @@ def test_optimizes_tagged_partial_cz():
     ), 'It should take 2 CZ gates to decompose a CZ**0.5 gate'
 
 
-def test_not_decompose_czs():
+def test_not_decompose_czs() -> None:
     circuit = cirq.Circuit(
         cirq.CZPowGate(exponent=1, global_shift=-0.5).on(*cirq.LineQubit.range(2))
     )
@@ -201,7 +201,7 @@ def test_not_decompose_czs():
         ),
     ),
 )
-def test_decompose_partial_czs(circuit):
+def test_decompose_partial_czs(circuit) -> None:
     circuit = cirq.optimize_for_target_gateset(
         circuit, gateset=cirq.CZTargetGateset(), ignore_failures=False
     )
@@ -216,7 +216,7 @@ def test_decompose_partial_czs(circuit):
     assert num_part_cz == 0
 
 
-def test_not_decompose_partial_czs():
+def test_not_decompose_partial_czs() -> None:
     circuit = cirq.Circuit(
         cirq.CZPowGate(exponent=0.1, global_shift=-0.5)(*cirq.LineQubit.range(2))
     )
@@ -232,7 +232,7 @@ def test_not_decompose_partial_czs():
     assert num_part_cz == 1
 
 
-def test_avoids_decompose_when_matrix_available():
+def test_avoids_decompose_when_matrix_available() -> None:
     class OtherXX(cirq.testing.TwoQubitGate):  # pragma: no cover
         def _has_unitary_(self) -> bool:
             return True
@@ -261,7 +261,7 @@ def test_avoids_decompose_when_matrix_available():
     assert len(c) == 0
 
 
-def test_composite_gates_without_matrix():
+def test_composite_gates_without_matrix() -> None:
     class CompositeExample(cirq.testing.SingleQubitGate):
         def _decompose_(self, qubits):
             yield cirq.X(qubits[0])
@@ -289,7 +289,7 @@ def test_composite_gates_without_matrix():
     )
 
 
-def test_unsupported_gate():
+def test_unsupported_gate() -> None:
     class UnsupportedExample(cirq.testing.TwoQubitGate):
         pass
 
@@ -319,11 +319,11 @@ def test_unsupported_gate():
         cirq.CZTargetGateset(additional_gates=()),
     ],
 )
-def test_repr(gateset):
+def test_repr(gateset) -> None:
     cirq.testing.assert_equivalent_repr(gateset)
 
 
-def test_with_commutation():
+def test_with_commutation() -> None:
     c = cirq.Circuit(
         cirq.CZ(cirq.q(0), cirq.q(1)), cirq.CZ(cirq.q(1), cirq.q(2)), cirq.CZ(cirq.q(0), cirq.q(1))
     )
@@ -335,7 +335,7 @@ def test_with_commutation():
     assert got == cirq.Circuit(cirq.CZ(cirq.q(1), cirq.q(2)))
 
 
-def test_reorder_operations_and_preserve_moment_structure_raises():
+def test_reorder_operations_and_preserve_moment_structure_raises() -> None:
     with pytest.raises(
         ValueError, match='reorder_operations and preserve_moment_structure can not both be True'
     ):

cirq/transformers/target_gatesets/sqrt_iswap_gateset_test.py

@@ -60,7 +60,7 @@ def assert_optimization_not_broken(
     )
 
 
-def test_convert_to_sqrt_iswap_preserving_moment_structure():
+def test_convert_to_sqrt_iswap_preserving_moment_structure() -> None:
     q = cirq.LineQubit.range(5)
     op = lambda q0, q1: cirq.H(q1).controlled_by(q0)
     c_orig = cirq.Circuit(
@@ -112,7 +112,7 @@ def test_convert_to_sqrt_iswap_preserving_moment_structure():
     ],
 )
 @pytest.mark.parametrize('use_sqrt_iswap_inv', [True, False])
-def test_two_qubit_gates_with_symbols(gate: cirq.Gate, use_sqrt_iswap_inv: bool):
+def test_two_qubit_gates_with_symbols(gate: cirq.Gate, use_sqrt_iswap_inv: bool) -> None:
     # Note that even though these gates are not natively supported by
     # `cirq.parameterized_2q_op_to_sqrt_iswap_operations`, the transformation succeeds because
     # `cirq.optimize_for_target_gateset` also relies on `cirq.decompose` as a fallback.
@@ -142,12 +142,12 @@ def test_two_qubit_gates_with_symbols(gate: cirq.Gate, use_sqrt_iswap_inv: bool)
         )
 
 
-def test_sqrt_iswap_gateset_raises():
+def test_sqrt_iswap_gateset_raises() -> None:
     with pytest.raises(ValueError, match="`required_sqrt_iswap_count` must be 0, 1, 2, or 3"):
         _ = cirq.SqrtIswapTargetGateset(required_sqrt_iswap_count=4)
 
 
-def test_sqrt_iswap_gateset_eq():
+def test_sqrt_iswap_gateset_eq() -> None:
     eq = cirq.testing.EqualsTester()
     eq.add_equality_group(
         cirq.SqrtIswapTargetGateset(), cirq.SqrtIswapTargetGateset(use_sqrt_iswap_inv=False)
@@ -179,11 +179,11 @@ def test_sqrt_iswap_gateset_eq():
         cirq.SqrtIswapTargetGateset(additional_gates=()),
     ],
 )
-def test_sqrt_iswap_gateset_repr(gateset):
+def test_sqrt_iswap_gateset_repr(gateset) -> None:
     cirq.testing.assert_equivalent_repr(gateset)
 
 
-def test_simplifies_sqrt_iswap():
+def test_simplifies_sqrt_iswap() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimizes(
         before=cirq.Circuit(
@@ -204,7 +204,7 @@ def test_simplifies_sqrt_iswap():
     )
 
 
-def test_simplifies_sqrt_iswap_inv():
+def test_simplifies_sqrt_iswap_inv() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimizes(
         use_sqrt_iswap_inv=True,
@@ -226,7 +226,7 @@ def test_simplifies_sqrt_iswap_inv():
     )
 
 
-def test_works_with_tags():
+def test_works_with_tags() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimizes(
         before=cirq.Circuit(
@@ -240,7 +240,7 @@ def test_works_with_tags():
     )
 
 
-def test_no_touch_single_sqrt_iswap():
+def test_no_touch_single_sqrt_iswap() -> None:
     a, b = cirq.LineQubit.range(2)
     circuit = cirq.Circuit(
         [
@@ -252,7 +252,7 @@ def test_no_touch_single_sqrt_iswap():
     assert_optimizes(before=circuit, expected=circuit)
 
 
-def test_no_touch_single_sqrt_iswap_inv():
+def test_no_touch_single_sqrt_iswap_inv() -> None:
     a, b = cirq.LineQubit.range(2)
     circuit = cirq.Circuit(
         [
@@ -264,19 +264,19 @@ def test_no_touch_single_sqrt_iswap_inv():
     assert_optimizes(before=circuit, expected=circuit, use_sqrt_iswap_inv=True)
 
 
-def test_cnots_separated_by_single_gates_correct():
+def test_cnots_separated_by_single_gates_correct() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimization_not_broken(cirq.Circuit(cirq.CNOT(a, b), cirq.H(b), cirq.CNOT(a, b)))
 
 
-def test_czs_separated_by_single_gates_correct():
+def test_czs_separated_by_single_gates_correct() -> None:
     a, b = cirq.LineQubit.range(2)
     assert_optimization_not_broken(
         cirq.Circuit(cirq.CZ(a, b), cirq.X(b), cirq.X(b), cirq.X(b), cirq.CZ(a, b))
     )
 
 
-def test_inefficient_circuit_correct():
+def test_inefficient_circuit_correct() -> None:
     t = 0.1
     v = 0.11
     a, b = cirq.LineQubit.range(2)
@@ -301,7 +301,7 @@ def test_inefficient_circuit_correct():
     )
 
 
-def test_optimizes_single_iswap():
+def test_optimizes_single_iswap() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.ISWAP(a, b))
     assert_optimization_not_broken(c)
@@ -311,7 +311,7 @@ def test_optimizes_single_iswap():
     assert len([1 for op in c.all_operations() if len(op.qubits) == 2]) == 2
 
 
-def test_optimizes_single_inv_sqrt_iswap():
+def test_optimizes_single_inv_sqrt_iswap() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.SQRT_ISWAP_INV(a, b))
     assert_optimization_not_broken(c)
@@ -321,7 +321,7 @@ def test_optimizes_single_inv_sqrt_iswap():
     assert len([1 for op in c.all_operations() if len(op.qubits) == 2]) == 1
 
 
-def test_optimizes_single_iswap_require0():
+def test_optimizes_single_iswap_require0() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.CNOT(a, b), cirq.CNOT(a, b))  # Minimum 0 sqrt-iSWAP
     assert_optimization_not_broken(c, required_sqrt_iswap_count=0)
@@ -331,7 +331,7 @@ def test_optimizes_single_iswap_require0():
     assert len([1 for op in c.all_operations() if len(op.qubits) == 2]) == 0
 
 
-def test_optimizes_single_iswap_require0_raises():
+def test_optimizes_single_iswap_require0_raises() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.CNOT(a, b))  # Minimum 2 sqrt-iSWAP
     with pytest.raises(ValueError, match='cannot be decomposed into exactly 0 sqrt-iSWAP gates'):
@@ -342,7 +342,7 @@ def test_optimizes_single_iswap_require0_raises():
         )
 
 
-def test_optimizes_single_iswap_require1():
+def test_optimizes_single_iswap_require1() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.SQRT_ISWAP_INV(a, b))  # Minimum 1 sqrt-iSWAP
     assert_optimization_not_broken(c, required_sqrt_iswap_count=1)
@@ -352,7 +352,7 @@ def test_optimizes_single_iswap_require1():
     assert len([1 for op in c.all_operations() if len(op.qubits) == 2]) == 1
 
 
-def test_optimizes_single_iswap_require1_raises():
+def test_optimizes_single_iswap_require1_raises() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.CNOT(a, b))  # Minimum 2 sqrt-iSWAP
     with pytest.raises(ValueError, match='cannot be decomposed into exactly 1 sqrt-iSWAP gates'):
@@ -363,7 +363,7 @@ def test_optimizes_single_iswap_require1_raises():
         )
 
 
-def test_optimizes_single_iswap_require2():
+def test_optimizes_single_iswap_require2() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.SQRT_ISWAP_INV(a, b))  # Minimum 1 sqrt-iSWAP but 2 possible
     assert_optimization_not_broken(c, required_sqrt_iswap_count=2)
@@ -373,7 +373,7 @@ def test_optimizes_single_iswap_require2():
     assert len([1 for op in c.all_operations() if len(op.qubits) == 2]) == 2
 
 
-def test_optimizes_single_iswap_require2_raises():
+def test_optimizes_single_iswap_require2_raises() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.SWAP(a, b))  # Minimum 3 sqrt-iSWAP
     with pytest.raises(ValueError, match='cannot be decomposed into exactly 2 sqrt-iSWAP gates'):
@@ -384,7 +384,7 @@ def test_optimizes_single_iswap_require2_raises():
         )
 
 
-def test_optimizes_single_iswap_require3():
+def test_optimizes_single_iswap_require3() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.ISWAP(a, b))  # Minimum 2 sqrt-iSWAP but 3 possible
     assert_optimization_not_broken(c, required_sqrt_iswap_count=3)
@@ -394,7 +394,7 @@ def test_optimizes_single_iswap_require3():
     assert len([1 for op in c.all_operations() if len(op.qubits) == 2]) == 3
 
 
-def test_optimizes_single_inv_sqrt_iswap_require3():
+def test_optimizes_single_inv_sqrt_iswap_require3() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.SQRT_ISWAP_INV(a, b))
     assert_optimization_not_broken(c, required_sqrt_iswap_count=3)

cirq/value/angle_test.py

@@ -19,7 +19,7 @@ import sympy
 import cirq
 
 
-def test_canonicalize_half_turns():
+def test_canonicalize_half_turns() -> None:
     assert cirq.canonicalize_half_turns(0) == 0
     assert cirq.canonicalize_half_turns(1) == +1
     assert cirq.canonicalize_half_turns(-1) == +1
@@ -34,7 +34,7 @@ def test_canonicalize_half_turns():
     assert cirq.canonicalize_half_turns(sympy.Symbol('a') * 0 + 3) == 1
 
 
-def test_chosen_angle_to_half_turns():
+def test_chosen_angle_to_half_turns() -> None:
     assert cirq.chosen_angle_to_half_turns() == 1
     assert cirq.chosen_angle_to_half_turns(default=0.5) == 0.5
     assert cirq.chosen_angle_to_half_turns(half_turns=0.25, default=0.75) == 0.25
@@ -54,7 +54,7 @@ def test_chosen_angle_to_half_turns():
         _ = cirq.chosen_angle_to_half_turns(half_turns=0, rads=0, degs=0)
 
 
-def test_chosen_angle_to_canonical_half_turns():
+def test_chosen_angle_to_canonical_half_turns() -> None:
     assert cirq.chosen_angle_to_canonical_half_turns() == 1
     assert cirq.chosen_angle_to_canonical_half_turns(default=0.5) == 0.5
     assert cirq.chosen_angle_to_canonical_half_turns(half_turns=0.25, default=0.75) == 0.25

cirq/value/classical_data_test.py

@@ -22,7 +22,7 @@ mkey_c = cirq.MeasurementKey('c')
 two_qubits = tuple(cirq.LineQubit.range(2))
 
 
-def test_init():
+def test_init() -> None:
     cd = cirq.ClassicalDataDictionaryStore()
     assert cd.records == {}
     assert cd.keys() == ()
@@ -44,7 +44,7 @@ def test_init():
     }
 
 
-def test_record_measurement():
+def test_record_measurement() -> None:
     cd = cirq.ClassicalDataDictionaryStore()
     cd.record_measurement(mkey_m, (0, 1), two_qubits)
     assert cd.records == {mkey_m: [(0, 1)]}
@@ -52,7 +52,7 @@ def test_record_measurement():
     assert cd.measured_qubits == {mkey_m: [two_qubits]}
 
 
-def test_record_measurement_errors():
+def test_record_measurement_errors() -> None:
     cd = cirq.ClassicalDataDictionaryStore()
     with pytest.raises(ValueError, match='3 measurements but 2 qubits'):
         cd.record_measurement(mkey_m, (0, 1, 2), two_qubits)
@@ -68,14 +68,14 @@ def test_record_measurement_errors():
         cd.record_measurement(mkey_m, (1, 0), tuple(cirq.LineQid.range(2, dimension=3)))
 
 
-def test_record_channel_measurement():
+def test_record_channel_measurement() -> None:
     cd = cirq.ClassicalDataDictionaryStore()
     cd.record_channel_measurement(mkey_m, 1)
     assert cd.channel_records == {mkey_m: [1]}
     assert cd.keys() == (mkey_m,)
 
 
-def test_record_channel_measurement_errors():
+def test_record_channel_measurement_errors() -> None:
     cd = cirq.ClassicalDataDictionaryStore()
     cd.record_channel_measurement(mkey_m, 1)
     cd.record_channel_measurement(mkey_m, 1)
@@ -88,7 +88,7 @@ def test_record_channel_measurement_errors():
         cd.record_channel_measurement(mkey_m, 1)
 
 
-def test_get_int():
+def test_get_int() -> None:
     cd = cirq.ClassicalDataDictionaryStore()
     cd.record_measurement(mkey_m, (0, 1), two_qubits)
     assert cd.get_int(mkey_m) == 1
@@ -106,7 +106,7 @@ def test_get_int():
         cd.get_int(mkey_m)
 
 
-def test_copy():
+def test_copy() -> None:
     cd = cirq.ClassicalDataDictionaryStore(
         _records={mkey_m: [(0, 1)]},
         _measured_qubits={mkey_m: [two_qubits]},
@@ -129,7 +129,7 @@ def test_copy():
     assert cd1.measurement_types == cd.measurement_types
 
 
-def test_repr():
+def test_repr() -> None:
     cd = cirq.ClassicalDataDictionaryStore(
         _records={mkey_m: [(0, 1)]},
         _measured_qubits={mkey_m: [two_qubits]},

cirq/value/periodic_value_test.py

@@ -18,7 +18,7 @@ import sympy
 import cirq
 
 
-def test_periodic_value_equality():
+def test_periodic_value_equality() -> None:
     eq = cirq.testing.EqualsTester()
     eq.add_equality_group(
         cirq.PeriodicValue(1, 2),
@@ -34,7 +34,7 @@ def test_periodic_value_equality():
     eq.add_equality_group(cirq.PeriodicValue(2, 4))
 
 
-def test_periodic_value_approx_eq_basic():
+def test_periodic_value_approx_eq_basic() -> None:
     assert cirq.approx_eq(cirq.PeriodicValue(1.0, 2.0), cirq.PeriodicValue(1.0, 2.0), atol=0.1)
     assert cirq.approx_eq(cirq.PeriodicValue(1.0, 2.0), cirq.PeriodicValue(1.2, 2.0), atol=0.3)
     assert not cirq.approx_eq(cirq.PeriodicValue(1.0, 2.0), cirq.PeriodicValue(1.2, 2.0), atol=0.1)
@@ -49,12 +49,12 @@ def test_periodic_value_approx_eq_basic():
     )
 
 
-def test_periodic_value_approx_eq_normalized():
+def test_periodic_value_approx_eq_normalized() -> None:
     assert cirq.approx_eq(cirq.PeriodicValue(1.0, 3.0), cirq.PeriodicValue(4.1, 3.0), atol=0.2)
     assert cirq.approx_eq(cirq.PeriodicValue(1.0, 3.0), cirq.PeriodicValue(-2.1, 3.0), atol=0.2)
 
 
-def test_periodic_value_approx_eq_boundary():
+def test_periodic_value_approx_eq_boundary() -> None:
     assert cirq.approx_eq(cirq.PeriodicValue(0.0, 2.0), cirq.PeriodicValue(1.9, 2.0), atol=0.2)
     assert cirq.approx_eq(cirq.PeriodicValue(0.1, 2.0), cirq.PeriodicValue(1.9, 2.0), atol=0.3)
     assert cirq.approx_eq(cirq.PeriodicValue(1.9, 2.0), cirq.PeriodicValue(0.1, 2.0), atol=0.3)
@@ -64,7 +64,7 @@ def test_periodic_value_approx_eq_boundary():
     assert cirq.approx_eq(cirq.PeriodicValue(0.4, 1.0), cirq.PeriodicValue(0.6, 1.0), atol=0.3)
 
 
-def test_periodic_value_types_mismatch():
+def test_periodic_value_types_mismatch() -> None:
     assert not cirq.approx_eq(cirq.PeriodicValue(0.0, 2.0), 0.0, atol=0.2)
     assert not cirq.approx_eq(0.0, cirq.PeriodicValue(0.0, 2.0), atol=0.2)
 
@@ -79,7 +79,9 @@ def test_periodic_value_types_mismatch():
     ],
 )
 @pytest.mark.parametrize('resolve_fn', [cirq.resolve_parameters, cirq.resolve_parameters_once])
-def test_periodic_value_is_parameterized(value, is_parameterized, parameter_names, resolve_fn):
+def test_periodic_value_is_parameterized(
+    value, is_parameterized, parameter_names, resolve_fn
+) -> None:
     assert cirq.is_parameterized(value) == is_parameterized
     assert cirq.parameter_names(value) == parameter_names
     resolved = resolve_fn(value, {p: 1 for p in parameter_names})
@@ -98,5 +100,5 @@ def test_periodic_value_is_parameterized(value, is_parameterized, parameter_name
         cirq.PeriodicValue(sympy.Symbol('v'), 3),
     ],
 )
-def test_periodic_value_repr(val):
+def test_periodic_value_repr(val) -> None:
     cirq.testing.assert_equivalent_repr(val)

cirq/value/probability_test.py

@@ -18,11 +18,11 @@ import cirq
 
 
 @pytest.mark.parametrize('p', [0.0, 0.1, 0.6, 1.0])
-def test_validate_probability_valid(p):
+def test_validate_probability_valid(p) -> None:
     assert p == cirq.validate_probability(p, 'p')
 
 
 @pytest.mark.parametrize('p', [-0.1, 1.1])
-def test_validate_probability_invalid(p):
+def test_validate_probability_invalid(p) -> None:
     with pytest.raises(ValueError, match='p'):
         cirq.validate_probability(p, 'p')

cirq/value/value_equality_attr_test.py

@@ -54,7 +54,7 @@ class BasicCb(BasicC):
     pass
 
 
-def test_value_equality_basic():
+def test_value_equality_basic() -> None:
 
     # Lookup works across equivalent types.
     v = {BasicC(1): 4, BasicCa(2): 5}
@@ -69,7 +69,7 @@ def test_value_equality_basic():
     eq.add_equality_group(BasicCa(3))
 
 
-def test_value_equality_manual():
+def test_value_equality_manual() -> None:
     eq = cirq.testing.EqualsTester()
     eq.add_equality_group(MasqueradePositiveD(3), BasicD(3))
     eq.add_equality_group(MasqueradePositiveD(4), MasqueradePositiveD(4), BasicD(4))
@@ -105,7 +105,7 @@ class UnhashableCb(UnhashableC):
     pass
 
 
-def test_value_equality_unhashable():
+def test_value_equality_unhashable() -> None:
     # Not possible to use as a dictionary key.
     with pytest.raises(TypeError, match='unhashable'):
         _ = {UnhashableC(1): 4}
@@ -143,7 +143,7 @@ class DistinctCb(DistinctC):
     pass
 
 
-def test_value_equality_distinct_child_types():
+def test_value_equality_distinct_child_types() -> None:
     # Lookup is distinct across child types.
     v = {DistinctC(1): 4, DistinctCa(1): 5, DistinctCb(1): 6}
     assert v[DistinctC(1)] == 4
@@ -168,7 +168,7 @@ class ApproxE:
         return self.x
 
 
-def test_value_equality_approximate():
+def test_value_equality_approximate() -> None:
     assert cirq.approx_eq(ApproxE(0.0), ApproxE(0.0), atol=0.1)
     assert cirq.approx_eq(ApproxE(0.0), ApproxE(0.2), atol=0.3)
     assert not cirq.approx_eq(ApproxE(0.0), ApproxE(0.2), atol=0.1)
@@ -187,13 +187,13 @@ class PeriodicF:
         return self.x % self.n
 
 
-def test_value_equality_approximate_specialized():
+def test_value_equality_approximate_specialized() -> None:
     assert PeriodicF(1, 4) != PeriodicF(5, 4)
     assert cirq.approx_eq(PeriodicF(1, 4), PeriodicF(5, 4), atol=0.1)
     assert not cirq.approx_eq(PeriodicF(1, 4), PeriodicF(6, 4), atol=0.1)
 
 
-def test_value_equality_approximate_not_supported():
+def test_value_equality_approximate_not_supported() -> None:
     assert not cirq.approx_eq(BasicC(0.0), BasicC(0.1), atol=0.2)
 
 
@@ -222,7 +222,7 @@ class ApproxGb(ApproxG):
     pass
 
 
-def test_value_equality_approximate_typing():
+def test_value_equality_approximate_typing() -> None:
     assert not cirq.approx_eq(ApproxE(0.0), PeriodicF(0.0, 1.0), atol=0.1)
     assert cirq.approx_eq(ApproxEa(0.0), ApproxEb(0.0), atol=0.1)
     assert cirq.approx_eq(ApproxG(0.0), ApproxG(0.0), atol=0.1)
@@ -230,7 +230,7 @@ def test_value_equality_approximate_typing():
     assert not cirq.approx_eq(ApproxG(0.0), ApproxGb(0.0), atol=0.1)
 
 
-def test_value_equality_forgot_method():
+def test_value_equality_forgot_method() -> None:
     with pytest.raises(TypeError, match='_value_equality_values_'):
 
         @cirq.value_equality
@@ -238,7 +238,7 @@ def test_value_equality_forgot_method():
             pass
 
 
-def test_bad_manual_cls_incompatible_args():
+def test_bad_manual_cls_incompatible_args() -> None:
     with pytest.raises(ValueError, match='incompatible'):
 
         @cirq.value_equality(manual_cls=True, distinct_child_types=True)
@@ -246,7 +246,7 @@ def test_bad_manual_cls_incompatible_args():
             pass
 
 
-def test_bad_manual_cls_forgot_method():
+def test_bad_manual_cls_forgot_method() -> None:
     with pytest.raises(TypeError, match='_value_equality_values_cls_'):
 
         @cirq.value_equality(manual_cls=True)

cirq/vis/histogram_test.py

@@ -22,7 +22,7 @@ from cirq.vis import integrated_histogram
 
 @pytest.mark.usefixtures('closefigures')
 @pytest.mark.parametrize('data', [range(10), {f'key_{i}': i for i in range(10)}])
-def test_integrated_histogram(data):
+def test_integrated_histogram(data) -> None:
     ax = integrated_histogram(
         data,
         title='Test Plot',
@@ -40,7 +40,7 @@ def test_integrated_histogram(data):
 
 
 @pytest.mark.usefixtures('closefigures')
-def test_multiple_plots():
+def test_multiple_plots() -> None:
     _, ax = plt.subplots(1, 1)
     n = 53
     data = np.random.random_sample((2, n))

cirq/vis/state_histogram_test.py

@@ -24,7 +24,7 @@ from cirq.devices import GridQubit
 from cirq.vis import state_histogram
 
 
-def test_get_state_histogram():
+def test_get_state_histogram() -> None:
     simulator = cirq.Simulator()
 
     q0 = GridQubit(0, 0)
@@ -40,7 +40,7 @@ def test_get_state_histogram():
     np.testing.assert_equal(values_to_plot, expected_values)
 
 
-def test_get_state_histogram_multi_1():
+def test_get_state_histogram_multi_1() -> None:
     qubits = cirq.LineQubit.range(4)
     c = cirq.Circuit(
         cirq.X.on_each(*qubits[1:]), cirq.measure(*qubits)  # One multi-qubit measurement
@@ -51,7 +51,7 @@ def test_get_state_histogram_multi_1():
     np.testing.assert_equal(values_to_plot, expected_values)
 
 
-def test_get_state_histogram_multi_2():
+def test_get_state_histogram_multi_2() -> None:
     qubits = cirq.LineQubit.range(4)
     c = cirq.Circuit(
         cirq.X.on_each(*qubits[1:]),
@@ -65,7 +65,7 @@ def test_get_state_histogram_multi_2():
 
 
 @pytest.mark.usefixtures('closefigures')
-def test_plot_state_histogram_result():
+def test_plot_state_histogram_result() -> None:
     qubits = cirq.LineQubit.range(4)
     c = cirq.Circuit(
         cirq.X.on_each(*qubits[1:]), cirq.measure(*qubits)  # One multi-qubit measurement
@@ -83,7 +83,7 @@ def test_plot_state_histogram_result():
 
 
 @pytest.mark.usefixtures('closefigures')
-def test_plot_state_histogram_collection():
+def test_plot_state_histogram_collection() -> None:
     qubits = cirq.LineQubit.range(4)
     c = cirq.Circuit(
         cirq.X.on_each(*qubits[1:]), cirq.measure(*qubits)  # One multi-qubit measurement

cirq/vis/vis_utils_test.py

@@ -16,7 +16,7 @@ import numpy as np
 import cirq
 
 
-def test_relative_luminance():
+def test_relative_luminance() -> None:
     rl = cirq.vis.relative_luminance([100, 100, 100])
     assert np.isclose(rl, 55560.6360)
     rl = cirq.vis.relative_luminance([0, 1, 2])