code: def setup_method(self):
      super().setup_method()
      self.batch_size = 5
      self.input_width = 10
      self.input_height = 10
      flat_dim = self.input_width * self.input_height * 3
      self.obs_input = np.ones((self.batch_size, flat_dim))
      self.input_dim = (self.input_width, self.input_height, 3)
      self._input_ph = tf.compat.v1.placeholder(
          tf.float32, shape=(None, flat_dim), name="input"
      )
  
file_path: garage/tests/garage/tf/models/test_cnn_model.py
imports: import pickle
  import numpy
  import pytest
  import tensorflow
  from garage.tf.models import CNNModel, CNNModelWithMaxPooling
  from tests.fixtures import TfGraphTestCase
label_desc: Use image node_id 1 for calling the TestCNNModel obj's underlying member method code with example usage: obj.setup_method() without return types