code: def setUp(self):
      super(CustomModelTest, self).setUp()
      self._model = MockCustomModel(model_spec=None, shuffle=False)
      self._model._model = test_util.build_model(
          input_shape=[4], num_classes=2
      )
  
file_path: mediapipe/mediapipe/model_maker/python/core/tasks/custom_model_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import tensorflow
  from mediapipe.model_maker.python.core.tasks import custom_model
  from mediapipe.model_maker.python.core.utils import test_util
label_desc: Use image node_id 1 for calling the CustomModelTest obj's underlying member method code with example usage: obj.setUp() without return types