cirq-core 1.6.0.dev20250508231110__py3-none-any.whl → 1.6.0.dev20250512192405__py3-none-any.whl

cirq/contrib/qasm_import/_parser_test.py

@@ -29,7 +29,7 @@ from cirq.contrib.qasm_import._parser import QasmParser
 from cirq.testing import consistent_qasm as cq
 
 
-def test_format_header_circuit():
+def test_format_header_circuit() -> None:
     parser = QasmParser()
 
     parsed_qasm = parser.parse("OPENQASM 2.0;")
@@ -39,7 +39,7 @@ def test_format_header_circuit():
     ct.assert_same_circuits(parsed_qasm.circuit, Circuit())
 
 
-def test_unsupported_format():
+def test_unsupported_format() -> None:
     qasm = "OPENQASM 2.1;"
     parser = QasmParser()
 
@@ -47,7 +47,7 @@ def test_unsupported_format():
         parser.parse(qasm)
 
 
-def test_format_header_with_quelibinc_circuit():
+def test_format_header_with_quelibinc_circuit() -> None:
     qasm = """OPENQASM 2.0;
 include "qelib1.inc";
 """
@@ -61,14 +61,14 @@ include "qelib1.inc";
 
 
 @pytest.mark.parametrize('qasm', ["include \"qelib1.inc\";", "", "qreg q[3];"])
-def test_error_not_starting_with_format(qasm: str):
+def test_error_not_starting_with_format(qasm: str) -> None:
     parser = QasmParser()
 
     with pytest.raises(QasmException, match="Missing 'OPENQASM 2.0;' statement"):
         parser.parse(qasm)
 
 
-def test_comments():
+def test_comments() -> None:
     parser = QasmParser()
 
     parsed_qasm = parser.parse(
@@ -87,7 +87,7 @@ def test_comments():
     ct.assert_same_circuits(parsed_qasm.circuit, Circuit())
 
 
-def test_multiple_qreg_declaration():
+def test_multiple_qreg_declaration() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg a_quantum_register [ 1337 ];
@@ -116,7 +116,7 @@ def test_multiple_qreg_declaration():
                """,
     ],
 )
-def test_already_defined_error(qasm: str):
+def test_already_defined_error(qasm: str) -> None:
     parser = QasmParser()
 
     with pytest.raises(QasmException, match=r"q.*already defined.* line 3"):
@@ -134,14 +134,14 @@ def test_already_defined_error(qasm: str):
                """,
     ],
 )
-def test_zero_length_register(qasm: str):
+def test_zero_length_register(qasm: str) -> None:
     parser = QasmParser()
 
     with pytest.raises(QasmException, match=".* zero-length.*'q'.*line 2"):
         parser.parse(qasm)
 
 
-def test_unexpected_end_of_file():
+def test_unexpected_end_of_file() -> None:
     qasm = """OPENQASM 2.0;
               include "qelib1.inc";
               creg
@@ -152,7 +152,7 @@ def test_unexpected_end_of_file():
         parser.parse(qasm)
 
 
-def test_multiple_creg_declaration():
+def test_multiple_creg_declaration() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      creg a_classical_register [1337];
@@ -170,7 +170,7 @@ def test_multiple_creg_declaration():
     assert parsed_qasm.cregs == {'a_classical_register': 1337, 'c': 42}
 
 
-def test_syntax_error():
+def test_syntax_error() -> None:
     qasm = """OPENQASM 2.0;
          qreg q[2] bla;
          foobar q[0];
@@ -181,7 +181,7 @@ def test_syntax_error():
         parser.parse(qasm)
 
 
-def test_CX_gate():
+def test_CX_gate() -> None:
     qasm = """OPENQASM 2.0;
      qreg q1[2];
      qreg q2[2];
@@ -215,7 +215,7 @@ def test_CX_gate():
     assert parsed_qasm.qregs == {'q1': 2, 'q2': 2}
 
 
-def test_classical_control():
+def test_classical_control() -> None:
     qasm = """OPENQASM 2.0;
         qreg q[2];
         creg a[1];
@@ -259,7 +259,7 @@ if (m_a_0==1) cx q[0],q[1];
     assert cirq.qasm(parsed_qasm.circuit) == expected_generated_qasm
 
 
