docstr: Make sure we correctly handle non-contiguous range of blocks.
docstr_tok: ["Make","sure","we","correctly","handle","non-contiguous","range","of","blocks","."]
code: def test_ids_nonrange(self):
      """Make sure we correctly handle non-contiguous range of blocks."""
      np.random.seed(0)
      # TODO(b/172564129): This test is not correct.
      block_ints = [
          b.value
          for b in block_builder.BlockType
          if b.value not in [126, 127, 128, 129]
      ]
      blocks_to_use = np.random.choice(block_ints, 3)
      architectures = np.array([blocks_to_use])
      losses = [1.0]
      best = _get_suggestion(architectures, blocks_to_use, losses)
      for blockid in best:
          self.assertIn(blockid, blocks_to_use)
  
imports: from absl.testing import parameterized
  import mock
  from model_search import block_builder
  from model_search import hparam
  from model_search.metadata import trial
  from model_search.proto import phoenix_spec_pb2
  from model_search.search import linear_model
  from model_search.search import test_utils
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 6 for calling the LinearModelTest obj's underlying member method code with example usage: obj.test_ids_nonrange() without return types