code: def gen_data1():
      with tempfile.TemporaryDirectory() as tempdir:
          df = pd.DataFrame(
              {
                  "a": [3, 4, 5, 3, 5, 4, 1, 2, 3],
                  "b": [1, 3, 4, 5, 6, 5, 4, 4, 4],
                  "c": list("aabaaddce"),
                  "d": list("abaaaddce"),
              }
          )
          yield df, tempdir
  
file_path: mars/mars/optimization/logical/chunk/tests/test_column_pruning.py
imports: import os
  import tempfile
  import pandas
  import pytest
  from .....None import dataframe
  from .....core import enter_mode, TileableGraph, TileableGraphBuilder, ChunkGraphBuilder, TileContext
  from ..None import optimize
label_desc: Use image node_id 1 for calling a global function with example usage: gen_data1() without return types