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