cirq-core 1.6.0.dev20250529194600__py3-none-any.whl → 1.6.0.dev20250530203656__py3-none-any.whl

cirq/circuits/qasm_output_test.py

@@ -29,12 +29,12 @@ def _make_qubits(n):
     return [cirq.NamedQubit(f'q{i}') for i in range(n)]
 
 
-def test_u_gate_repr():
+def test_u_gate_repr() -> None:
     gate = QasmUGate(0.1, 0.2, 0.3)
     assert repr(gate) == 'cirq.circuits.qasm_output.QasmUGate(theta=0.1, phi=0.2, lmda=0.3)'
 
 
-def test_u_gate_eq():
+def test_u_gate_eq() -> None:
     gate = QasmUGate(0.1, 0.2, 0.3)
     gate2 = QasmUGate(0.1, 0.2, 0.3)
     cirq.approx_eq(gate, gate2, atol=1e-16)
@@ -43,13 +43,13 @@ def test_u_gate_eq():
     cirq.approx_eq(gate4, gate3, atol=1e-16)
 
 
-def test_qasm_two_qubit_gate_repr():
+def test_qasm_two_qubit_gate_repr() -> None:
     cirq.testing.assert_equivalent_repr(
         QasmTwoQubitGate.from_matrix(cirq.testing.random_unitary(4))
     )
 
 
-def test_qasm_u_qubit_gate_unitary():
+def test_qasm_u_qubit_gate_unitary() -> None:
     u = cirq.testing.random_unitary(2)
     g = QasmUGate.from_matrix(u)
     cirq.testing.assert_allclose_up_to_global_phase(cirq.unitary(g), u, atol=1e-7)
@@ -62,13 +62,13 @@ def test_qasm_u_qubit_gate_unitary():
     cirq.testing.assert_implements_consistent_protocols(g)
 
 
-def test_qasm_two_qubit_gate_unitary():
+def test_qasm_two_qubit_gate_unitary() -> None:
     u = cirq.testing.random_unitary(4)
     g = QasmTwoQubitGate.from_matrix(u)
     np.testing.assert_allclose(cirq.unitary(g), u)
 
 
-def test_empty_circuit_one_qubit():
+def test_empty_circuit_one_qubit() -> None:
     (q0,) = _make_qubits(1)
     output = cirq.QasmOutput((), (q0,))
     assert (
@@ -83,7 +83,7 @@ qreg q[1];
     )
 
 
-def test_empty_circuit_no_qubits():
+def test_empty_circuit_no_qubits() -> None:
     output = cirq.QasmOutput((), ())
     assert (
         str(output)
@@ -96,7 +96,7 @@ include "qelib1.inc";
     )
 
 
-def test_header():
+def test_header() -> None:
     (q0,) = _make_qubits(1)
     output = cirq.QasmOutput(
         (),
@@ -143,7 +143,7 @@ qreg q[1];
     )
 
 
-def test_single_gate_no_parameter():
+def test_single_gate_no_parameter() -> None:
     (q0,) = _make_qubits(1)
     output = cirq.QasmOutput((cirq.X(q0),), (q0,))
     assert (
@@ -161,7 +161,7 @@ x q[0];
     )
 
 
-def test_single_gate_with_parameter():
+def test_single_gate_with_parameter() -> None:
     (q0,) = _make_qubits(1)
     output = cirq.QasmOutput((cirq.X(q0) ** 0.25,), (q0,))
     assert (
@@ -179,7 +179,7 @@ rx(pi*0.25) q[0];
     )
 
 
-def test_h_gate_with_parameter():
+def test_h_gate_with_parameter() -> None:
     (q0,) = _make_qubits(1)
     output = cirq.QasmOutput((cirq.H(q0) ** 0.25,), (q0,))
     assert (
@@ -200,7 +200,7 @@ ry(pi*-0.25) q[0];
     )
 
 
