def test_returns_computation(self):
proto, _ = tensorflow_computation_factory.create_empty_tuple()
self.assertIsInstance(proto, pb.Computation)
actual_type = type_serialization.deserialize_type(proto.type)
expected_type = computation_types.FunctionType(None, [])
expected_type.check_assignable_from(actual_type)
actual_result = test_utils.run_tensorflow(proto)
expected_result = structure.Struct([])
self.assertEqual(actual_result, expected_result)
def test_returns_coputation(self):
proto = tensorflow_computation_factory.create_empty_tuple()
self.assertIsInstance(proto, pb.Computation)
actual_type = type_serialization.deserialize_type(proto.type)
expected_type = computation_types.FunctionType(None, [])
self.assertEqual(actual_type, expected_type)
actual_result = test_utils.run_tensorflow(proto)
expected_result = anonymous_tuple.AnonymousTuple([])
self.assertEqual(actual_result, expected_result)
def create_dummy_computation_tensorflow_empty():
"""Returns a tensorflow computation and type `( -> <>)`."""
value, type_signature = tensorflow_computation_factory.create_empty_tuple()
return value, type_signature
