code: def generate(generate_if_exists: bool, copy_dir: str):
      row_n = 9000
      sort_dir = "./test/data/csv"
      slt_dir = "./test/sql/dql"
  
      table_name = "test_sort"
      sort_path = sort_dir + "/test_sort.csv"
      slt_path = slt_dir + "/sort.slt"
      copy_path = copy_dir + "/test_sort.csv"
  
      os.makedirs(sort_dir, exist_ok=True)
      os.makedirs(slt_dir, exist_ok=True)
      if (
          os.path.exists(sort_path)
          and os.path.exists(slt_path)
          and generate_if_exists
      ):
          print(
              "File {} and {} already existed exists. Skip Generating.".format(
                  slt_path, sort_path
              )
          )
          return
      with open(sort_path, "w") as sort_file, open(
          slt_path, "w"
      ) as slt_file:
          slt_file.write("statement ok\n")
          slt_file.write(
              "DROP TABLE IF EXISTS {};\n".format(table_name)
          )
          slt_file.write("\n")
          slt_file.write("statement ok\n")
          slt_file.write(
              "CREATE TABLE {} (c1 int, c2 int);\n".format(table_name)
          )
          slt_file.write("\n")
          slt_file.write("query I\n")
          slt_file.write(
              "COPY {} FROM '{}' WITH ( DELIMITER ',' );\n".format(
                  table_name, copy_path
              )
          )
          slt_file.write("----\n")
          slt_file.write("\n")
          slt_file.write("query I\n")
          slt_file.write(
              "SELECT * FROM {} order by c1, c2;\n".format(table_name)
          )
          slt_file.write("----\n")
  
          random_integers = np.random.randint(
              low=1, high=row_n, size=row_n
          )
  
          for i in random_integers:
              sort_file.write("0," + str(i))
              sort_file.write("\n")
          for i in sorted(random_integers):
              slt_file.write("0 " + str(i))
              slt_file.write("\n")
  
          slt_file.write("\n")
          slt_file.write("statement ok\n")
          slt_file.write("DROP TABLE {};\n".format(table_name))
      random.random()
  
imports: import numpy
  import random
  import os
  import argparse
label_desc: Use image node_id 1 for calling a global function with example usage: generate(generate_if_exists, copy_dir) without return types