-def test_qasm_global_pahse():
+def test_qasm_global_pahse() -> None:
     output = cirq.QasmOutput((cirq.global_phase_operation(np.exp(1j * 5))), ())
     assert (
         str(output)
@@ -213,7 +213,7 @@ include "qelib1.inc";
     )
 
 
-def test_precision():
+def test_precision() -> None:
     (q0,) = _make_qubits(1)
     output = cirq.QasmOutput((cirq.X(q0) ** 0.1234567,), (q0,), precision=3)
     assert (
@@ -231,14 +231,14 @@ rx(pi*0.123) q[0];
     )
 
 
-def test_version():
+def test_version() -> None:
     (q0,) = _make_qubits(1)
     with pytest.raises(ValueError):
         output = cirq.QasmOutput((), (q0,), version='4.0')
         _ = str(output)
 
 
-def test_save_to_file(tmpdir):
+def test_save_to_file(tmpdir) -> None:
     file_path = os.path.join(tmpdir, 'test.qasm')
     (q0,) = _make_qubits(1)
     output = cirq.QasmOutput((), (q0,))
@@ -257,7 +257,7 @@ qreg q[1];
     )
 
 
-def test_unsupported_operation():
+def test_unsupported_operation() -> None:
     (q0,) = _make_qubits(1)
 
     class UnsupportedOperation(cirq.Operation):
@@ -269,7 +269,7 @@ def test_unsupported_operation():
         _ = str(output)
 
 
-def _all_operations(q0, q1, q2, q3, q4, include_measurements=True):
+def _all_operations(q0, q1, q2, q3, q4, *, include_measurements=True):
     class ExampleOperation(cirq.Operation):
         qubits = (q0,)
         with_qubits = NotImplemented
@@ -357,7 +357,7 @@ def _all_operations(q0, q1, q2, q3, q4, include_measurements=True):
     )
 
 
-def test_output_unitary_same_as_qiskit():
+def test_output_unitary_same_as_qiskit() -> None:
     qubits = tuple(_make_qubits(5))
     operations = _all_operations(*qubits, include_measurements=False)
     output = cirq.QasmOutput(operations, qubits, header='Generated from Cirq', precision=10)
@@ -368,7 +368,7 @@ def test_output_unitary_same_as_qiskit():
     cq.assert_qiskit_parsed_qasm_consistent_with_unitary(text, cirq_unitary)
 
 
-def test_fails_on_big_unknowns():
+def test_fails_on_big_unknowns() -> None:
     class UnrecognizedGate(cirq.testing.ThreeQubitGate):
         pass
 
@@ -377,7 +377,7 @@ def test_fails_on_big_unknowns():
         _ = c.to_qasm()
 
 
-def test_output_format():
+def test_output_format() -> None:
     def filter_unpredictable_numbers(text):
         return re.sub(r'u3\(.+\)', r'u3(<not-repeatable>)', text)
 
@@ -571,7 +571,7 @@ x q[0];
     )
 
 
-def test_reset():
+def test_reset() -> None:
     a, b = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.H(a), cirq.CNOT(a, b), cirq.reset(a), cirq.reset(b))
     output = cirq.QasmOutput(
@@ -601,7 +601,7 @@ reset q[1];
     )
 
 
-def test_different_sized_registers():
+def test_different_sized_registers() -> None:
     qubits = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.measure(qubits[0], key='c'), cirq.measure(qubits, key='c'))
     output = cirq.QasmOutput(

cirq/circuits/text_diagram_drawer_test.py

@@ -59,7 +59,7 @@ def assert_has_rendering(actual: TextDiagramDrawer, desired: str, **kwargs) -> N
     )
 
 
-def test_draw_entries_and_lines_with_options():
+def test_draw_entries_and_lines_with_options() -> None:
     d = TextDiagramDrawer()
     d.write(0, 0, '!')
     d.write(6, 2, 'span')