-def test_classical_control_multi_bit():
+def test_classical_control_multi_bit() -> None:
     qasm = """OPENQASM 2.0;
         qreg q[2];
         creg a[2];
@@ -296,7 +296,7 @@ def test_classical_control_multi_bit():
         _ = cirq.qasm(parsed_qasm.circuit)
 
 
-def test_CX_gate_not_enough_args():
+def test_CX_gate_not_enough_args() -> None:
     qasm = """OPENQASM 2.0;
      qreg q[2];
      CX q[0];
@@ -307,7 +307,7 @@ def test_CX_gate_not_enough_args():
         parser.parse(qasm)
 
 
-def test_CX_gate_mismatched_registers():
+def test_CX_gate_mismatched_registers() -> None:
     qasm = """OPENQASM 2.0;
      qreg q1[2];
      qreg q2[3];
@@ -319,7 +319,7 @@ def test_CX_gate_mismatched_registers():
         parser.parse(qasm)
 
 
-def test_CX_gate_bounds():
+def test_CX_gate_bounds() -> None:
     qasm = """OPENQASM 2.0;
      qreg q1[2];
      qreg q2[3];
@@ -331,7 +331,7 @@ def test_CX_gate_bounds():
         parser.parse(qasm)
 
 
-def test_CX_gate_arg_overlap():
+def test_CX_gate_arg_overlap() -> None:
     qasm = """OPENQASM 2.0;
      qreg q1[2];
      qreg q2[3];
@@ -343,7 +343,7 @@ def test_CX_gate_arg_overlap():
         parser.parse(qasm)
 
 
-def test_U_gate():
+def test_U_gate() -> None:
     qasm = """
      OPENQASM 2.0;
      qreg q[2];
@@ -376,7 +376,7 @@ def test_U_gate():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_U_angles():
+def test_U_angles() -> None:
     qasm = """
     OPENQASM 2.0;
     qreg q[1];
@@ -389,7 +389,7 @@ def test_U_angles():
     )
 
 
-def test_U_gate_zero_params_error():
+def test_U_gate_zero_params_error() -> None:
     qasm = """OPENQASM 2.0;
      qreg q[2];
      U q[1];"""
@@ -400,7 +400,7 @@ def test_U_gate_zero_params_error():
         parser.parse(qasm)
 
 
-def test_U_gate_too_much_params_error():
+def test_U_gate_too_much_params_error() -> None:
     qasm = """OPENQASM 2.0;
      qreg q[2];
      U(pi, pi, pi, pi) q[1];"""
@@ -439,7 +439,7 @@ def test_U_gate_too_much_params_error():
         'atan(0.2)',
     ],
 )
-def test_expressions(expr: str):
+def test_expressions(expr: str) -> None:
     qasm = f"""OPENQASM 2.0;
      qreg q[1];
      U({expr}, 2 * pi, pi / 2.0) q[0];
@@ -463,7 +463,7 @@ def test_expressions(expr: str):
     assert parsed_qasm.qregs == {'q': 1}
 
 
-def test_unknown_function():
+def test_unknown_function() -> None:
     qasm = """OPENQASM 2.0;
      qreg q[1];
      U(nonexistent(3), 2 * pi, pi / 3.0) q[0];
@@ -491,7 +491,7 @@ single_qubit_gates = [
 
 
 @pytest.mark.parametrize('qasm_gate,cirq_gate', rotation_gates)
-def test_rotation_gates(qasm_gate: str, cirq_gate: Callable[[float], cirq.Gate]):
+def test_rotation_gates(qasm_gate: str, cirq_gate: Callable[[float], cirq.Gate]) -> None:
     qasm = f"""OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -518,7 +518,7 @@ def test_rotation_gates(qasm_gate: str, cirq_gate: Callable[[float], cirq.Gate])
 
 
 @pytest.mark.parametrize('qasm_gate', [g[0] for g in rotation_gates])
-def test_rotation_gates_wrong_number_of_args(qasm_gate: str):
+def test_rotation_gates_wrong_number_of_args(qasm_gate: str) -> None:
     qasm = f"""
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -533,7 +533,7 @@ def test_rotation_gates_wrong_number_of_args(qasm_gate: str):
 
 
 @pytest.mark.parametrize('qasm_gate', [g[0] for g in rotation_gates])
-def test_rotation_gates_zero_params_error(qasm_gate: str):
+def test_rotation_gates_zero_params_error(qasm_gate: str) -> None:
     qasm = f"""OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -546,7 +546,7 @@ def test_rotation_gates_zero_params_error(qasm_gate: str):
         parser.parse(qasm)
 
 
