code: def setUp(self):
      self.extraction = models.Extraction()
      self.extraction.main_dataset["source"] = "".join(
          [
              "from sklearn.datasets import load_digits\n",
              "\n",
              "\n",
              "def extract_main_dataset():\n",
              "    X, y = load_digits(return_X_y=True)\n",
              "    return X, y",
          ]
      )
  
file_path: xcessiv/xcessiv/tests/test_models.py
imports: from __future__ import absolute_import, print_function, division, unicode_literals
  import unittest
  from xcessiv import models
  from sklearn.ensemble import RandomForestClassifier
label_desc: Use image node_id 1 for calling the TestReturnTrainDataFromJSON obj's underlying member method code with example usage: obj.setUp() without return types