code: def setup_method(self, method):
      tf.keras.backend.clear_session()
      super(TestTFDataset, self).setup_method(method)
  
imports: import pytest
  from pyspark.ml.linalg import DenseVector
  from zoo.feature.common import ChainedPreprocessing, FeatureSet
  from zoo.feature.image import *
  from test.zoo.pipeline.utils.test_utils import ZooTestCase
  from zoo.pipeline.api.keras.optimizers import Adam
  from zoo.tfpark import TFNet, TFOptimizer
  import tensorflow
  import numpy
  import os
  from zoo.tfpark import KerasModel, TFDataset
label_desc: Use image node_id 1 for calling the TestTFDataset obj's underlying member method code with example usage: obj.setup_method(method) without return types