code: def test_valid_rank(self, input_fn):
      for rank in range(2, 5):
          inputs = input_fn(rank)
          utils.assert_rank(inputs, rank)
          utils.assert_minimum_rank(inputs, rank - 2)
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import initializers
  from sonnet.src import test_utils
  from sonnet.src import utils
  import tensorflow
label_desc: Use image node_id 1 for calling the AssertRankTest obj's underlying member method code with example usage: obj.test_valid_rank(input_fn) without return types