def test_unloadable_proto_from_literal_type(): with pytest.raises(_user_exceptions.FlyteAssertion): _flyte_engine.FlyteDefaultTypeEngine().get_sdk_type_from_literal_type( _type_models.LiteralType( simple=_type_models.SimpleType.BINARY, metadata={_proto.Protobuf.PB_FIELD_KEY: "flyteidl.core.errors_pb2.ContainerErrorNoExist"}, ) )
def test_generic_proto_from_literal_type(): sdk_type = _flyte_engine.FlyteDefaultTypeEngine().get_sdk_type_from_literal_type( _type_models.LiteralType( simple=_type_models.SimpleType.STRUCT, metadata={_proto.Protobuf.PB_FIELD_KEY: "flyteidl.core.errors_pb2.ContainerError"}, ) ) assert sdk_type.pb_type == _errors_pb2.ContainerError
def test_infer_proto_from_literal(): sdk_type = _flyte_engine.FlyteDefaultTypeEngine().infer_sdk_type_from_literal( _literal_models.Literal( scalar=_literal_models.Scalar( binary=_literal_models.Binary( value="", tag="{}{}".format(_proto.Protobuf.TAG_PREFIX, "flyteidl.core.errors_pb2.ContainerError",), ) ) ) ) assert sdk_type.pb_type == _errors_pb2.ContainerError