@@ -132,7 +132,7 @@ def test_draw_entries_and_lines_with_options():
     )
 
 
-def test_draw_entries_and_lines_with_emphasize():
+def test_draw_entries_and_lines_with_emphasize() -> None:
     d = TextDiagramDrawer()
     d.write(0, 0, '!')
     d.write(6, 2, 'span')
@@ -161,27 +161,27 @@ def test_draw_entries_and_lines_with_emphasize():
     )
 
 
-def test_line_detects_horizontal():
+def test_line_detects_horizontal() -> None:
     d = TextDiagramDrawer()
     with mock.patch.object(d, 'vertical_line') as vertical_line:
         d.grid_line(1, 2, 1, 5, True)
         vertical_line.assert_called_once_with(1, 2, 5, True, False)
 
 
-def test_line_detects_vertical():
+def test_line_detects_vertical() -> None:
     d = TextDiagramDrawer()
     with mock.patch.object(d, 'horizontal_line') as horizontal_line:
         d.grid_line(2, 1, 5, 1, True)
         horizontal_line.assert_called_once_with(1, 2, 5, True, False)
 
 
-def test_line_fails_when_not_aligned():
+def test_line_fails_when_not_aligned() -> None:
     d = TextDiagramDrawer()
     with pytest.raises(ValueError):
         d.grid_line(1, 2, 3, 4)
 
 
-def test_multiline_entries():
+def test_multiline_entries() -> None:
     d = TextDiagramDrawer()
     d.write(0, 0, 'hello\nthere')
     d.write(0, 1, 'next')
@@ -225,7 +225,7 @@ short     │ │         │
     )
 
 
-def test_drawer_copy():
+def test_drawer_copy() -> None:
     orig_entries = {(0, 0): _DiagramText('entry', '')}
     orig_vertical_lines = [_VerticalLine(1, 1, 3, True, False)]
     orig_horizontal_lines = [_HorizontalLine(0, 0, 3, False, False)]
@@ -238,9 +238,9 @@ def test_drawer_copy():
         'vertical_padding': orig_vertical_padding,
         'horizontal_padding': orig_horizontal_padding,
     }
-    orig_drawer = TextDiagramDrawer(**kwargs)
+    orig_drawer = TextDiagramDrawer(**kwargs)  # type: ignore[arg-type]
 
-    same_drawer = TextDiagramDrawer(**kwargs)
+    same_drawer = TextDiagramDrawer(**kwargs)  # type: ignore[arg-type]
     assert orig_drawer == same_drawer
 
     copy_drawer = orig_drawer.copy()
@@ -266,7 +266,7 @@ def test_drawer_copy():
     assert copy_drawer != orig_drawer
 
 
-def test_drawer_stack():
+def test_drawer_stack() -> None:
     d = TextDiagramDrawer()
     d.write(0, 0, 'A')
     d.write(1, 0, 'B')
@@ -336,7 +336,7 @@ AB D
     with pytest.raises(ValueError):
         TextDiagramDrawer.vstack((d, dd))
 
-    vstacked = TextDiagramDrawer.vstack((dd, d), padding_resolver=max)
+    vstacked = TextDiagramDrawer.vstack((dd, d), padding_resolver=max)  # type: ignore[arg-type]
     expected = """
 D
 
@@ -350,7 +350,7 @@ A  B
     """.strip()
     assert_has_rendering(vstacked, expected)
 
-    hstacked = TextDiagramDrawer.hstack((d, dd), padding_resolver=max)
+    hstacked = TextDiagramDrawer.hstack((d, dd), padding_resolver=max)  # type: ignore[arg-type]
     expected = """
 AB D
 
@@ -360,7 +360,7 @@ AB D
     """.strip()
     assert_has_rendering(hstacked, expected)
 
