def test_reconcile_value_type_with_type_spec_raises_type_error_value_type_and_bad_type_spec(
self):
value_type = computation_types.TensorType(tf.int32)
type_spec = computation_types.TensorType(tf.string)
with self.assertRaises(TypeError):
executor_utils.reconcile_value_type_with_type_spec(value_type, type_spec)
def _serialize_computation(
comp: computation_pb2.Computation,
type_spec: Optional[computation_types.Type]) -> _SerializeReturnType:
"""Serializes a TFF computation."""
type_spec = executor_utils.reconcile_value_type_with_type_spec(
type_serialization.deserialize_type(comp.type), type_spec)
return executor_pb2.Value(computation=comp), type_spec
def test_reconcile_value_type_with_type_spec_returns_type(
self, value_type, type_spec, expected_type):
actual_type = executor_utils.reconcile_value_type_with_type_spec(
value_type, type_spec)
self.assertEqual(actual_type, expected_type)