code: def createPartitions(fileSchemaType, dir_data_file):
      testsWithNulls = Settings.data["RunSettings"]["testsWithNulls"]
      if testsWithNulls == "true":
          dir_data = dir_data_file + "/tpch-with-nulls"
      else:
          dir_data = dir_data_file + "/tpch"
  
      ext = cs.get_extension(fileSchemaType)
  
      # orders table
      utilityHive.test_hive_partition_data(
          input=("%s/%s_[0-9]*.%s") % (dir_data, "orders", ext),
          file_format=ext,
          table_name="orders",
          partitions={
              "o_orderpriority": [
                  "1-URGENT",
                  "2-HIGH",
                  "3-MEDIUM",
                  "4-NOT SPECIFIED",
                  "5-LOW",
              ],
              "o_orderstatus": ["F", "O", "P"],
              "o_orderdate": [
                  pandas.Timestamp("1992-01-01 00:00:00"),
                  pandas.Timestamp("1992-01-02 00:00:00"),
                  pandas.Timestamp("1992-01-03 00:00:00"),
                  pandas.Timestamp("1992-01-04 00:00:00"),
                  pandas.Timestamp("1992-01-05 00:00:00"),
                  pandas.Timestamp("1992-01-06 00:00:00"),
                  pandas.Timestamp("1992-01-07 00:00:00"),
                  pandas.Timestamp("1992-01-08 00:00:00"),
                  pandas.Timestamp("1992-01-09 00:00:00"),
                  pandas.Timestamp("1992-01-10 00:00:00"),
                  pandas.Timestamp("1992-01-11 00:00:00"),
                  pandas.Timestamp("1992-01-12 00:00:00"),
                  pandas.Timestamp("1992-01-13 00:00:00"),
                  pandas.Timestamp("1992-01-14 00:00:00"),
                  pandas.Timestamp("1992-01-15 00:00:00"),
                  pandas.Timestamp("1992-01-16 00:00:00"),
                  pandas.Timestamp("1992-01-17 00:00:00"),
                  pandas.Timestamp("1992-01-18 00:00:00"),
                  pandas.Timestamp("1992-01-19 00:00:00"),
                  pandas.Timestamp("1992-01-20 00:00:00"),
                  pandas.Timestamp("1992-01-21 00:00:00"),
                  pandas.Timestamp("1992-01-22 00:00:00"),
                  pandas.Timestamp("1992-01-23 00:00:00"),
                  pandas.Timestamp("1992-01-24 00:00:00"),
                  pandas.Timestamp("1992-01-25 00:00:00"),
                  pandas.Timestamp("1992-01-26 00:00:00"),
                  pandas.Timestamp("1992-01-27 00:00:00"),
                  pandas.Timestamp("1992-01-28 00:00:00"),
                  pandas.Timestamp("1992-01-29 00:00:00"),
                  pandas.Timestamp("1992-01-30 00:00:00"),
                  pandas.Timestamp("1992-01-31 00:00:00"),
              ],
          },
          partitions_schema=[
              ("o_orderpriority", "str"),
              ("o_orderstatus", "str"),
              ("o_orderdate", "timestamp"),
          ],
          output=tmpPath + ext + "/orders/",
          num_files=1,
      )
  
      # customer table
      utilityHive.test_hive_partition_data(
          input=("%s/%s_[0-9]*.%s") % (dir_data, "customer", ext),
          file_format=ext,
          table_name="customer",
          partitions={
              "c_nationkey": [
                  0,
                  1,
                  2,
                  3,
                  4,
                  5,
                  6,
                  7,
                  8,
                  9,
                  10,
                  11,
                  12,
                  13,
                  14,
                  15,
                  16,
                  17,
                  18,
                  19,
                  20,
                  21,
                  22,
                  23,
                  24,
              ],
              "c_mktsegment": [
                  "AUTOMOBILE",
                  "BUILDING",
                  "FURNITURE",
                  "HOUSEHOLD",
                  "MACHINERY",
              ],
          },
          partitions_schema=[
              ("c_nationkey", "int32"),
              ("c_mktsegment", "str"),
          ],
          output=tmpPath + ext + "/customer/",
          num_files=1,
      )
  
      # lineitem table
      utilityHive.test_hive_partition_data(
          input=("%s/%s_[0-9]*.%s") % (dir_data, "lineitem", ext),
          file_format=ext,
          table_name="lineitem",
          partitions={
              "l_shipmode": [
                  "AIR",
                  "FOB",
                  "MAIL",
                  "RAIL",
                  "REG",
                  "AIR",
                  "SHIP",
                  "TRUCK",
              ],
              "l_linestatus": ["F", "O"],
              "l_returnflag": ["A", "N", "R"],
          },
          partitions_schema=[
              ("l_shipmode", "str"),
              ("l_linestatus", "str"),
              ("l_returnflag", "str"),
          ],
          output=tmpPath + ext + "/lineitem/",
          num_files=1,
      )
  
      # nation table
      utilityHive.test_hive_partition_data(
          input=("%s/%s_[0-9]*.%s") % (dir_data, "nation", ext),
          file_format=ext,
          table_name="nation",
          partitions={"n_regionkey": [0, 1, 2, 3, 4]},
          partitions_schema=[("n_regionkey", "int32")],
          output=tmpPath + ext + "/nation/",
          num_files=1,
      )
  
      # part table
      utilityHive.test_hive_partition_data(
          input=("%s/%s_[0-9]*.%s") % (dir_data, "part", ext),
          file_format=ext,
          table_name="part",
          partitions={
              "p_container": [
                  "JUMBO BAG",
                  "JUMBO BOX",
                  "JUMBO CAN",
                  "JUMBO CASE",
                  "JUMBO DRUM",
                  "JUMBO JAR",
                  "JUMBO PACK",
                  "JUMBO PKG",
                  "LG BAG",
                  "LG BOX",
                  "LG CAN",
                  "LG CASE",
                  "LG DRUM",
                  "LG JAR",
                  "LG PACK",
                  "LG PKG",
                  "MED BAG",
                  "MED BOX",
                  "MED CAN",
                  "MED CASE",
                  "MED DRUM",
                  "MED JAR",
                  "MED PACK",
                  "MED PKG",
                  "SM BAG",
                  "SM BOX",
                  "SM CAN",
                  "SM CASE",
                  "SM DRUM",
                  "SM JAR",
                  "SM PACK",
                  "SM PKG",
                  "WRAP BAG",
                  "WRAP BOX",
                  "WRAP CAN",
                  "WRAP CASE",
                  "WRAP DRUM",
                  "WRAP JAR",
                  "WRAP PACK",
                  "WRAP PKG",
              ]
          },
          partitions_schema=[("p_container", "str")],
          output=tmpPath + ext + "/part/",
          num_files=1,
      )
  
      # supplier table
      utilityHive.test_hive_partition_data(
          input=("%s/%s_[0-9]*.%s") % (dir_data, "supplier", ext),
          file_format=ext,
          table_name="supplier",
          partitions={
              "s_nationkey": [
                  0,
                  1,
                  2,
                  3,
                  4,
                  5,
                  6,
                  7,
                  8,
                  9,
                  10,
                  11,
                  12,
                  13,
                  14,
                  15,
                  16,
                  17,
                  18,
                  19,
                  20,
                  21,
                  22,
                  23,
                  24,
              ]
          },
          partitions_schema=[("s_nationkey", "int32")],
          output=tmpPath + ext + "/supplier/",
          num_files=1,
      )
  
file_path: blazingsql/tests/BlazingSQLTest/EndToEndTests/oldScripts/hiveFileTest.py
imports: from blazingsql import DataType
  from Configuration import ExecutionMode
  from Configuration import Settings
  from DataBase import createSchema
  from pynvml import nvmlInit
  from Runner import runTest
  from Utils import Execution, gpuMemory, init_context, skip_test, utilityHive
  import shutil
  import pandas
  import tempfile
label_desc: Use image node_id 5 for calling a global function with example usage: createPartitions(fileSchemaType, dir_data_file) without return types