-def test_qelib_gate_without_include_statement():
+def test_qelib_gate_without_include_statement() -> None:
     qasm = """OPENQASM 2.0;
          qreg q[2];
          x q[0];
@@ -557,7 +557,7 @@ def test_qelib_gate_without_include_statement():
         parser.parse(qasm)
 
 
-def test_undefined_register_from_qubit_arg():
+def test_undefined_register_from_qubit_arg() -> None:
     qasm = """OPENQASM 2.0;
             qreg q[2];
             CX q[0], q2[1];
@@ -568,7 +568,7 @@ def test_undefined_register_from_qubit_arg():
         parser.parse(qasm)
 
 
-def test_undefined_register_from_register_arg():
+def test_undefined_register_from_register_arg() -> None:
     qasm = """OPENQASM 2.0;
             qreg q[2];
             qreg q2[2];
@@ -580,7 +580,7 @@ def test_undefined_register_from_register_arg():
         parser.parse(qasm)
 
 
-def test_measure_individual_bits():
+def test_measure_individual_bits() -> None:
     qasm = """
          OPENQASM 2.0;
          include "qelib1.inc";
@@ -609,7 +609,7 @@ def test_measure_individual_bits():
     assert parsed_qasm.cregs == {'c1': 2}
 
 
-def test_measure_registers():
+def test_measure_registers() -> None:
     qasm = """OPENQASM 2.0;
          include "qelib1.inc";
          qreg q1[3];
@@ -638,7 +638,7 @@ def test_measure_registers():
     assert parsed_qasm.cregs == {'c1': 3}
 
 
-def test_measure_mismatched_register_size():
+def test_measure_mismatched_register_size() -> None:
     qasm = """OPENQASM 2.0;
          include "qelib1.inc";
          qreg q1[2];
@@ -652,7 +652,7 @@ def test_measure_mismatched_register_size():
         parser.parse(qasm)
 
 
-def test_measure_to_quantum_register():
+def test_measure_to_quantum_register() -> None:
     qasm = """OPENQASM 2.0;
          include "qelib1.inc";
          qreg q1[3];
@@ -667,7 +667,7 @@ def test_measure_to_quantum_register():
         parser.parse(qasm)
 
 
-def test_measure_undefined_classical_bit():
+def test_measure_undefined_classical_bit() -> None:
     qasm = """OPENQASM 2.0;
          include "qelib1.inc";
          qreg q1[3];
@@ -681,7 +681,7 @@ def test_measure_undefined_classical_bit():
         parser.parse(qasm)
 
 
-def test_measure_from_classical_register():
+def test_measure_from_classical_register() -> None:
     qasm = """OPENQASM 2.0;
          include "qelib1.inc";
          qreg q1[2];
@@ -696,7 +696,7 @@ def test_measure_from_classical_register():
         parser.parse(qasm)
 
 
-def test_measurement_bounds():
+def test_measurement_bounds() -> None:
     qasm = """OPENQASM 2.0;
      qreg q1[3];
      creg c1[3];
@@ -708,7 +708,7 @@ def test_measurement_bounds():
         parser.parse(qasm)
 
 
-def test_reset():
+def test_reset() -> None:
     qasm = textwrap.dedent(
         """\
         OPENQASM 2.0;
@@ -737,7 +737,7 @@ def test_reset():
     assert parsed_qasm.cregs == {'c': 1}
 
 
-def test_u1_gate():
+def test_u1_gate() -> None:
     qasm = """
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -760,7 +760,7 @@ def test_u1_gate():
     assert parsed_qasm.qregs == {'q': 1}
 
 
-def test_u2_gate():
+def test_u2_gate() -> None:
     qasm = """
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -783,7 +783,7 @@ def test_u2_gate():
     assert parsed_qasm.qregs == {'q': 1}
 
 
-def test_id_gate():
+def test_id_gate() -> None:
     qasm = """
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -808,7 +808,7 @@ def test_id_gate():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_u3_gate():
+def test_u3_gate() -> None:
     qasm = """
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -842,7 +842,7 @@ def test_u3_gate():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_r_gate():
+def test_r_gate() -> None:
     qasm = """
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -869,7 +869,7 @@ def test_r_gate():
     'qasm_gate',
     ['id', 'u2', 'u3', 'r'] + [g[0] for g in rotation_gates] + [g[0] for g in single_qubit_gates],
 )
