code: def setup_method(self, method):
      OrcaContext.train_data_store = "DRAM"
  
imports: import os
  import shutil
  import tempfile
  from unittest import TestCase
  import numpy
  import tensorflow
  from bigdl.orca.learn.trigger import SeveralIteration
  from pyspark.sql.context import SQLContext
  import bigdl.orca.data.pandas
  from bigdl.dllib.nncontext import init_nncontext
  from bigdl.orca import OrcaContext
  from bigdl.orca.data.tf.data import Dataset
  from bigdl.orca.learn.tf.estimator import Estimator
  from bigdl.dllib.utils.tf import save_tf_checkpoint, load_tf_checkpoint, get_checkpoint_state
label_desc: Use image node_id 1 for calling the TestEstimatorForGraph obj's underlying member method code with example usage: obj.setup_method(method) without return types