-    vstacked_min = TextDiagramDrawer.vstack((dd, d), padding_resolver=min)
+    vstacked_min = TextDiagramDrawer.vstack((dd, d), padding_resolver=min)  # type: ignore[arg-type]
     expected = """
 D
 
@@ -374,7 +374,7 @@ AB
     """.strip()
     assert_has_rendering(vstacked_min, expected)
 
-    hstacked_min = TextDiagramDrawer.hstack((d, dd), padding_resolver=min)
+    hstacked_min = TextDiagramDrawer.hstack((d, dd), padding_resolver=min)  # type: ignore[arg-type]
     expected = """
 AB D
 
@@ -383,7 +383,7 @@ AB D
     assert_has_rendering(hstacked_min, expected)
 
 
-def test_drawer_eq():
+def test_drawer_eq() -> None:
     assert TextDiagramDrawer().__eq__(23) == NotImplemented
 
     eq = ct.EqualsTester()
@@ -407,7 +407,7 @@ def test_drawer_eq():
     eq.add_equality_group(dd)
 
 
-def test_drawer_superimposed():
+def test_drawer_superimposed() -> None:
     empty_drawer = TextDiagramDrawer()
     assert not empty_drawer
     drawer_with_something = TextDiagramDrawer()
@@ -418,7 +418,7 @@ def test_drawer_superimposed():
     assert not empty_drawer
 
 
-def test_pick_charset():
+def test_pick_charset() -> None:
     assert pick_charset(use_unicode=False, emphasize=False, doubled=False) == ASCII_BOX_CHARS
     assert pick_charset(use_unicode=False, emphasize=False, doubled=True) == ASCII_BOX_CHARS
     assert pick_charset(use_unicode=False, emphasize=True, doubled=False) == ASCII_BOX_CHARS

cirq/contrib/custom_simulators/custom_state_simulator_test.py

@@ -64,26 +64,27 @@ def create_test_circuit():
     )
 
 
-def test_basis_state_simulator():
+def test_basis_state_simulator() -> None:
     sim = CustomStateSimulator(ComputationalBasisSimState)
     circuit = create_test_circuit()
     r = sim.simulate(circuit)
     assert r.measurements == {'a': np.array([1]), 'b': np.array([2])}
-    assert r._final_simulator_state._state.basis == [2, 2]
+    assert r._final_simulator_state._state.basis == [2, 2]  # type: ignore[attr-defined]
 
 
-def test_built_in_states():
+def test_built_in_states() -> None:
     # Verify this works for the built-in states too, you just lose the custom step/trial results.
     sim = CustomStateSimulator(cirq.StateVectorSimulationState)
     circuit = create_test_circuit()
     r = sim.simulate(circuit)
     assert r.measurements == {'a': np.array([1]), 'b': np.array([2])}
     assert np.allclose(
-        r._final_simulator_state._state._state_vector, [[0, 0, 0], [0, 0, 0], [0, 0, 1]]
+        r._final_simulator_state._state._state_vector,  # type: ignore[attr-defined]
+        [[0, 0, 0], [0, 0, 0], [0, 0, 1]],
     )
 
 
-def test_product_state_mode_built_in_state():
+def test_product_state_mode_built_in_state() -> None:
     sim = CustomStateSimulator(cirq.StateVectorSimulationState, split_untangled_states=True)
     circuit = create_test_circuit()
     r = sim.simulate(circuit)
@@ -99,16 +100,16 @@ def test_product_state_mode_built_in_state():
     )
 
 
-def test_noise():
+def test_noise() -> None:
     x = cirq.XPowGate(dimension=3)
     sim = CustomStateSimulator(ComputationalBasisSimState, noise=x**2)
     circuit = create_test_circuit()
     r = sim.simulate(circuit)
     assert r.measurements == {'a': np.array([2]), 'b': np.array([2])}
-    assert r._final_simulator_state._state.basis == [1, 2]
+    assert r._final_simulator_state._state.basis == [1, 2]  # type: ignore[attr-defined]
 
 
-def test_run():
+def test_run() -> None:
     sim = CustomStateSimulator(ComputationalBasisSimState)
     circuit = create_test_circuit()
     r = sim.run(circuit)
@@ -116,7 +117,7 @@ def test_run():
     assert np.allclose(r.records['b'], np.array([[1], [2]]))
 
 
-def test_parameterized_repetitions():
+def test_parameterized_repetitions() -> None:
     q = cirq.LineQid(0, dimension=5)
     x = cirq.XPowGate(dimension=5)
     circuit = cirq.Circuit(
@@ -182,7 +183,7 @@ class ComputationalBasisSimProductState(cirq.SimulationState[ComputationalBasisP
             return True
 
 
-def test_product_state_mode():
+def test_product_state_mode() -> None:
     sim = CustomStateSimulator(ComputationalBasisSimProductState, split_untangled_states=True)
     circuit = create_test_circuit()
     r = sim.simulate(circuit)

cirq/contrib/qasm_import/_lexer_test.py

@@ -20,14 +20,14 @@ from cirq.contrib.qasm_import import QasmException
 from cirq.contrib.qasm_import._lexer import QasmLexer
 
 
-def test_empty_circuit():
+def test_empty_circuit() -> None:
     lexer = QasmLexer()
     lexer.input("")
     assert lexer.token() is None
 
 
 @pytest.mark.parametrize('number', ["00000", "03", "3", "0045", "21"])
-def test_natural_numbers(number: str):
+def test_natural_numbers(number: str) -> None:
     lexer = QasmLexer()
     lexer.input(number)
     token = lexer.token()
@@ -36,7 +36,7 @@ def test_natural_numbers(number: str):
     assert token.value == int(number)
 
 
-def test_supported_format():
+def test_supported_format() -> None:
     lexer = QasmLexer()
     lexer.input("OPENQASM 2.0;")
     token = lexer.token()
@@ -45,7 +45,7 @@ def test_supported_format():
     assert token.value == '2.0'
 
 
-def test_qelib_inc():
+def test_qelib_inc() -> None:
     lexer = QasmLexer()
     lexer.input('include "qelib1.inc";')
     token = lexer.token()
@@ -54,26 +54,31 @@ def test_qelib_inc():
     assert token.value == 'include "qelib1.inc";'
 
 
-def test_measurement():
+def test_measurement() -> None:
     lexer = QasmLexer()
     lexer.input("measure q -> c;")
     token = lexer.token()
+    assert token is not None
     assert token.type == "MEASURE"
     assert token.value == 'measure'
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == 'q'
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ARROW"
     assert token.value == '->'
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == 'c'
 
     token = lexer.token()
+    assert token is not None
     assert token.type == ";"
     assert token.value == ';'
 
@@ -81,7 +86,7 @@ def test_measurement():
 @pytest.mark.parametrize(
     'identifier', ['b', 'CX', 'abc', 'aXY03', 'a_valid_name_with_02_digits_and_underscores']
 )
-def test_valid_ids(identifier: str):
+def test_valid_ids(identifier: str) -> None:
     lexer = QasmLexer()
     lexer.input(identifier)
     token = lexer.token()
@@ -95,7 +100,7 @@ def test_valid_ids(identifier: str):
     'number',
     ['1e2', '1e0', '3.', '4.e10', '.333', '1.0', '0.1', '2.0e-05', '1.2E+05', '123123.2132312'],
 )
-def test_numbers(number: str):
+def test_numbers(number: str) -> None:
     lexer = QasmLexer()
     lexer.input(number)
     token = lexer.token()
@@ -106,146 +111,175 @@ def test_numbers(number: str):
 
 
 @pytest.mark.parametrize('token', QasmLexer.reserved.keys())
-def test_keywords(token):
+def test_keywords(token) -> None:
     lexer = QasmLexer()
     identifier = f'{token} {token}'
     lexer.input(identifier)
     t = lexer.token()
+    assert t is not None
     assert t.type == QasmLexer.reserved[token]
     assert t.value == token
     t2 = lexer.token()
+    assert t2 is not None
     assert t2.type == QasmLexer.reserved[token]
     assert t2.value == token
 
 
 @pytest.mark.parametrize('token', QasmLexer.reserved.keys())
 @pytest.mark.parametrize('separator', ['', '_'])
-def test_identifier_starts_or_ends_with_keyword(token, separator):
+def test_identifier_starts_or_ends_with_keyword(token, separator) -> None:
     lexer = QasmLexer()
     identifier = f'{token}{separator}{token}'
     lexer.input(identifier)
     t = lexer.token()
+    assert t is not None
     assert t.type == "ID"
     assert t.value == identifier
 
 
-def test_qreg():
+def test_qreg() -> None:
     lexer = QasmLexer()
     lexer.input('qreg [5];')
     token = lexer.token()
+    assert token is not None
     assert token.type == "QREG"
     assert token.value == "qreg"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "["
     assert token.value == "["
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "NATURAL_NUMBER"
     assert token.value == 5
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "]"
     assert token.value == "]"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == ";"
     assert token.value == ";"
 
 
-def test_creg():
+def test_creg() -> None:
     lexer = QasmLexer()
     lexer.input('creg [8];')
     token = lexer.token()
+    assert token is not None
     assert token.type == "CREG"
     assert token.value == "creg"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "["
     assert token.value == "["
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "NATURAL_NUMBER"
     assert token.value == 8
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "]"
     assert token.value == "]"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == ";"
     assert token.value == ";"
 
 
-def test_custom_gate():
+def test_custom_gate() -> None:
     lexer = QasmLexer()
     lexer.input('gate name(param1,param2) q1, q2 {X(q1)}')
     token = lexer.token()
+    assert token is not None
     assert token.type == "GATE"
     assert token.value == "gate"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == "name"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "("
     assert token.value == "("
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == "param1"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == ","
     assert token.value == ","
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == "param2"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == ")"
     assert token.value == ")"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == "q1"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == ","
     assert token.value == ","
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == "q2"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "{"
     assert token.value == "{"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == "X"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "("
     assert token.value == "("
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "ID"
     assert token.value == "q1"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == ")"
     assert token.value == ")"
 
     token = lexer.token()
+    assert token is not None
     assert token.type == "}"
     assert token.value == "}"
 
 
-def test_error():
+def test_error() -> None:
     lexer = QasmLexer()
     lexer.input('θ')
 

cirq/contrib/qasm_import/_parser_test.py

@@ -1516,7 +1516,7 @@ def test_nested_custom_gate_has_keyword_in_name() -> None:
     assert parsed_qasm.circuit == expected
 
 
-def test_rzz_gate():
+def test_rzz_gate() -> None:
     qasm = """
     OPENQASM 2.0;
     include "qelib1.inc";
@@ -1539,7 +1539,7 @@ def test_rzz_gate():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_rxx_gate():
+def test_rxx_gate() -> None:
     qasm = """
     OPENQASM 2.0;
     include "qelib1.inc";
@@ -1562,7 +1562,7 @@ def test_rxx_gate():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_ryy_gate():
+def test_ryy_gate() -> None:
     qasm = """
     OPENQASM 2.0;
     include "qelib1.inc";
@@ -1585,7 +1585,7 @@ def test_ryy_gate():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_crx_gate():
+def test_crx_gate() -> None:
     qasm = """
     OPENQASM 2.0;
     include "qelib1.inc";
@@ -1608,7 +1608,7 @@ def test_crx_gate():
     assert parsed_qasm.qregs == {'q': 2}
 
 
-def test_iswap_gate():
+def test_iswap_gate() -> None:
     qasm = """
     OPENQASM 2.0;
     include "qelib1.inc";