-def test_standard_single_qubit_gates_wrong_number_of_args(qasm_gate):
+def test_standard_single_qubit_gates_wrong_number_of_args(qasm_gate) -> None:
     qasm = f"""
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -888,7 +888,7 @@ def test_standard_single_qubit_gates_wrong_number_of_args(qasm_gate):
     [['id', 0], ['u2', 2], ['u3', 3], ['rx', 1], ['ry', 1], ['rz', 1], ['r', 2]]
     + [[g[0], 0] for g in single_qubit_gates],
 )
-def test_standard_gates_wrong_params_error(qasm_gate: str, num_params: int):
+def test_standard_gates_wrong_params_error(qasm_gate: str, num_params: int) -> None:
     qasm = f"""OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -934,7 +934,7 @@ two_qubit_param_gates = {
 
 
 @pytest.mark.parametrize('qasm_gate,cirq_gate', two_qubit_gates)
-def test_two_qubit_gates(qasm_gate: str, cirq_gate: cirq.testing.TwoQubitGate):
+def test_two_qubit_gates(qasm_gate: str, cirq_gate: cirq.testing.TwoQubitGate) -> None:
     qasm = f"""
     OPENQASM 2.0;
     include "qelib1.inc";
@@ -1015,7 +1015,7 @@ def test_two_qubit_param_gates(
 @pytest.mark.parametrize(
     'qasm_gate', [g[0] for g in two_qubit_gates] + [g[0] for g in two_qubit_param_gates.keys()]
 )
-def test_two_qubit_gates_not_enough_qubits(qasm_gate: str):
+def test_two_qubit_gates_not_enough_qubits(qasm_gate: str) -> None:
     if qasm_gate in ('cu1', 'crz'):
         qasm = f"""
         OPENQASM 2.0;
@@ -1045,7 +1045,7 @@ def test_two_qubit_gates_not_enough_qubits(qasm_gate: str):
 
 
 @pytest.mark.parametrize('qasm_gate', [g[0] for g in two_qubit_param_gates.keys()])
-def test_two_qubit_gates_not_enough_args(qasm_gate: str):
+def test_two_qubit_gates_not_enough_args(qasm_gate: str) -> None:
     qasm = f"""
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -1062,7 +1062,7 @@ def test_two_qubit_gates_not_enough_args(qasm_gate: str):
 @pytest.mark.parametrize(
     'qasm_gate', [g[0] for g in two_qubit_gates] + [g[0] for g in two_qubit_param_gates.keys()]
 )
-def test_two_qubit_gates_with_too_much_parameters(qasm_gate: str):
+def test_two_qubit_gates_with_too_much_parameters(qasm_gate: str) -> None:
     if qasm_gate in ('cu1', 'cu3', 'crz'):
         num_params_needed = 3 if qasm_gate == 'cu3' else 1
     else:
@@ -1088,7 +1088,7 @@ three_qubit_gates = [('ccx', cirq.TOFFOLI), ('cswap', cirq.CSWAP)]
 
 
 @pytest.mark.parametrize('qasm_gate,cirq_gate', three_qubit_gates)
-def test_three_qubit_gates(qasm_gate: str, cirq_gate: cirq.testing.TwoQubitGate):
+def test_three_qubit_gates(qasm_gate: str, cirq_gate: cirq.testing.TwoQubitGate) -> None:
     qasm = f"""
      OPENQASM 2.0;
      include "qelib1.inc";
@@ -1128,7 +1128,7 @@ def test_three_qubit_gates(qasm_gate: str, cirq_gate: cirq.testing.TwoQubitGate)
 
 
 @pytest.mark.parametrize('qasm_gate', [g[0] for g in three_qubit_gates])
-def test_three_qubit_gates_not_enough_args(qasm_gate: str):
+def test_three_qubit_gates_not_enough_args(qasm_gate: str) -> None:
     qasm = f"""OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -1142,7 +1142,7 @@ def test_three_qubit_gates_not_enough_args(qasm_gate: str):
 
 
 @pytest.mark.parametrize('qasm_gate', [g[0] for g in three_qubit_gates])
-def test_three_qubit_gates_with_too_much_parameters(qasm_gate: str):
+def test_three_qubit_gates_with_too_much_parameters(qasm_gate: str) -> None:
     qasm = f"""OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[3];
@@ -1156,7 +1156,7 @@ def test_three_qubit_gates_with_too_much_parameters(qasm_gate: str):
 
 
 @pytest.mark.parametrize('qasm_gate,cirq_gate', single_qubit_gates)
-def test_single_qubit_gates(qasm_gate: str, cirq_gate: cirq.Gate):
+def test_single_qubit_gates(qasm_gate: str, cirq_gate: cirq.Gate) -> None:
     qasm = f"""OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -1180,7 +1180,7 @@ def test_single_qubit_gates(qasm_gate: str, cirq_gate: cirq.Gate):
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_openqasm_3_0_qubits():
+def test_openqasm_3_0_qubits() -> None:
     qasm = """OPENQASM 3.0;
      include "stdgates.inc";
      qubit[2] q;
@@ -1205,7 +1205,7 @@ def test_openqasm_3_0_qubits():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_openqasm_3_0_scalar_qubit():
+def test_openqasm_3_0_scalar_qubit() -> None:
     qasm = """OPENQASM 3.0;
      include "stdgates.inc";
      qubit q;
@@ -1230,7 +1230,7 @@ def test_openqasm_3_0_scalar_qubit():
     assert parsed_qasm.qregs == {'q': 1}
 
 
-def test_custom_gate():
+def test_custom_gate() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -1271,7 +1271,7 @@ def test_custom_gate():
     cq.assert_qiskit_parsed_qasm_consistent_with_unitary(qasm, cirq.unitary(unrolled))
 
 
-def test_custom_gate_parameterized():
+def test_custom_gate_parameterized() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -1320,7 +1320,7 @@ def test_custom_gate_parameterized():
     cq.assert_qiskit_parsed_qasm_consistent_with_unitary(qasm, cirq.unitary(unrolled))
 
 
-def test_custom_gate_broadcast():
+def test_custom_gate_broadcast() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[3];
@@ -1370,7 +1370,7 @@ def test_custom_gate_broadcast():
     cq.assert_qiskit_parsed_qasm_consistent_with_unitary(qasm, cirq.unitary(unrolled))
 
 
-def test_custom_gate_undefined_qubit_error():
+def test_custom_gate_undefined_qubit_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];
@@ -1384,7 +1384,7 @@ def test_custom_gate_undefined_qubit_error():
     )
 
 
-def test_custom_gate_qubit_scope_closure_error():
+def test_custom_gate_qubit_scope_closure_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];
@@ -1398,7 +1398,7 @@ def test_custom_gate_qubit_scope_closure_error():
     )
 
 
-def test_custom_gate_qubit_index_error():
+def test_custom_gate_qubit_index_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];
@@ -1412,7 +1412,7 @@ def test_custom_gate_qubit_index_error():
     )
 
 
-def test_custom_gate_qreg_count_error():
+def test_custom_gate_qreg_count_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[2];
@@ -1426,7 +1426,7 @@ def test_custom_gate_qreg_count_error():
     )
 
 
-def test_custom_gate_missing_param_error():
+def test_custom_gate_missing_param_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];
@@ -1440,7 +1440,7 @@ def test_custom_gate_missing_param_error():
     )
 
 
-def test_custom_gate_extra_param_error():
+def test_custom_gate_extra_param_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];
@@ -1454,7 +1454,7 @@ def test_custom_gate_extra_param_error():
     )
 
 
-def test_custom_gate_undefined_param_error():
+def test_custom_gate_undefined_param_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];
@@ -1468,7 +1468,7 @@ def test_custom_gate_undefined_param_error():
     )
 
 
-def test_top_level_param_error():
+def test_top_level_param_error() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];
@@ -1495,7 +1495,7 @@ def _test_parse_exception(qasm: str, cirq_err: str, qiskit_err: str | None):
         qiskit.QuantumCircuit.from_qasm_str(qasm)
 
 
-def test_nested_custom_gate_has_keyword_in_name():
+def test_nested_custom_gate_has_keyword_in_name() -> None:
     qasm = """OPENQASM 2.0;
      include "qelib1.inc";
      qreg q[1];

cirq/contrib/qcircuit/qcircuit_pdf_test.py

@@ -21,7 +21,7 @@ import cirq.contrib.qcircuit.qcircuit_pdf as qcircuit_pdf
 
 
 @mock.patch.object(pylatex.Document, "generate_pdf")
-def test_qcircuit_pdf(mock_generate_pdf):
+def test_qcircuit_pdf(mock_generate_pdf) -> None:
     circuit = cirq.Circuit(cirq.X(cirq.q(0)), cirq.CZ(cirq.q(0), cirq.q(1)))
     qcircuit_pdf.circuit_to_pdf_using_qcircuit_via_tex(circuit, "/tmp/test_file")
     mock_generate_pdf.assert_called_once_with(

cirq/contrib/qcircuit/qcircuit_test.py

@@ -43,7 +43,7 @@ def assert_has_qcircuit_diagram(actual: cirq.Circuit, desired: str, **kwargs) ->
     )
 
 
-def test_fallback_diagram():
+def test_fallback_diagram() -> None:
     class MagicGate(cirq.testing.ThreeQubitGate):
         def __str__(self):
             return 'MagicGate'
@@ -77,7 +77,7 @@ def test_fallback_diagram():
     assert_has_qcircuit_diagram(circuit, expected_diagram)
 
 
-def test_teleportation_diagram():
+def test_teleportation_diagram() -> None:
     ali = cirq.NamedQubit('alice')
     car = cirq.NamedQubit('carrier')
     bob = cirq.NamedQubit('bob')
@@ -106,7 +106,7 @@ def test_teleportation_diagram():
     )
 
 
