code: def helper_get_data():
      train_data = [
          {
              "action": 1,
              "cost": 2,
              "probability": 0.4,
              "feature1": "a",
              "feature2": "c",
              "feature3": "",
          },
          {
              "action": 3,
              "cost": 0,
              "probability": 0.2,
              "feature1": "b",
              "feature2": "d",
              "feature3": "",
          },
          {
              "action": 4,
              "cost": 1,
              "probability": 0.5,
              "feature1": "a",
              "feature2": "b",
              "feature3": "",
          },
          {
              "action": 2,
              "cost": 1,
              "probability": 0.3,
              "feature1": "a",
              "feature2": "b",
              "feature3": "c",
          },
          {
              "action": 3,
              "cost": 1,
              "probability": 0.7,
              "feature1": "a",
              "feature2": "d",
              "feature3": "",
          },
      ]
  
      train_df = pd.DataFrame(train_data)
  
      train_df["index"] = range(1, len(train_df) + 1)
      train_df = train_df.set_index("index")
  
      test_data = [
          {"feature1": "b", "feature2": "c", "feature3": ""},
          {"feature1": "a", "feature2": "", "feature3": "b"},
          {"feature1": "b", "feature2": "b", "feature3": ""},
          {"feature1": "a", "feature2": "", "feature3": "b"},
      ]
  
      test_df = pd.DataFrame(test_data)
  
      # Add index to data frame
      test_df["index"] = range(1, len(test_df) + 1)
      test_df = test_df.set_index("index")
  
      return train_df, test_df
  
file_path: vowpal_wabbit/python/tests/test_cb.py
imports: import pandas
  from os import path
  import vowpalwabbit
  import unittest
  import platform
  import math
  import re
label_desc: Use image node_id 2 for calling a global function with example usage: helper_get_data() and returns: train_df,  test_df