code: def keras_test(func):
      """Function wrapper to clean up after TensorFlow tests.
      # Arguments
          func: test function to clean up after.
      # Returns
          A function wrapping the input function.
      """
  
      @six.wraps(func)
      def wrapper(*args, **kwargs):
          output = func(*args, **kwargs)
          keras.backend.clear_session()
          return output
  
      return wrapper
  
file_path: segmentation_models/tests/test_models.py
imports: import os
  import pytest
  import random
  import six
  import numpy
  import segmentation_models
  from segmentation_models import Unet
  from segmentation_models import Linknet
  from segmentation_models import PSPNet
  from segmentation_models import FPN
  from segmentation_models import get_available_backbone_names
label_desc: Use image node_id 3 for calling a global function with example usage: keras_test(func) and returns: wrapper, output