code: def get_data():
      def get_linear_data(a, b, size):
          x = np.arange(0, 10, 10 / size, dtype=np.float32)
          y = a * x + b
          return x, y
  
      train_x, train_y = get_linear_data(2, 5, 1000)
      val_x, val_y = get_linear_data(2, 5, 400)
      data = {
          "x": train_x,
          "y": train_y,
          "val_x": val_x,
          "val_y": val_y,
      }
      return data
  
file_path: analytics-zoo/pyzoo/test/zoo/orca/automl/model/test_base_keras_model.py
imports: from unittest import TestCase
  from zoo.orca.automl.model.base_keras_model import KerasBaseModel, KerasModelBuilder
  import numpy
  import tensorflow
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: get_data() and returns: data, x,  y