code: def data(request):
      graph = pd.DataFrame(get_test_graph())
      graph.to_csv(request.param, header=None, index=None, sep="\t")
      data_loader = GraphDataLoader(request.param)
      yield data_loader
      data_loader.clean()
      try:
          os.remove(request.param)
      except:
          del request.param
  
imports: import pytest
  from ampligraph.datasets.graph_partitioner import AbstractGraphPartitioner, RandomVerticesGraphPartitioner, RandomEdgesGraphPartitioner, SortedEdgesGraphPartitioner, NaiveGraphPartitioner, DoubleSortedEdgesGraphPartitioner, get_number_of_partitions, PARTITION_ALGO_REGISTRY, GraphDataLoader
  import numpy
  import pandas
  from scipy import special
  from itertools import permutations
  import os
label_desc: Use image node_id 3 for calling a global function with example usage: data(request) without return types