code: def test_get_allowed_depth(
      self, num_completed_trials, expected, depth_thresholds=None
  ):
      actual = common.get_allowed_depth(
          num_completed_trials, depth_thresholds, max_depth=5
      )
      self.assertEqual(expected, actual)
  
file_path: model_search/model_search/search/common_test.py
imports: from absl.testing import parameterized
  from model_search.search import common
  import tensorflow.compat.v2
label_desc: Use image node_id 1 for calling the CommonTest obj's underlying member method code with example usage: obj.test_get_allowed_depth(num_completed_trials, expected, depth_thresholds) without return types