cirq-core 1.7.0.dev20250818230756__py3-none-any.whl → 1.7.0.dev20250820205919__py3-none-any.whl

cirq/ops/permutation_gate.py

@@ -69,7 +69,7 @@ class QubitPermutationGate(raw_types.Gate):
     def _value_equality_values_(self):
         return self.permutation
 
-    def num_qubits(self):
+    def num_qubits(self) -> int:
         return len(self.permutation)
 
     def _has_unitary_(self):

cirq/ops/raw_types.py

@@ -33,7 +33,6 @@ from typing import (
 )
 
 import numpy as np
-from typing_extensions import Self
 
 from cirq import protocols, value
 from cirq._compat import __cirq_debug__, _method_cache_name, cached_method
@@ -521,7 +520,7 @@ class Operation(metaclass=abc.ABCMeta):
         return protocols.qid_shape(self.qubits)
 
     @abc.abstractmethod
-    def with_qubits(self, *new_qubits: cirq.Qid) -> Self:
+    def with_qubits(self, *new_qubits: cirq.Qid) -> cirq.Operation:
         """Returns the same operation, but applied to different qubits.
 
         Args:
@@ -567,7 +566,7 @@ class Operation(metaclass=abc.ABCMeta):
 
     def transform_qubits(
         self, qubit_map: dict[cirq.Qid, cirq.Qid] | Callable[[cirq.Qid], cirq.Qid]
-    ) -> Self:
+    ) -> cirq.Operation:
         """Returns the same operation, but with different qubits.
 
         This function will return a new operation with the same gate but
@@ -769,7 +768,7 @@ class TaggedOperation(Operation):
     def gate(self) -> cirq.Gate | None:
         return self.sub_operation.gate
 
-    def with_qubits(self, *new_qubits: cirq.Qid):
+    def with_qubits(self, *new_qubits: cirq.Qid) -> TaggedOperation:
         return TaggedOperation(self.sub_operation.with_qubits(*new_qubits), *self._tags)
 
     def _with_measurement_key_mapping_(self, key_map: Mapping[str, str]):

cirq/ops/raw_types_test.py

@@ -48,7 +48,7 @@ class ValidQid(cirq.Qid):
         self.validate_dimension(dimension)
 
     @property
-    def dimension(self):
+    def dimension(self) -> int:
         return self._dimension
 
     def with_dimension(self, dimension) -> ValidQid:

cirq/ops/three_qubit_gates.py

@@ -423,8 +423,8 @@ class CCXPowGate(gate_features.InterchangeableQubitsGate, eigen_gate.EigenGate):
             }
         )
 
-    def qubit_index_to_equivalence_group_key(self, index):
-        return index < 2
+    def qubit_index_to_equivalence_group_key(self, index) -> int:
+        return 1 if index < 2 else 0
 
     def _apply_unitary_(self, args: protocols.ApplyUnitaryArgs) -> np.ndarray:
         if protocols.is_parameterized(self):
@@ -508,7 +508,7 @@ class CCXPowGate(gate_features.InterchangeableQubitsGate, eigen_gate.EigenGate):
 class CSwapGate(gate_features.InterchangeableQubitsGate, raw_types.Gate):
     """A controlled swap gate. The Fredkin gate."""
 
-    def qubit_index_to_equivalence_group_key(self, index):
+    def qubit_index_to_equivalence_group_key(self, index) -> int:
         return 0 if index == 0 else 1
 
     def _pauli_expansion_(self) -> value.LinearDict[str]:

cirq/protocols/act_on_protocol.py