-def test_other_diagram():
+def test_other_diagram() -> None:
     a, b, c = cirq.LineQubit.range(3)
 
     circuit = cirq.Circuit(cirq.X(a), cirq.Y(b), cirq.Z(c))
@@ -122,7 +122,7 @@ def test_other_diagram():
     assert_has_qcircuit_diagram(circuit, expected_diagram)
 
 
-def test_qcircuit_qubit_namer():
+def test_qcircuit_qubit_namer() -> None:
     from cirq.contrib.qcircuit import qcircuit_diagram
 
     assert qcircuit_diagram.qcircuit_qubit_namer(cirq.NamedQubit('q')) == r'\lstick{\text{q}}&'
@@ -137,7 +137,7 @@ def test_qcircuit_qubit_namer():
     )
 
 
-def test_two_cx_diagram():
+def test_two_cx_diagram() -> None:
     # test for no moment indication
     q0, q1, q2, q3 = cirq.LineQubit.range(4)
     circuit = cirq.Circuit(cirq.CX(q0, q2), cirq.CX(q1, q3), cirq.CX(q0, q2), cirq.CX(q1, q3))
@@ -153,7 +153,7 @@ def test_two_cx_diagram():
     assert_has_qcircuit_diagram(circuit, expected_diagram)
 
 
