code: def test_servable_mapping(
      tmp_path, lightning_squeezenet1_1_obj, squeezenet_servable
  ):
      from tests.core.serve.models import (
          ClassificationInferenceModelMapping,
      )
  
      squeezenet_gm, _ = squeezenet_servable
      model_map = {
          "model_one": squeezenet_gm,
          "model_two": squeezenet_gm,
      }
      comp = ClassificationInferenceModelMapping(model_map)
  
      composit = Composition(comp=comp)
      assert (
          composit.components["callnum_1"]._flashserve_meta_.models
          == model_map
      )
      assert (
          composit.components["callnum_1"].model1
          == model_map["model_one"]
      )
      assert (
          composit.components["callnum_1"].model2
          == model_map["model_two"]
      )
  
file_path: lightning-flash/tests/core/serve/test_composition.py
imports: import base64
  from dataclasses import asdict
  import pytest
  from flash.core.serve import Composition, Endpoint
  from flash.core.utilities.imports import _FASTAPI_AVAILABLE, _TOPIC_SERVE_AVAILABLE
label_desc: Use image node_id 5 for calling a global function with example usage: test_servable_mapping(tmp_path, lightning_squeezenet1_1_obj, squeezenet_servable) without return types