@@ -94,7 +94,7 @@ def act_on(
     qubits: Sequence[cirq.Qid] | None = None,
     *,
     allow_decompose: bool = True,
-):
+) -> None:
     """Applies an action to a state argument.
 
     For example, the action may be a `cirq.Operation` and the state argument may

cirq/protocols/act_on_protocol_test.py

@@ -24,11 +24,11 @@ import cirq
 
 
 class ExampleQuantumState(cirq.QuantumStateRepresentation):
-    def copy(self, deep_copy_buffers=True):
-        pass
+    def copy(self, deep_copy_buffers=True) -> Self:
+        return self.__class__()  # pragma: no cover
 
-    def measure(self, axes, seed=None):
-        pass
+    def measure(self, axes, seed=None) -> list[int]:
+        return []  # pragma: no cover
 
 
 class ExampleSimulationState(cirq.SimulationState):

cirq/protocols/apply_channel_protocol_test.py

@@ -20,7 +20,7 @@ import pytest
 import cirq
 
 
-def make_buffers(shape, dtype):
+def make_buffers(shape, dtype) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
     return (
         np.empty(shape, dtype=dtype),
         np.empty(shape, dtype=dtype),
@@ -28,7 +28,7 @@ def make_buffers(shape, dtype):
     )
 
 
-def apply_channel(val, rho, left_axes, right_axes, assert_result_is_out_buf=False):
+def apply_channel(val, rho, left_axes, right_axes, assert_result_is_out_buf=False) -> np.ndarray:
     out_buf, buf0, buf1 = make_buffers(rho.shape, rho.dtype)
     result = cirq.apply_channel(
         val,
@@ -48,7 +48,7 @@ def apply_channel(val, rho, left_axes, right_axes, assert_result_is_out_buf=Fals
     return result
 
 
-def test_apply_channel_bad_args():
+def test_apply_channel_bad_args() -> None:
     target = np.zeros((3,) + (1, 2, 3) + (3, 1, 2) + (3,))
     with pytest.raises(ValueError, match='Invalid target_tensor shape'):
         cirq.apply_channel(
@@ -77,7 +77,7 @@ def test_apply_channel_bad_args():
         )
 
 
-def test_apply_channel_simple():
+def test_apply_channel_simple() -> None:
     x = np.array([[0, 1], [1, 0]], dtype=np.complex128)
 
     class HasApplyChannel:
@@ -108,7 +108,7 @@ def test_apply_channel_simple():
     np.testing.assert_almost_equal(result, x)
 
 
-def test_apply_channel_inline():
+def test_apply_channel_inline() -> None:
     x = np.array([[0, 1], [1, 0]], dtype=np.complex128)
 
     class HasApplyChannel:
@@ -123,7 +123,7 @@ def test_apply_channel_inline():
     np.testing.assert_almost_equal(result, x)
 
 
-def test_apply_channel_returns_aux_buffer():
+def test_apply_channel_returns_aux_buffer() -> None:
     rho = np.array([[1, 0], [0, 0]], dtype=np.complex128)
 
     class ReturnsAuxBuffer0:
@@ -141,7 +141,7 @@ def test_apply_channel_returns_aux_buffer():
         _ = apply_channel(ReturnsAuxBuffer1(), rho, [0], [1])
 
 
-def test_apply_channel_channel_fallback_simple():
+def test_apply_channel_channel_fallback_simple() -> None:
     x = np.array([[0, 1], [1, 0]], dtype=np.complex128)
 
     class HasChannel:
@@ -153,7 +153,7 @@ def test_apply_channel_channel_fallback_simple():
     np.testing.assert_almost_equal(result, x)
 
 
-def test_apply_channel_channel_fallback_one_qubit_random_on_qubit():
+def test_apply_channel_channel_fallback_one_qubit_random_on_qubit() -> None:
     for _ in range(25):
         state = cirq.testing.random_superposition(2)
         rho = np.outer(np.conjugate(state), state)
@@ -171,7 +171,7 @@ def test_apply_channel_channel_fallback_one_qubit_random_on_qubit():
         np.testing.assert_almost_equal(result, expected)
 
 
-def test_apply_channel_channel_fallback_one_qubit_random_on_two_qubits():
+def test_apply_channel_channel_fallback_one_qubit_random_on_two_qubits() -> None:
     for _ in range(25):
         state = cirq.testing.random_superposition(4)
         rho = np.outer(np.conjugate(state), state)
@@ -193,7 +193,7 @@ def test_apply_channel_channel_fallback_one_qubit_random_on_two_qubits():
         np.testing.assert_almost_equal(result, expected)
 
 
-def test_apply_channel_channel_fallback_two_qubit_random():
+def test_apply_channel_channel_fallback_two_qubit_random() -> None:
     for _ in range(25):
         state = cirq.testing.random_superposition(4)
         rho = np.outer(np.conjugate(state), state)
@@ -214,7 +214,7 @@ def test_apply_channel_channel_fallback_two_qubit_random():
         np.testing.assert_almost_equal(result, expected)
 
 
-def test_apply_channel_no_protocols_implemented():
+def test_apply_channel_no_protocols_implemented() -> None:
     class NoProtocols:
         pass
 
@@ -224,23 +224,24 @@ def test_apply_channel_no_protocols_implemented():
         apply_channel(NoProtocols(), rho, left_axes=[1], right_axes=[1])
 
 
-def test_apply_channel_no_protocols_implemented_default():
+def test_apply_channel_no_protocols_implemented_default() -> None:
     class NoProtocols:
         pass
 
+    out_buffer, auxiliary_buffer0, auxiliary_buffer1 = make_buffers((2, 2), float)
     args = cirq.ApplyChannelArgs(
         target_tensor=np.eye(2),
         left_axes=[0],
         right_axes=[1],
-        out_buffer=None,
-        auxiliary_buffer0=None,
-        auxiliary_buffer1=None,
+        out_buffer=out_buffer,
+        auxiliary_buffer0=auxiliary_buffer0,
+        auxiliary_buffer1=auxiliary_buffer1,
     )
     result = cirq.apply_channel(NoProtocols(), args, 'cirq')
     assert result == 'cirq'
 
 
-def test_apply_channel_unitary():
+def test_apply_channel_unitary() -> None:
     m = np.diag([1, 1j])
 
     shape = (2, 2, 2, 2)
@@ -263,7 +264,7 @@ def test_apply_channel_unitary():
         )
 
 
-def test_apply_channel_apply_unitary():
+def test_apply_channel_apply_unitary() -> None:
     shape = (2, 2, 2, 2)
     rho = np.ones(shape, dtype=np.complex128)
 
@@ -291,7 +292,7 @@ def test_apply_channel_apply_unitary():
         )
 
 
-def test_apply_channel_apply_unitary_not_implemented():
+def test_apply_channel_apply_unitary_not_implemented() -> None:
     class ApplyUnitaryNotImplemented:
         def _apply_unitary_(self, args: cirq.ApplyUnitaryArgs):
             return NotImplemented

cirq/protocols/apply_mixture_protocol_test.py

@@ -21,7 +21,7 @@ import pytest
 import cirq
 
 
-def make_buffers(shape, dtype):
+def make_buffers(shape, dtype) -> tuple[np.ndarray, np.ndarray, np.ndarray]:
     return (
         np.empty(shape, dtype=dtype),
         np.empty(shape, dtype=dtype),
@@ -36,7 +36,7 @@ def assert_apply_mixture_returns(
     right_axes: Iterable[int] | None,
     assert_result_is_out_buf: bool = False,
     expected_result: np.ndarray | None = None,
-):
+) -> None:
     out_buf, buf0, buf1 = make_buffers(rho.shape, rho.dtype)
     result = cirq.apply_mixture(
         val,
@@ -58,7 +58,7 @@ def assert_apply_mixture_returns(
     np.testing.assert_array_almost_equal(result, expected_result)
 
 
-def test_apply_mixture_bad_args():
+def test_apply_mixture_bad_args() -> None:
     target = np.zeros((3,) + (1, 2, 3) + (3, 1, 2) + (3,))
     with pytest.raises(ValueError, match='Invalid target_tensor shape'):
         cirq.apply_mixture(
@@ -89,7 +89,7 @@ def test_apply_mixture_bad_args():
         )
 
 
-def test_apply_mixture_simple():
+def test_apply_mixture_simple() -> None:
     x = np.array([[0, 1], [1, 0]], dtype=np.complex128)
 
     class HasApplyMixture:
@@ -121,7 +121,7 @@ def test_apply_mixture_simple():
     )
 
 
-def test_apply_mixture_inline():
+def test_apply_mixture_inline() -> None:
     x = np.array([[0, 1], [1, 0]], dtype=np.complex128)
 
     class HasApplyMixture:
@@ -135,7 +135,7 @@ def test_apply_mixture_inline():
     assert_apply_mixture_returns(HasApplyMixture(), rho, [0], [1], expected_result=x)
 
 
-def test_apply_mixture_returns_aux_buffer():
+def test_apply_mixture_returns_aux_buffer() -> None:
     rho = np.array([[1, 0], [0, 0]], dtype=np.complex128)
 
     class ReturnsAuxBuffer0:
@@ -153,7 +153,7 @@ def test_apply_mixture_returns_aux_buffer():
         assert_apply_mixture_returns(ReturnsAuxBuffer1(), rho, [0], [1])
 
 
-def test_apply_mixture_simple_state_vector():
+def test_apply_mixture_simple_state_vector() -> None:
     for _ in range(25):
         state = cirq.testing.random_superposition(2)
         u1 = cirq.testing.random_unitary(2)
@@ -172,7 +172,7 @@ def test_apply_mixture_simple_state_vector():
         )
 
 
-def test_apply_mixture_simple_split_fallback():
+def test_apply_mixture_simple_split_fallback() -> None:
     x = np.array([[0, 1], [1, 0]], dtype=np.complex128)
 
     class HasMixture:
@@ -185,7 +185,7 @@ def test_apply_mixture_simple_split_fallback():
     )
 
 
-def test_apply_mixture_fallback_one_qubit_random_on_qubit():
+def test_apply_mixture_fallback_one_qubit_random_on_qubit() -> None:
     for _ in range(25):
         state = cirq.testing.random_superposition(2)
         rho = np.outer(np.conjugate(state), state)
@@ -203,7 +203,7 @@ def test_apply_mixture_fallback_one_qubit_random_on_qubit():
         )
 
 
-def test_apply_mixture_fallback_two_qubit_random():
+def test_apply_mixture_fallback_two_qubit_random() -> None:
     for _ in range(25):
         state = cirq.testing.random_superposition(4)
         rho = np.outer(np.conjugate(state), state)
@@ -229,7 +229,7 @@ def test_apply_mixture_fallback_two_qubit_random():
         )
 
 
-def test_apply_mixture_no_protocols_implemented():
+def test_apply_mixture_no_protocols_implemented() -> None:
     class NoProtocols:
         pass
 
@@ -239,7 +239,7 @@ def test_apply_mixture_no_protocols_implemented():
         assert_apply_mixture_returns(NoProtocols(), rho, left_axes=[1], right_axes=[1])
 
 
-def test_apply_mixture_mixture_returns_not_implemented():
+def test_apply_mixture_mixture_returns_not_implemented() -> None:
     class NoMixture:
         def _mixture_(self):
             return NotImplemented
@@ -250,23 +250,24 @@ def test_apply_mixture_mixture_returns_not_implemented():
         assert_apply_mixture_returns(NoMixture(), rho, left_axes=[1], right_axes=[1])
 
 
-def test_apply_mixture_no_protocols_implemented_default():
+def test_apply_mixture_no_protocols_implemented_default() -> None:
     class NoProtocols:
         pass
 
+    out_buffer, auxiliary_buffer0, auxiliary_buffer1 = make_buffers((2, 2), float)
     args = cirq.ApplyMixtureArgs(
         target_tensor=np.eye(2),
         left_axes=[0],
         right_axes=[1],
-        out_buffer=None,
-        auxiliary_buffer0=None,
-        auxiliary_buffer1=None,
+        out_buffer=out_buffer,
+        auxiliary_buffer0=auxiliary_buffer0,
+        auxiliary_buffer1=auxiliary_buffer1,
     )
     result = cirq.apply_mixture(NoProtocols(), args, default='cirq')
     assert result == 'cirq'
 
 
-def test_apply_mixture_unitary():
+def test_apply_mixture_unitary() -> None:
     m = np.diag([1, 1j])
 
     shape = (2, 2, 2, 2)
@@ -291,7 +292,7 @@ def test_apply_mixture_unitary():
         )
 
 
-def test_apply_mixture_apply_unitary():
+def test_apply_mixture_apply_unitary() -> None:
     shape = (2, 2, 2, 2)
     rho = np.ones(shape, dtype=np.complex128)
 
@@ -320,7 +321,7 @@ def test_apply_mixture_apply_unitary():
         )
 
 
-def test_apply_mixture_apply_unitary_not_implemented():
+def test_apply_mixture_apply_unitary_not_implemented() -> None:
     class ApplyUnitaryNotImplemented:
         def _apply_unitary_(self, args: cirq.ApplyUnitaryArgs):
             return NotImplemented

cirq/protocols/apply_unitary_protocol_test.py

@@ -21,7 +21,7 @@ import cirq
 from cirq.protocols.apply_unitary_protocol import _incorporate_result_into_target
 
 
-def test_apply_unitary_presence_absence():
+def test_apply_unitary_presence_absence() -> None:
     m = np.diag([1, -1])
 
     class NoUnitaryEffect:
@@ -102,7 +102,7 @@ def test_apply_unitary_presence_absence():
         )
 
 
-def test_apply_unitary_args_tensor_manipulation():
+def test_apply_unitary_args_tensor_manipulation() -> None:
     # All below are qubit swap operations with 1j global phase
 
     class ModifyTargetTensor:
@@ -215,7 +215,7 @@ def test_apply_unitary_args_tensor_manipulation():
             args.available_buffer[...] = 98
             return ret
 
-    operations = [
+    operations: list[cirq.SupportsConsistentApplyUnitary] = [
         ModifyTargetTensor(),
         TransposeTargetTensor(),
         ReshapeTargetTensor(),
@@ -266,7 +266,7 @@ def test_apply_unitary_args_tensor_manipulation():
         assert_is_swap(op)
 
 
-def test_big_endian_subspace_index():
+def test_big_endian_subspace_index() -> None:
     state = np.zeros(shape=(2, 3, 4, 5, 1, 6, 1, 1))
     args = cirq.ApplyUnitaryArgs(state, np.empty_like(state), [1, 3])
     s = slice(None)
@@ -274,12 +274,13 @@ def test_big_endian_subspace_index():
     assert args.subspace_index(big_endian_bits_int=1) == (s, 0, s, 1, s, s, s, s)
 
 
-def test_apply_unitaries():
+def test_apply_unitaries() -> None:
     a, b, c = cirq.LineQubit.range(3)
 
     result = cirq.apply_unitaries(
         unitary_values=[cirq.H(a), cirq.CNOT(a, b), cirq.H(c).controlled_by(b)], qubits=[a, b, c]
     )
+    assert result is not None
     np.testing.assert_allclose(
         result.reshape(8), [np.sqrt(0.5), 0, 0, 0, 0, 0, 0.5, 0.5], atol=1e-8
     )
@@ -288,6 +289,7 @@ def test_apply_unitaries():
     result = cirq.apply_unitaries(
         unitary_values=[cirq.H(a), cirq.CNOT(a, b), cirq.H(c).controlled_by(b)], qubits=[a, c, b]
     )
+    assert result is not None
     np.testing.assert_allclose(
         result.reshape(8), [np.sqrt(0.5), 0, 0, 0, 0, 0.5, 0, 0.5], atol=1e-8
     )
@@ -298,14 +300,17 @@ def test_apply_unitaries():
         qubits=[a, b, c],
         args=cirq.ApplyUnitaryArgs.default(num_qubits=3),
     )
+    assert result is not None
     np.testing.assert_allclose(
         result.reshape(8), [np.sqrt(0.5), 0, 0, 0, 0, 0, 0.5, 0.5], atol=1e-8
     )
 
     # Empty.
     result = cirq.apply_unitaries(unitary_values=[], qubits=[])
+    assert result is not None
     np.testing.assert_allclose(result, [1])
     result = cirq.apply_unitaries(unitary_values=[], qubits=[], default=None)
+    assert result is not None
     np.testing.assert_allclose(result, [1])
 
     # Non-unitary operation.
@@ -327,7 +332,7 @@ def test_apply_unitaries():
         )
 
 
-def test_apply_unitaries_mixed_qid_shapes():
+def test_apply_unitaries_mixed_qid_shapes() -> None:
     class PlusOneMod3Gate(cirq.testing.SingleQubitGate):
         def _qid_shape_(self):
             return (3,)
@@ -358,6 +363,7 @@ def test_apply_unitaries_mixed_qid_shapes():
         ],
         qubits=[a, b],
     )
+    assert result is not None
     np.testing.assert_allclose(result.reshape(12), [1] + [0] * 11, atol=1e-8)
 
     result = cirq.apply_unitaries(
@@ -377,6 +383,7 @@ def test_apply_unitaries_mixed_qid_shapes():
             axes=(0, 1),
         ),
     )
+    assert result is not None
     np.testing.assert_allclose(result.reshape(12, 12), np.eye(12), atol=1e-8)
 
     result = cirq.apply_unitaries(
@@ -403,6 +410,7 @@ def test_apply_unitaries_mixed_qid_shapes():
             axes=(0, 1),
         ),
     )
+    assert result is not None
     np.testing.assert_allclose(
         result.reshape(12, 12),
         np.array(
@@ -426,7 +434,7 @@ def test_apply_unitaries_mixed_qid_shapes():
 
 
 # fmt: off
-def test_subspace_size_2():
+def test_subspace_size_2() -> None:
     result = cirq.apply_unitary(
         unitary_value=cirq.X,
         args=cirq.ApplyUnitaryArgs(
@@ -513,7 +521,7 @@ def test_subspace_size_2():
     )
 
 
-def test_subspaces_size_3():
+def test_subspaces_size_3() -> None:
     plus_one_mod_3_gate = cirq.XPowGate(dimension=3)
 
     result = cirq.apply_unitary(
@@ -581,7 +589,7 @@ def test_subspaces_size_3():
     )
 
 
-def test_subspaces_size_1():
+def test_subspaces_size_1() -> None:
     phase_gate = cirq.MatrixGate(np.array([[1j]]))
 
     result = cirq.apply_unitary(
@@ -646,7 +654,7 @@ def test_subspaces_size_1():
 # fmt: on
 
 
-def test_invalid_subspaces():
+def test_invalid_subspaces() -> None:
     with pytest.raises(ValueError, match='Subspace specified does not exist in axis'):
         _ = cirq.ApplyUnitaryArgs(
             target_tensor=cirq.eye_tensor((2,), dtype=np.complex64),
@@ -677,7 +685,7 @@ def test_invalid_subspaces():
         )
 
 
-def test_incorporate_result_not_view():
+def test_incorporate_result_not_view() -> None:
     tensor = np.zeros((2, 2))
     tensor2 = np.zeros((2, 2))
     buffer = np.empty_like(tensor)
@@ -687,12 +695,12 @@ def test_incorporate_result_not_view():
         _incorporate_result_into_target(args, not_sub_args, tensor2)
 
 
-def test_default_method_arguments():
+def test_default_method_arguments() -> None:
     with pytest.raises(TypeError, match='exactly one of'):
         cirq.ApplyUnitaryArgs.default(1, qid_shape=(2,))
 
 
-def test_apply_unitary_args_with_axes_transposed_to_start():
+def test_apply_unitary_args_with_axes_transposed_to_start() -> None:
     target = np.zeros((2, 3, 4, 5))
     buffer = np.zeros((2, 3, 4, 5))
     args = cirq.ApplyUnitaryArgs(target, buffer, [1, 3])
@@ -709,7 +717,8 @@ def test_apply_unitary_args_with_axes_transposed_to_start():
     assert args.available_buffer[1, 2, 3, 4] == 2
 
 
-def test_cast_to_complex():
+def test_cast_to_complex() -> None:
+    y0: cirq.PauliString[cirq.LineQubit]
     y0 = cirq.PauliString({cirq.LineQubit(0): cirq.Y})
     state = 0.5 * np.eye(2)
     args = cirq.ApplyUnitaryArgs(
@@ -724,7 +733,7 @@ def test_cast_to_complex():
 
 
 class NotDecomposableGate(cirq.Gate):
-    def num_qubits(self):
+    def num_qubits(self) -> int:
         return 1
 
 
@@ -734,7 +743,7 @@ class DecomposableGate(cirq.Gate):
         self._sub_gate = sub_gate
         self._allocate_ancilla = allocate_ancilla
 
-    def num_qubits(self):
+    def num_qubits(self) -> int:
         return 1
 
     def _decompose_(self, qubits):
@@ -743,17 +752,16 @@ class DecomposableGate(cirq.Gate):
         yield self._sub_gate(qubits[0])
 
 
-def test_strat_apply_unitary_from_decompose():
+def test_strat_apply_unitary_from_decompose() -> None:
     state = np.eye(2, dtype=np.complex128)
     args = cirq.ApplyUnitaryArgs(
         target_tensor=state, available_buffer=np.zeros_like(state), axes=(0,)
     )
-    np.testing.assert_allclose(
-        cirq.apply_unitaries(
-            [DecomposableGate(cirq.X, False)(cirq.LineQubit(0))], [cirq.LineQubit(0)], args
-        ),
-        [[0, 1], [1, 0]],
+    result = cirq.apply_unitaries(
+        [DecomposableGate(cirq.X, False)(cirq.LineQubit(0))], [cirq.LineQubit(0)], args
     )
+    assert result is not None
+    np.testing.assert_allclose(result, [[0, 1], [1, 0]])
 
     with pytest.raises(TypeError):
         _ = cirq.apply_unitaries(
@@ -763,7 +771,7 @@ def test_strat_apply_unitary_from_decompose():
         )
 
 
-def test_unitary_construction():
+def test_unitary_construction() -> None:
     with pytest.raises(TypeError):
         _ = cirq.ApplyUnitaryArgs.for_unitary()