-def test_sqrt_iswap_diagram():
+def test_sqrt_iswap_diagram() -> None:
     # test for proper rendering of ISWAP^{0.5}
     q0, q1 = cirq.LineQubit.range(2)
     circuit = cirq.Circuit(cirq.ISWAP(q0, q1) ** 0.5)

cirq/contrib/routing/greedy_test.py

@@ -21,7 +21,7 @@ import cirq.contrib.routing as ccr
 from cirq.contrib.routing.greedy import route_circuit_greedily
 
 
-def test_bad_args():
+def test_bad_args() -> None:
     """Test zero valued arguments in greedy router."""
     circuit = cirq.testing.random_circuit(4, 2, 0.5, random_state=5)
     device_graph = ccr.get_grid_device_graph(3, 2)
@@ -50,7 +50,7 @@ def create_hanging_routing_instance(circuit, device_graph):
     )
 
 
-def test_router_hanging():
+def test_router_hanging() -> None:
     """Run a separate process and check if greedy router hits timeout (20s)."""
     circuit, device_graph = create_circuit_and_device()
     process = Process(target=create_hanging_routing_instance, args=[circuit, device_graph])

cirq/contrib/routing/initialization_test.py

@@ -30,7 +30,7 @@ def get_seeded_initial_mapping(graph_seed, init_seed):
 
 
 @pytest.mark.parametrize('seed', [random.randint(0, 2**32) for _ in range(10)])
-def test_initialization_reproducible_with_seed(seed):
+def test_initialization_reproducible_with_seed(seed) -> None:
     wrappers = (lambda s: s, np.random.RandomState)
     mappings = [
         get_seeded_initial_mapping(seed, wrapper(seed)) for wrapper in wrappers for _ in range(5)
@@ -39,7 +39,7 @@ def test_initialization_reproducible_with_seed(seed):
     eq.add_equality_group(*mappings)
 
 
-def test_initialization_with_no_seed():
+def test_initialization_with_no_seed() -> None:
     graph_seed = random.randint(0, 2**32)
     state = np.random.get_state()
     mappings = []
@@ -50,7 +50,7 @@ def test_initialization_with_no_seed():
     eq.add_equality_group(*mappings)
 
 
-def test_initialization_reproducible_between_runs():
+def test_initialization_reproducible_between_runs() -> None:
     seed = 45
     logical_graph = nx.erdos_renyi_graph(6, 0.5, seed=seed)
     logical_graph = nx.relabel_nodes(logical_graph, cirq.LineQubit)