code: def test_regression(self, mock_handler):
      df = pd.DataFrame(
          {
              "age": [19, 18, 28, 33, 32, 31, 46, 37],
              "sex": [
                  "female",
                  "male",
                  "male",
                  "male",
                  "male",
                  "female",
                  "female",
                  "female",
              ],
              "bmi": [
                  27.9,
                  33.77,
                  33,
                  22.705,
                  28.88,
                  25.74,
                  33.44,
                  27.74,
              ],
              "children": [0, 1, 3, 0, 0, 0, 1, 3],
              "smoker": [
                  "yes",
                  "no",
                  "no",
                  "no",
                  "no",
                  "no",
                  "no",
                  "no",
              ],
              "region": [
                  "southwest",
                  "southeast",
                  "southeast",
                  "northwest",
                  "northwest",
                  "southeast",
                  "southeast",
                  "northwest",
              ],
              "charges": [
                  16884.924,
                  1725.5523,
                  4449.462,
                  21984.47061,
                  3866.8552,
                  3756.6216,
                  8240.5896,
                  7281.5056,
              ],
          }
      )
  
      self.set_handler(
          mock_handler, name="pg", tables={"insurance": df}
      )
  
      # create project
      self.run_sql("create database proj;")
  
      # create predictor
      self.run_sql(
          """
              CREATE MODEL proj.my_pycaret_regr_model
              FROM pg
                  (SELECT age, sex, bmi, children, smoker, region, charges FROM insurance)
              PREDICT charges
              USING
                engine = 'pycaret',
                model_type = 'regression',
                model_name = 'xgboost',
                setup_session_id = 123,
                setup_fold = 2;
          """
      )
      self.wait_predictor("proj", "my_pycaret_regr_model")
  
      # run predict
      ret = self.run_sql(
          """
              SELECT prediction_label
              FROM pg.insurance as t
              JOIN proj.my_pycaret_regr_model AS m;
          """
      )
  
      assert int(ret["prediction_label"].iloc[0]) == 3822
  
file_path: mindsdb/mindsdb/integrations/handlers/pycaret_handler/test/test_pycaret.py
imports: import importlib
  import time
  from unittest.mock import patch
  import pandas
  import pytest
  from mindsdb_sql import parse_sql
  from tests.unit.executor_test_base import BaseExecutorTest
label_desc: Use image node_id 4 for calling the TestPyCaret obj's underlying member method code with example usage: obj.test_regression(mock_handler) without return types